Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 API_TOKEN = os.getenv("HF_TOKEN")
 client = InferenceClient(token=API_TOKEN)
 
-# Initialize session state for chat history
+# Initialize session state
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 if "corrected_sentence" not in st.session_state:
@@ -43,14 +43,22 @@ if st.session_state.corrected_sentence:
     for speaker, message in st.session_state.chat_history:
         st.write(f"**{speaker}:** {message}")
 
-    # Chat input
-    chat_input = st.text_input("Ask something about the corrected sentence:", key="chat_input")
-    if st.button("Send"):
-        if chat_input:
+    # Chat input with Enter key submission
+    chat_input = st.text_input(
+        "Ask something about the corrected sentence (press Enter to send):",
+        key="chat_input",
+        value="",  # Clear after each submission
+        on_change=lambda: submit_chat(),  # Trigger on Enter
+    )
+
+    # Function to handle chat submission
+    def submit_chat():
+        chat_text = st.session_state.chat_input
+        if chat_text:
             # Build prompt with corrected sentence as context
             prompt = (
                 f"The corrected sentence is: '{st.session_state.corrected_sentence}'. "
-                f"User asks: '{chat_input}'. Respond naturally."
+                f"User asks: '{chat_text}'. Respond naturally."
             )
             try:
                 response = client.text_generation(
@@ -60,13 +68,12 @@ if st.session_state.corrected_sentence:
                     temperature=0.7,
                 )
                 # Add to chat history
-                st.session_state.chat_history.append(("You", chat_input))
+                st.session_state.chat_history.append(("You", chat_text))
                 st.session_state.chat_history.append(("LLM", response.strip()))
-                # Rerun to refresh the display
-                st.rerun()
+                # Clear the input field by resetting the key
+                st.session_state.chat_input = ""
             except Exception as e:
                 st.error(f"Error in chat: {str(e)}")
-        else:
-            st.warning("Please enter a question!")
+
 else:
     st.write("Improve a sentence first to start chatting!")