Update app.py

app.py

@@ -108,16 +108,28 @@ def generate_answer(message, choice):
     addresses = extract_addresses(response['output'])
     return response['output'], addresses
 
+# def bot(history, choice):
+#     if not history:
+#         return history
+#     response = generate_answer(history[-1][0], choice)
+#     history[-1][1] = ""
+#     for character in response:
+#         history[-1][1] += character
+#         time.sleep(0.05)
+#         yield history
+
+# Define the bot function
 def bot(history, choice):
     if not history:
         return history
-    response = generate_answer(history[-1][0], choice)
+    response, addresses = generate_answer(history[-1][0], choice)
     history[-1][1] = ""
     for character in response:
         history[-1][1] += character
         time.sleep(0.05)
         yield history
-
+    if addresses:
+        return history, addresses
 def add_message(history, message):
     history.append((message, None))
     return history, gr.Textbox(value="", interactive=True, placeholder="Enter message or upload file...", show_label=False)