Update app.py

app.py

@@ -31,17 +31,22 @@ def start_bot(userid, password, api_key_input):
 
 
 def answer(message, history):
-    answer_md, tables_display, images_display, retrieved_display = bot.get_answer(message)
-    
-    # Combine all parts into a single response string for chat
-    combined_response = f"{answer_md}\n\n{tables_display}"
-    
-    # Add images as markdown
-    if images_display:
-        combined_response += "\n\n" + "\n\n".join(images_display)
-    
-    return combined_response
-
+    global bot
+    if bot is None:
+        return "Please log in or register to start the bot."
+    try:
+        answer_md, tables_display, images_display, retrieved_display = bot.get_answer(message)
+        
+        # Combine all parts into a single response string for chat
+        combined_response = f"{answer_md}\n\n{tables_display}"
+        
+        # Add images as markdown
+        if images_display:
+            combined_response += "\n\n" + "\n\n".join(images_display)
+        
+        return combined_response
+    except Exception as e:
+        return f"An error occurred: {e}"
 
 
 with gr.Blocks(fill_height=True, fill_width=True) as app:
@@ -83,7 +88,7 @@ with gr.Blocks(fill_height=True, fill_width=True) as app:
     # Function connections
     login_btn.click(
         start_bot,
-        inputs=[userid_login, password_login],
+        inputs=[userid_login, password_login, api_key_register],
         outputs=[login_output, login_register_section, chat_section]
     )
 
@@ -94,4 +99,4 @@ with gr.Blocks(fill_height=True, fill_width=True) as app:
     )
 
 
-app.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=False)
+app.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=False)