Update app.py

app.py

@@ -7,7 +7,7 @@ df = df[['drug_name', 'medical_condition', 'side_effects']]
 df.dropna(inplace=True)
 
 context_data = []
-for i in range(10):
+for i in range(2):
     context = " | ".join([f"{col}: {df.iloc[i][col]}" for col in df.columns])
     context_data.append(context)
 
@@ -90,20 +90,28 @@ Disclaimer: This chatbot is for informational purposes only and not a substitute
 """
 
 # Customizing Gradio interface for a better look
-demo = gr.Interface(
-    fn=rag_memory_stream,
-    inputs=gr.Textbox(
-        lines=5, 
-        placeholder="Type your medical question here...", 
-        label="Your Medical Question"
-    ),
-    outputs=gr.Textbox(
-        lines=15,  # Reduced line count for better layout
-        label="AI Response"
-    ),
-    title=title,
-    description=description,
-    allow_flagging="never"
+# demo = gr.Interface(
+#     fn=rag_memory_stream,
+#     inputs=gr.Textbox(
+#         lines=5, 
+#         placeholder="Type your medical question here...", 
+#         label="Your Medical Question"
+#     ),
+#     outputs=gr.Textbox(
+#         lines=15,  # Reduced line count for better layout
+#         label="AI Response"
+#     ),
+#     title=title,
+#     description=description,
+#     allow_flagging="never"
+# )
+
+demo = gr.ChatInterface(fn=rag_memory_stream,
+                        type="messages",
+                        title=title,
+                        description=description,
+                        fill_height=True,
+                        theme="glass",
 )
 
 if __name__ == "__main__":