Update app.py

app.py

@@ -46,8 +46,10 @@ retriever = vectorstore.as_retriever()
 
 from langchain_core.prompts import PromptTemplate
 
-template = ("""You are a pharmacist and medical expert.
+template = ("""You are a pharmacist and medical expert. 
     Use the provided context to answer the question.
+    If the question is related to medical condition, drug name 
+    and side effects that are not in the context, look online and answer them.
     If you don't know the answer, say so. Explain your answer in detail.
     Do not discuss the context in your response; just provide the answer directly.
 
@@ -82,63 +84,82 @@ def rag_memory_stream(text):
 # Title and description for the app
 title = "AI Medical Assistant for Drug Information and Side Effects"
 description = """
+<div class="description">
 This AI-powered chatbot is designed to provide reliable information about drugs, their side effects, and related medical conditions.
-It utilizes the Groq API and LangChain to deliver real-time, accurate responses. 
+It utilizes the Groq API and LangChain to deliver real-time, accurate responses.
 
 Ask questions like:
-1. What are the side effects of taking aspirin daily?
-2. What is the recommended treatment for a common cold?
-3. What is the disease for constant fatigue and muscle weakness?
-4. What are the symptoms of diabetes?
-5. How can hypertension be managed?
-
-**Disclaimer:** This chatbot is for informational purposes only and is not a substitute for professional medical advice.
+<ul>
+    <li>What are the side effects of taking aspirin daily?</li>
+    <li>What is the recommended treatment for a common cold?</li>
+    <li>What is the disease for constant fatigue and muscle weakness?</li>
+    <li>What are the symptoms of diabetes?</li>
+    <li>How can hypertension be managed?</li>
+</ul>
+
+<strong>Disclaimer:</strong> This chatbot is for informational purposes only and is not a substitute for professional medical advice.
+</div>
 """
 
 # Customizing Gradio interface for a better look
+
+# HTML for custom styling
+custom_css = """
+body {
+    background-color: #f9f9f9;
+    font-family: Arial, sans-serif;
+    margin: 0;
+    padding: 0;
+}
+
+#interface-container {
+    max-width: 800px;
+    margin: 50px auto;
+    padding: 20px;
+    background: linear-gradient(145deg, #ffffff, #f0f0f0);
+    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+    border-radius: 10px;
+    border: 1px solid #e0e0e0;
+}
+
+h1 {
+    text-align: center;
+    color: #333;
+}
+
+.description {
+    text-align: justify;
+    color: #555;
+    font-size: 1rem;
+    margin-bottom: 20px;
+}
+
+footer {
+    text-align: center;
+    color: #777;
+    margin-top: 30px;
+    font-size: 0.9rem;
+}
+"""
+
+# Customizing Gradio interface with additional CSS and content
 demo = gr.Interface(
     fn=rag_memory_stream,
     inputs=gr.Textbox(
-        lines=2, 
+        lines=5, 
         placeholder="Type your medical question here...", 
         label="Your Medical Question"
     ),
     outputs=gr.Textbox(
-        lines=10, 
+        lines=15,  # Reduced line count for better layout
         label="AI Response"
     ),
     title=title,
     description=description,
-    theme="compact",  # Adding a compact theme for a polished look
-    allow_flagging="never"
+    css=custom_css,
+    allow_flagging="never",
+    live=True
 )
 
-# # Launching the app
-# demo.launch(share=True)
-
-# import gradio as gr
-
-# def rag_memory_stream(text):
-#     partial_text = ""
-#     for new_text in rag_chain.stream(text):
-#         partial_text += new_text
-#         yield partial_text
-
-# examples = ['I feel dizzy', 'what is the possible sickness for fatigue']
-
-
-
-
-# title = "Real-time AI App with Groq API and LangChain to Answer medical questions"
-# demo = gr.Interface(
-#     title=title,
-#     fn=rag_memory_stream,
-#     inputs="text",
-#     outputs="text",
-#     examples=examples,
-#     allow_flagging="never",
-# )
-
-
 if __name__ == "__main__":
     demo.launch()