Update app.py

app.py

@@ -5,21 +5,66 @@ from sus_fls import get_sustainability,fls
 from Cuad_others import quad,summarize_text,fin_ner
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-     
-#CUAD STARTS    
 def load_questions():
     questions = []
-    with open('questions.txt') as f:
+    with open('questionshort.txt') as f:
         questions = f.readlines()
     return questions
 questions = load_questions()
 
+answer_main=''     
 def mainFun(query,file):
-    answer,answer_p=quad(query,file.name)
-    return answer_p,summarize_text(answer),fin_ner(answer),score_fincat(answer),get_sustainability(answer),fls(answer)
-                
-                   
-iface = gr.Interface(fn=mainFun, inputs=[gr.Dropdown(choices=questions,label='SEARCH QUERY'),gr.inputs.File(label='TXT FILE')], title="CONBERT",description="CONTRACT REVIEW TOOL",article='Article', outputs=[gr.outputs.Textbox(label='Answer'),gr.outputs.Textbox(label='Summary'),gr.HighlightedText(label='NER'),gr.HighlightedText(label='CLAIM'),gr.HighlightedText(label='SUSTAINABILITY'),gr.HighlightedText(label='FLS')], allow_flagging="never")
+    text=''
+    with open(file.name) as f:
+        text = f.read()
+    answer_main,answer_p=quad(query,file)
+    return text,answer_p,summarize_text(answer_main)
 
+def mainFun2(temp):
+    return fin_ner(temp.split('Probability:')[0])
 
-iface.launch()
+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()
+with demo:
+
+#FETCH AND DISPLAY CONTRACT
+    txt_file = gr.File(label='CONTRACT')
+    text = gr.Textbox(lines=10)
+    selected_ques=gr.Dropdown(choices=questions,label='SEARCH QUERY')
+    b1 = gr.Button("Analyze File")
+    
+#DISPLAY ANSWER AND SUMMARIZATION    
+    answer = gr.Textbox(lines=2)
+    summarize = gr.Textbox(lines=2)
+    b1.click(mainFun, inputs=[selected_ques,txt_file], outputs=[text,answer,summarize])
+
+#DISPLAY NER    
+    b2=gr.Button("Get NER")
+    label_ner = gr.HighlightedText()
+    b2.click(mainFun2,inputs=answer,outputs=label_ner)
+
+#DISPLAY CLAIMS        
+    b3=gr.Button("Get CLAIM")
+    label_claim = gr.HighlightedText()
+    b3.click(mainFun3,inputs=answer,outputs=label_claim)
+
+#DISPLAY SUSTAINABILITY        
+    b4=gr.Button("Get SUSTAINABILITY")
+    label_sus = gr.HighlightedText()
+    b4.click(mainFun4,inputs=answer,outputs=label_sus)
+
+#DISPLAY FLS        
+    b5=gr.Button("Get FLS")
+    label_fls = gr.HighlightedText()
+    b5.click(mainFun5,inputs=answer,outputs=label_fls)
+    
+demo.launch()