Update app.py

app.py

@@ -21,11 +21,16 @@ def mainFun(query,file):
     return text,answer_p,summarize_text(answer_main)
 
 def mainFun2(temp):
-    return fin_ner(temp)
-    #return temp
-
+    return fin_ner(temp.split('Probability:')[0])
 
+def mainFun3(temp):
+    return score_fincat(temp.split('Probability:')[0])
 
+def mainFun4(temp):
+    return get_sustainability(temp.split('Probability:')[0])
+    
+def mainFun5(temp):
+    return fls(temp.split('Probability:')[0])    
 
 
 demo = gr.Blocks()
@@ -34,6 +39,7 @@ with demo:
     text = gr.Textbox(lines=10)
     selected_ques=gr.Dropdown(choices=questions,label='SEARCH QUERY')
     b1 = gr.Button("Analyze File")
+    
     answer = gr.Textbox(lines=2)
     summarize = gr.Textbox(lines=2)
     b1.click(mainFun, inputs=[selected_ques,txt_file], outputs=[text,answer,summarize])
@@ -42,14 +48,16 @@ with demo:
     label_ner = gr.HighlightedText()
     b2.click(mainFun2,inputs=answer,outputs=label_ner)
     
-    
     b3=gr.Button("Get CLAIM")
-    label_claim = gr.Label()
+    label_claim = gr.HighlightedText()
+    b3.click(mainFun3,inputs=answer,outputs=label_claim)
     
     b4=gr.Button("Get SUSTAINABILITY")
     label_sus = gr.Label()
+    b4.click(mainFun4,inputs=answer,outputs=label_sus)
     
     b5=gr.Button("Get FLS")
     label_fls = gr.Label()
+    b5.click(mainFun5,inputs=answer,outputs=label_fls)
     
 demo.launch()