Update app.py

app.py

@@ -335,10 +335,11 @@ def main():
         st.session_state.chat_history = []
     if "vectorstore" not in st.session_state:
         st.session_state.vectorstore = None
+    if "feedback_submitted" not in st.session_state:
+        st.session_state.feedback_submitted = False
 
     st.header("Chat with MODTRAN Documents 📄")
 
-    # Preload the document once when app starts
     if not st.session_state.chat_ready:
         with st.spinner("Loading MODTRAN document..."):
             preload_modtran_document()
@@ -348,7 +349,7 @@ def main():
 
     user_question = st.text_input("Ask your question:", key="user_input")
 
-    if st.session_state.chat_ready and user_question:
+    if st.session_state.chat_ready and user_question and not st.session_state.feedback_submitted:
         with st.spinner("Generating answer..."):
             try:
                 set_global_vectorstore(st.session_state.vectorstore)
@@ -356,14 +357,11 @@ def main():
             except Exception as e:
                 response = f"⚠️ Something went wrong: {e}"
 
-        # ✅ Show the response
         st.markdown("### Answer:")
         st.write(response)
 
-        # Update chat history
         st.session_state.chat_history.append({"user": user_question, "bot": response})
 
-        # After bot responds, ask for feedback
         with st.form(key=f"feedback_form_{len(st.session_state.chat_history)}"):
             rating = st.radio(
                 "Rate this response:",
@@ -381,12 +379,17 @@ def main():
                 }
                 st.session_state.feedback_log.append(feedback)
 
-                # If 5 feedbacks collected, auto upload to Hugging Face
                 if len(st.session_state.feedback_log) >= 5:
                     save_feedback_to_huggingface()
 
-        # Clear input box automatically after feedback
-        st.rerun()
+                # Mark feedback as submitted to prevent rerun loop
+                st.session_state.feedback_submitted = True
+                st.rerun()
+
+    # Reset feedback flag if no question is asked (new session)
+    if not user_question:
+        st.session_state.feedback_submitted = False
+
 
 
 if __name__ == "__main__":