Update app.py

app.py

@@ -51,42 +51,37 @@ if clear_model_button:
     st.session_state["qa_pipeline"] = None
     st.success("Model cleared.")
 
-# Layout for chat
-chat_container = st.container()
-input_container = st.container()
-
 # Chat Conversation Display
-with chat_container:
-    st.subheader("Conversation")
-    for idx, (speaker, message) in enumerate(st.session_state["conversation"]):
-        st.markdown(f"**{speaker}:** {message}")
+st.subheader("Conversation")
+for idx, (speaker, message) in enumerate(st.session_state["conversation"]):
+    if speaker == "You":
+        st.markdown(f"**You:** {message}")
+    else:
+        st.markdown(f"**Model:** {message}")
 
 # Input Area
-with input_container:
-    if st.session_state["qa_pipeline"]:
-        user_input = st.text_input(
-            "Enter your query:",
-            value=st.session_state["user_input"],  # Use session state for persistence
-            key="chat_input",
-        )
-        send_button_clicked = st.button("Send", key="send_button")
-        if send_button_clicked and user_input.strip():
-            # Process the input
+if st.session_state["qa_pipeline"]:
+    user_input = st.text_input(
+        "Enter your query:",
+        value=st.session_state["user_input"],  # Use session state for persistence
+        key="chat_input",
+    )
+    if st.button("Send", key="send_button"):
+        if user_input:
             with st.spinner("Generating response..."):
                 try:
                     response = st.session_state["qa_pipeline"](f"Q: {user_input}", max_length=400)
                     generated_text = response[0]["generated_text"]
-
-                    # Append to conversation
                     st.session_state["conversation"].append(("You", user_input))
                     st.session_state["conversation"].append(("Model", generated_text))
-
-                    # Clear input after submission
-                    st.session_state["user_input"] = ""
+                    st.session_state["user_input"] = ""  # Clear input after submission
+                    st.experimental_rerun()  # Rerun to update the conversation display
                 except Exception as e:
                     st.error(f"Error generating response: {e}")
 
 # Clear Conversation
 if clear_conversation_button:
     st.session_state["conversation"] = []
+    st.session_state["user_input"] = ""  # Clear input field
     st.success("Conversation cleared.")
+    st.experimental_rerun()  # Rerun to refresh the cleared conversation