Update app.py

app.py

@@ -59,10 +59,7 @@ input_container = st.container()
 with chat_container:
     st.subheader("Conversation")
     for idx, (speaker, message) in enumerate(st.session_state["conversation"]):
-        if speaker == "You":
-            st.write(f"You ({idx}):", message, key=f"you_{idx}", disabled=False)
-        else:
-            st.write(f"Model ({idx}):", message, key=f"model_{idx}", disabled=False)
+        st.markdown(f"**{speaker}:** {message}")
 
 # Input Area
 with input_container:
@@ -71,20 +68,23 @@ with input_container:
             "Enter your query:",
             value=st.session_state["user_input"],  # Use session state for persistence
             key="chat_input",
-            label_visibility="visible",
-            on_change=lambda: st.session_state.update({"user_input": st.session_state.chat_input}),
         )
-        if st.button("Send", key="send_button"):
-            if st.session_state["user_input"]:
-                with st.spinner("Generating response..."):
-                    try:
-                        response = st.session_state["qa_pipeline"](f"Q: {st.session_state['user_input']}", max_length=400)
-                        generated_text = response[0]["generated_text"]
-                        st.session_state["conversation"].append(("You", st.session_state["user_input"]))
-                        st.session_state["conversation"].append(("Model", generated_text))
-                        st.session_state["user_input"] = ""  # Clear input after submission
-                    except Exception as e:
-                        st.error(f"Error generating response: {e}")
+        send_button_clicked = st.button("Send", key="send_button")
+        if send_button_clicked and user_input.strip():
+            # Process the 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"] = ""
+                except Exception as e:
+                    st.error(f"Error generating response: {e}")
 
 # Clear Conversation
 if clear_conversation_button: