Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import gradio as gr
 from src.auth.auth import handle_login
 from src.auth.db import initialize_db
@@ -9,8 +8,6 @@ from src.interface import Interface
 initialize_db()
 load_dotenv()
 
-os.getenv("OPENAI_API_KEY")
-
 bot = None  # Initially, no bot is created until the user logs in or registers
 
 
@@ -34,22 +31,17 @@ def start_bot(userid, password, api_key_input):
 
 
 def answer(message, history):
-    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}"
+    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
+
 
 
 with gr.Blocks(fill_height=True, fill_width=True) as app:
@@ -84,14 +76,14 @@ with gr.Blocks(fill_height=True, fill_width=True) as app:
                 answer,
                 title="🩺 Medico-Bot",
                 examples=["briefly explain me about cancer", "types of skin diseases?"],
-                flagging_options = ['Like', 'Dislike'],
+                flagging_options = ['Like', 'Dislike']
                 )
 
 
     # Function connections
     login_btn.click(
         start_bot,
-        inputs=[userid_login, password_login, api_key_register],
+        inputs=[userid_login, password_login],
         outputs=[login_output, login_register_section, chat_section]
     )