Upload app.py

app.py

@@ -15,30 +15,25 @@ llm = HuggingFaceEndpoint(
 )
 llm_engine_hf = ChatHuggingFace(llm=llm)
 
+# Update the template to extract topic information
 template_classify = '''
-Please carefully read the following text. The text is written in {LANG} language:
+Please read the following text written in {LANG} language and extract the main topics discussed in it.
+You can list more than one topic or topics sentence by sentence. List the topics clearly.
 
 <text>  
 {TEXT}  
 </text>
-
-After reading it, I want you to classify it in three groups: Positive, Negative, or Neutral.
-Your final response MUST contain only the response, no other text.
-Example:
-Positive
-Negative
-Neutral
 '''
 
 template_json = '''
-Your task is to read the following text, convert it to json format using 'Answer' as key and return it.
+Your task is to read the following extracted topics and convert them into JSON format using 'Topics' as the key.
+
 <text>
 {RESPONSE}
 </text>
 
-Your final response MUST contain only the response, no other text.
-Example:
-{{"Answer":"Positive"}}
+The final response should be in this format:
+{{"Topics": ["Topic1", "Topic2", ...]}}
 '''
 json_output_parser = JsonOutputParser()
 
@@ -67,7 +62,7 @@ def classify_text(text):
     parsed_output = json_output_parser.parse(response)
     end = time.time()
     duration = end - start
-    return parsed_output, duration #['Answer']
+    return parsed_output, duration
 
 # Create the Gradio interface
 def gradio_app(text):
@@ -77,11 +72,11 @@ def gradio_app(text):
 def create_gradio_interface():
     with gr.Blocks() as iface:
         text_input = gr.Textbox(label="Text to Classify")
-        output_text = gr.Textbox(label="Classification")
+        output_text = gr.Textbox(label="Extracted Topics")
         time_taken = gr.Textbox(label="Time Taken (seconds)")
-        submit_btn = gr.Button("Classify")
+        submit_btn = gr.Button("Extract Topics")
 
-        submit_btn.click(fn=classify_text, inputs=text_input, outputs=[output_text, time_taken])
+        submit_btn.click(fn=gradio_app, inputs=text_input, outputs=[output_text, time_taken])
 
     iface.launch()