Update app.py

app.py

@@ -1,38 +1,35 @@
-import os
-import gradio as gr
-from rag import generate_answer  # Import the generate_answer function from rag.py
-
-def chat_with_bot(query: str) -> tuple:
-    """
-    Chat with the bot using the provided query.
-    
-    Args:
-        query: User's question
-        
-    Returns:
-        tuple: Generated answer and extracted contents
-    """
-    response = generate_answer(query)  # Call the existing generate_answer function
-    
-    return response
-
-def main():
-    """
-    Main function to create and launch the Gradio interface.
-    """
-    try:
-        # Create Gradio interface
-        interface = gr.Interface(
-            fn=chat_with_bot,  # Use the chat function
-            inputs="text",  # Input type
-            outputs="text",  # Output type
-            title="RAG-LLM based Medical Chatbot",  # Title of the app
-            description="Ask your medical questions and get answers from the chatbot."  # Description
-        )
-        interface.launch(share=True)  # Launch the Gradio app
-        
-    except Exception as e:
-        print(f"An error occurred: {str(e)}")
-
-if __name__ == "__main__":
+import os
+import gradio as gr
+from rag import generate_answer  # Import the generate_answer function from rag.py
+
+def chat_with_bot(query: str) -> tuple:
+    """
+    Chat with the bot using the provided query.
+    """
+    try:
+        response = generate_answer(query)  # Ensure this function is working correctly
+    except Exception as e:
+        print(f"Error in generate_answer: {str(e)}")  # More specific error handling
+        return "Error generating answer", None  # Return a default response on error
+    
+    return response
+
+def main():
+    """
+    Main function to create and launch the Gradio interface.
+    """
+    try:
+        interface = gr.Interface(
+            fn=chat_with_bot,
+            inputs="text",
+            outputs="text",
+            title="RAG-LLM based Medical Chatbot",
+            description="Ask your medical questions and get answers from the chatbot."
+        )
+        interface.launch(share=True)
+        
+    except Exception as e:
+        print(f"An error occurred while launching the interface: {str(e)}")  # More context on errors
+
+if __name__ == "__main__":
     main()