Update app.py

app.py

@@ -240,51 +240,75 @@ def diabetes_bot(query, pdf_file=None):
     return answer + usage_message
 
 # Gradio interface
+import gradio as gr
 
-
+def chat_wrapper(query, pdf, history):
+    # Initialize history if empty
+    if history is None:
+        history = []
+    # If no query is provided, return the current history without changes
+    if query.strip() == "":
+        return history, "", None, history
+    # Call your existing diabetes_bot function to generate an answer
+    answer = diabetes_bot(query, pdf)
+    # Append the new interaction to the history as a tuple (user, bot)
+    history.append((query, answer))
+    # Return the updated conversation, and clear the query and pdf inputs
+    return history, "", None, history
+
+def clear_all():
+    # Clear conversation history and inputs
+    return [], "", None, []
 
 with gr.Blocks() as app:
-    gr.Markdown(
+    gr.HTML(
     """
-    # Diabetes Research ChatBot Powered By Gemini 1.5 Flash And Pinecone 🩺
-    **Powered by the latest diabetes research, Running on Gemini 1.5 Flash API**  
-    Ask questions about diabetes directly or upload a research paper (up to 10 pages) for specific Q&A.
-    
+    <h1 style="text-align:center;">Diabetes Research ChatBot Powered By Gemini 1.5 Flash And Pinecone 🩺</h1>
+    <p style="text-align:center;"><strong>Powered by the latest diabetes research, Running on Gemini 1.5 Flash API</strong></p>
+    <p style="text-align:center;">Ask questions about diabetes directly or upload a research paper (up to 10 pages) for specific Q&A.</p>
     <br>
-    <div style="border: 1px solid #ccc; border-radius: 5px; padding: 10px; background-color: #f9f9f9;">
+    <div style="border: 1px solid #ccc; border-radius: 5px; padding: 10px; background-color: #f9f9f9; margin:auto; width:80%;">
         <strong>Disclaimer:</strong>  
         The information provided by this chatbot is for research and informational purposes only and is not intended to substitute professional medical advice, diagnosis, or treatment. Always seek the advice of your physician or other qualified health provider with any questions you may have regarding a medical condition.
     </div>
     <br>
+    <p style="text-align:center;">Example Queries:</p>
+    <ul style="text-align:center;">
+        <li>"What is new with type 2 diabetes management research?"</li>
+        <li>"Tell me about new treatments in diabetes if any."</li>
+    </ul>
     """
     )
-
+    
+    # Chatbot component to display conversation history
+    chatbot = gr.Chatbot(label="Conversation").style(height=400)
+    
+    # Input row for query and PDF file (with PDF box sized smaller)
     with gr.Row():
-        query_input = gr.Textbox(label="Ask a Question", placeholder="Type your diabetes-related query here...", lines=3)
+        query_input = gr.Textbox(label="Ask a Question", placeholder="Type your diabetes-related query here...", lines=4)
         with gr.Column(scale=0.5):
             pdf_input = gr.File(label="Upload a PDF (optional, max 10 pages)", file_types=[".pdf"])
-
+    
+    # Row for Submit and Clear buttons
     with gr.Row():
-        submit_button = gr.Button("Submit")
+        submit_button = gr.Button("Ask", variant="primary")
         clear_button = gr.Button("Clear")
-
-    output = gr.Textbox(label="Answer", lines=15)
-
-    # Function to clear inputs and output
-    def clear_all():
-        return "", None, ""
-
+    
+    # State to maintain conversation history
+    state = gr.State([])
+    
+    # On submit, update the conversation and clear inputs; outputs: chatbot, query_input, pdf_input, state
     submit_button.click(
-        fn=diabetes_bot,
-        inputs=[query_input, pdf_input],
-        outputs=output
+        fn=chat_wrapper,
+        inputs=[query_input, pdf_input, state],
+        outputs=[chatbot, query_input, pdf_input, state]
     )
-
+    
+    # Clear all the components including conversation history
     clear_button.click(
         fn=clear_all,
         inputs=[],
-        outputs=[query_input, pdf_input, output]
+        outputs=[chatbot, query_input, pdf_input, state]
     )
 
-# Launch the app
-app.launch()
+app.launch()