Update app.py

app.py

@@ -336,10 +336,6 @@ def clear_user_input():
 def main():
     load_environment()
 
-    if "last_answered_question" not in st.session_state:
-        st.session_state.last_answered_question = ""
-
-    # Initialize session state safely
     if "chat_ready" not in st.session_state:
         st.session_state.chat_ready = False
     if "chat_history" not in st.session_state:
@@ -350,6 +346,8 @@ def main():
         st.session_state.feedback_log = []
     if "feedback_submitted" not in st.session_state:
         st.session_state.feedback_submitted = False
+    if "last_answered_question" not in st.session_state:
+        st.session_state.last_answered_question = ""
 
     st.header("Chat with MODTRAN Documents 📄")
 
@@ -360,17 +358,16 @@ def main():
             st.session_state.chat_ready = True
             st.success("MODTRAN User Manual loaded successfully!")
 
-    # Show entire conversation history
+    # Display entire chat history
     for i, exchange in enumerate(st.session_state.chat_history):
         st.markdown(f"**You:** {exchange['user']}")
         st.markdown(f"**MODTRAN Bot:** {exchange['bot']}")
 
-        # If rated, show rating
         if "rating" in exchange:
             st.markdown(f"⭐️ You rated this: {exchange['rating']}/5")
 
-        # If last answer not yet rated, show rating form
         elif i == len(st.session_state.chat_history) - 1:
+            # Show rating form for latest if not rated
             with st.form(key=f"feedback_form_{i}"):
                 rating = st.radio(
                     "Rate this response:",
@@ -380,7 +377,6 @@ def main():
                 )
                 submitted = st.form_submit_button("Submit Rating")
                 if submitted:
-                    # Save rating
                     st.session_state.chat_history[i]["rating"] = rating
                     st.session_state.feedback_log.append({
                         "question": exchange["user"],
@@ -391,7 +387,6 @@ def main():
 
                     print(f"📌 Feedbacks collected so far: {len(st.session_state.feedback_log)}")
 
-                    # Upload feedbacks when 5 are collected
                     if len(st.session_state.feedback_log) >= 5:
                         print("📦 Upload threshold reached — saving feedback to Hugging Face.")
                         save_feedback_to_huggingface()
@@ -400,12 +395,13 @@ def main():
                     st.session_state.feedback_submitted = True
                     st.rerun()
 
-    # Show how many feedbacks collected
+    # Feedback counter
     st.markdown(f"📝 Feedbacks collected: **{len(st.session_state.feedback_log)} / 5**")
 
-    # Show input box for next question
-    user_question = st.text_input("Ask your next question:", key="user_input", on_change=clear_user_input)
+    # Input box for new question
+    user_question = st.text_input("Ask your next question:", key="user_input")
 
+    # Only handle if user typed something new
     if user_question and user_question != st.session_state.last_answered_question:
         with st.spinner("Generating answer..."):
             try:
@@ -414,17 +410,21 @@ def main():
             except Exception as e:
                 response = f"⚠️ Something went wrong: {e}"
 
-        # Store answer
+        # Append to chat history
         st.session_state.chat_history.append({
             "user": user_question,
             "bot": response
         })
 
-        # ✅ Mark this question as answered
+        # Save the last answered question
         st.session_state.last_answered_question = user_question
 
+        # ✅ Clear the user input manually now
+        st.session_state.user_input = ""
+
         st.rerun()
 
+
 if __name__ == "__main__":
     load_environment()
     main()