Update app.py

app.py

@@ -137,7 +137,8 @@ def query(category, text):
         ]
     )
     answer = result.generations[0][0].text
-    yield gr.Chatbot.update([[question, answer]])
+    #yield gr.Chatbot.update([[question, answer]])
+    return answer
 
 
 PARSING_REGEXP = r"\*\*System:\*\*\n([\s\S]+)\n\n\*\*Human:\*\*\n([\s\S]+)"
@@ -147,12 +148,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