Update app.py

app.py

@@ -137,8 +137,7 @@ def query(category, text):
         ]
     )
     answer = result.generations[0][0].text
-    #yield gr.Chatbot.update([[question, answer]])
-    return answer
+    yield gr.Chatbot.update([[question, answer]])
 
 
 PARSING_REGEXP = r"\*\*System:\*\*\n([\s\S]+)\n\n\*\*Human:\*\*\n([\s\S]+)"
@@ -148,12 +147,12 @@ def parse(category, chatbot):
     """Takes the chatbot prompt and response and updates the extracted information"""
     global PARSING_REGEXP
 
-    answer = chatbot
+    
         
-    # try:
-    #     answer = next(chatbot)[1]  
-    # except StopIteration:
-    #     answer = "" 
+    try:
+        answer = next(chatbot)[1]  
+    except StopIteration:
+        answer = "" 
     
     if category not in Category.__members__:
         # Handle the case when an invalid category is provided
@@ -205,7 +204,7 @@ def process_and_output_files(input_files):
             "File Name": file.name,
             "Extracted Text": text,
             "Category": category,
-            "Chatbot Response": chatbot_response,  # Access the first element as a list
+            # "Chatbot Response": chatbot_response[0][1],  # Access the first element as a list
             "Parsed Information": parsed_info,
         })
 
@@ -418,4 +417,4 @@ page.queue(
     concurrency_count=20,
     max_size=1,
 )
-page.launch(show_api=True, show_error=True, debug=True)
+page.launch(show_api=True, show_error=True, debug=True)