Update app.py

app.py

@@ -13,7 +13,7 @@ clear_conversation_button = st.sidebar.button("Clear Conversation")
 clear_model_button = st.sidebar.button("Clear Model")
 
 # Main UI
-st.title("Chat Conversation")
+st.title("Chat Conversation UI")
 
 # Session states
 if "model" not in st.session_state:
@@ -49,27 +49,34 @@ if clear_model_button:
     st.session_state["qa_pipeline"] = None
     st.success("Model cleared.")
 
-# Chat Input and Output
-if st.session_state["qa_pipeline"]:
-    user_input = st.text_input("Enter your query:", key="chat_input")
-    if st.button("Send"):
-        if user_input:
-            with st.spinner("Generating response..."):
-                try:
-                    response = st.session_state["qa_pipeline"](f"Q: {user_input}", max_length=300)
-                    generated_text = response[0]["generated_text"]
-                    st.session_state["conversation"].append(("You", user_input))
-                    st.session_state["conversation"].append(("Model", generated_text))
-                except Exception as e:
-                    st.error(f"Error generating response: {e}")
+# Layout for chat
+chat_container = st.container()
+input_container = st.container()
 
-    # Display conversation
+# Chat Conversation Display
+with chat_container:
+    st.subheader("Conversation")
     for idx, (speaker, message) in enumerate(st.session_state["conversation"]):
         if speaker == "You":
             st.text_area(f"You ({idx}):", message, key=f"you_{idx}", disabled=False)
         else:
             st.text_area(f"Model ({idx}):", message, key=f"model_{idx}", disabled=False)
 
+# Input Area
+with input_container:
+    if st.session_state["qa_pipeline"]:
+        user_input = st.text_input("Enter your query:", key="chat_input", label_visibility="visible")
+        if st.button("Send", key="send_button"):
+            if user_input:
+                with st.spinner("Generating response..."):
+                    try:
+                        response = st.session_state["qa_pipeline"](user_input, max_length=300)
+                        generated_text = response[0]["generated_text"]
+                        st.session_state["conversation"].append(("You", user_input))
+                        st.session_state["conversation"].append(("Model", generated_text))
+                    except Exception as e:
+                        st.error(f"Error generating response: {e}")
+
 # Clear Conversation
 if clear_conversation_button:
     st.session_state["conversation"] = []