Update app.py

app.py

@@ -37,22 +37,22 @@ def get_sustainability(text):
   return highlight
 #SUSTAINABILITY ENDS
 
-##Summarization 
+##Summarization
+summarizer = pipeline("summarization", model="knkarthick/MEETING_SUMMARY") 
 def summarize_text(text):
-    summarizer = pipeline("summarization", model="knkarthick/MEETING_SUMMARY")
     resp = summarizer(text)
     stext = resp[0]['summary_text']
     return stext
 
 ##Forward Looking Statement
+fls_model = pipeline("text-classification", model="yiyanghkust/finbert-fls", tokenizer="yiyanghkust/finbert-fls")
 def fls(text):
-    fls_model = pipeline("text-classification", model="yiyanghkust/finbert-fls", tokenizer="yiyanghkust/finbert-fls")
     results = fls_model(split_in_sentences(text))
     return make_spans(text,results) 
     
 ##Company Extraction
+ner=pipeline('ner',model='Jean-Baptiste/camembert-ner-with-dates',tokenizer='Jean-Baptiste/camembert-ner-with-dates', aggregation_strategy="simple")
 def fin_ner(text):
-    ner=pipeline('ner',model='Jean-Baptiste/camembert-ner-with-dates',tokenizer='Jean-Baptiste/camembert-ner-with-dates', aggregation_strategy="simple")
     replaced_spans = ner(text)
     return replaced_spans  
     
@@ -89,10 +89,12 @@ def quad(query,file):
                 raw_answer=data['0'][i]['text']
                 answer += f"Answer {i+1}: {data['0'][i]['text']} -- \n"
                 answer += f"Probability: {round(data['0'][i]['probability']*100,1)}%\n\n"
-    summarizer = pipeline("summarization", model="knkarthick/MEETING_SUMMARY")
-    resp = summarizer(answer)
-    stext = resp[0]['summary_text']
-    return stext,answer            
+    #summarizer = pipeline("summarization", model="knkarthick/MEETING_SUMMARY")
+    #resp = summarizer(answer)
+    #stext = resp[0]['summary_text']
+    
+    
+    return answer,summarize_text(answer),fin_ner(answer),get_sustainability(answer),fls(answer)           
                 
                    
 # b6 = gr.Button("Get Sustainability")
@@ -102,5 +104,7 @@ def quad(query,file):
 #iface = gr.Interface(fn=get_sustainability, inputs="textbox", title="CONBERT",description="SUSTAINABILITY TOOL", outputs=gr.HighlightedText(), allow_flagging="never")
 #iface.launch()
 
-iface = gr.Interface(fn=quad, inputs=[gr.inputs.Textbox(label='SEARCH QUERY'),gr.inputs.File(label='TXT FILE')], title="CONBERT",description="SUSTAINABILITY TOOL",article='Article', outputs=[gr.outputs.Textbox(label='Answer'),gr.outputs.Textbox(label='Summary')], allow_flagging="never")
+iface = gr.Interface(fn=quad, inputs=[gr.inputs.Textbox(label='SEARCH QUERY'),gr.inputs.File(label='TXT FILE')], title="CONBERT",description="SUSTAINABILITY TOOL",article='Article', outputs=[gr.outputs.Textbox(label='Answer'),gr.outputs.Textbox(label='Summary'),gr.HighlightedText(label='NER'),gr.HighlightedText(label='SUSTAINABILITY'),gr.HighlightedText(label='FLS')], allow_flagging="never")
+
+
 iface.launch()