Update app.py

app.py

@@ -45,49 +45,35 @@ rag_chain = (
     | llm
     | StrOutputParser()
 )
+
 import gradio as gr
 
-# Function to handle prompt and query the RAG chain
-def handle_prompt(message, history):
+def rag_memory_stream(message, history):
     partial_text = ""
     for new_text in rag_chain.stream(message):
         partial_text += new_text
         yield partial_text
 
-
-
-if __name__=="__main__":
-
-# Predefined messages and examples
-    greetingsmessage = """Hello! Welcome to MediGuide ChatBot. I'm here to provide you with quick and accurate information on medical drugs. 
-    Whether you need details on usage, side effects, etc., feel free to ask. Let's enhance patient care together!"""
-    
-    description = """
-    
-    Hello! Welcome to MediGuide ChatBot,AI-powered assistant designed to facilitate healthcare providers to make informed decision-making by providing reliable information about various medical drugs, including their uses, side effects, contraindications and classification."""
-    
-    example_questions = [
+examples = [
                     "What are the Side effects of Doxycycline",
                     "Can Doxycycline Treat Acnes",
                     "Which Classification is Doxycycline"
                     ]
-    
-    # Define customized Gradio chatbot
-    chatbot = gr.Chatbot([{"role":"Medical Expert", "content":greetingsmessage}],
-                    type="messages",
-                    avatar_images=["./avatar-1.jpg","./images.png"])
-    
-    # Define Gradio interface
-    demo = gr.ChatInterface(handle_prompt,
+
+description = "Hello! Welcome to MediGuide ChatBot,AI-powered assistant designed to facilitate healthcare providers to make informed decision-making by providing reliable information about various medical drugs, including their uses, side effects, contraindications and classification"
+
+
+title = "MediGuide ChatBot"
+demo = gr.ChatInterface(fn=rag_memory_stream,
                         type="messages",
-                        title="MediGuide ChatBot",
-                        examples=example_questions,
-                        theme=gr.themes.Soft(),
+                        title=title,
                         description=description,
-                        #cache_examples=False,
-                        chatbot=chatbot)
-
-    demo.launch()
+                        fill_height=True,
+                        examples=examples,
+                        theme=gr.themes.Soft(),
+)
 
 
+if __name__ == "__main__":
+    demo.launch()