Update app.py

app.py

@@ -2,33 +2,17 @@ import os
 import gradio as gr
 from rag import generate_answer  
 
-def chat_with_bot(query: str) -> str:
+def chat_with_bot(query: str) -> tuple:
     """
-    Chat with the bot using the provided query and format the output for readability.
+    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"  # Return a default response on error
+        return "Error generating answer", None  # Return a default response on error
     
-    # Formatting response as Markdown
-    formatted_response = f"""
-    **Summary:**
-    {response.get("summary", "No summary available.")}
-
-    **Details:**
-    {response.get("details", "No details available.")}
-
-    **Cautions:**
-    {response.get("cautions", "No cautions available.")}
-
-    **Disclaimer:**
-    {response.get("disclaimer", "No disclaimer available.")}
-
-    If you have any further questions or need more details, feel free to ask!
-    """
-    return formatted_response
+    return response
 
 def main():
     """
@@ -38,7 +22,7 @@ def main():
         interface = gr.Interface(
             fn=chat_with_bot,
             inputs="text",
-            outputs="markdown",  # Use Markdown for better formatting
+            outputs="text",
             title="RAG-LLM based Medical Chatbot",
             description="Ask your medical questions and get answers from the chatbot."
         )
@@ -48,4 +32,4 @@ def main():
         print(f"An error occurred while launching the interface: {str(e)}")  # More context on errors
 
 if __name__ == "__main__":
-    main()
+    main()