Update app.py

app.py

@@ -17,44 +17,51 @@ def get_gemini_response(prompt):
 
 # Streamlit app configuration
 st.set_page_config(page_title="Med ChatBot")
-st.header("Medical ChatBot")
+st.title("Medical ChatBot")
 
 # Initialize session state for chat history
 if "chat_history" not in st.session_state:
     st.session_state["chat_history"] = []
 
-# Input and submission button
-input_text = st.text_input("Input: ", key="input")
-submit = st.button("Ask the question")
-
-if submit and input_text:
-    # Context for the LLM with history included
-    chat_history_text = " ".join([f"{role}: {text}" for role, text in st.session_state["chat_history"]])
+# Function to handle submission (without the button)
+def submit_input():
+    input_text = st.session_state["input"]
     
-    context = (
-        "You are a medical chatbot designed to assist users in understanding their symptoms. "
-        "Provide clear, concise, and informative responses based on NHS guidelines. "
-        "Avoid technical jargon and code snippets. If asked a question unrelated to medical topics, "
-        "respond with: 'I am a medical bot and I don't have that knowledge.' "
-        f"Previous conversation: {chat_history_text} "
-    )
-
-    prompt = f"{context} User's latest input: {input_text}"  # Include the latest user input
-    response = get_gemini_response(prompt)
-
-    # Add user query to session state chat history
-    st.session_state['chat_history'].append(("You", input_text))
-
-    st.subheader("The Response is")
-    # If the response is a list of chunks, combine them into a single string
-    if isinstance(response, list):
-        full_response = " ".join([chunk.text for chunk in response])
-        st.write(full_response)
-        st.session_state['chat_history'].append(("Bot", full_response))
-    else:
-        st.write(response)
-        st.session_state['chat_history'].append(("Bot", response))
+    if input_text:
+        # Context for the LLM with history included
+        chat_history_text = " ".join([f"{role}: {text}" for role, text in st.session_state["chat_history"]])
+        
+        context = (
+            "You are a medical chatbot designed to assist users in understanding their symptoms. "
+            "Provide clear, concise, and informative responses based on NHS guidelines. "
+            "Avoid technical jargon and code snippets. If asked a question unrelated to medical topics, "
+            "respond with: 'I am a medical bot and I don't have that knowledge.' "
+            f"Previous conversation: {chat_history_text} "
+        )
+
+        prompt = f"{context} User's latest input: {input_text}"  # Include the latest user input
+        response = get_gemini_response(prompt)
+
+        # Add user query to session state chat history
+        st.session_state['chat_history'].append(("You", input_text))
 
-st.subheader("The Chat History is")
+        # Add bot response to session state chat history
+        if isinstance(response, list):
+            full_response = " ".join([chunk.text for chunk in response])
+            st.session_state['chat_history'].append(("Bot", full_response))
+        else:
+            st.session_state['chat_history'].append(("Bot", response))
+
+        # Clear input field after submission
+        st.session_state["input"] = ""
+
+# Chat input (without button, submit on "Enter")
+st.text_input("Ask the medical chatbot:", key="input", on_change=submit_input)
+
+# Display chat history in a chatbox style
+st.subheader("Chat History")
 for role, text in st.session_state['chat_history']:
-    st.write(f"{role}: {text}")
+    if role == "You":
+        st.markdown(f"<div style='text-align: right; color: blue;'>{role}: {text}</div>", unsafe_allow_html=True)
+    else:
+        st.markdown(f"<div style='text-align: left; color: green;'>{role}: {text}</div>", unsafe_allow_html=True)