Update app.py

app.py

@@ -163,6 +163,7 @@ def set_global_vectorstore(vectorstore):
     vectorstore_global = vectorstore
 
 kw_model = None
+reranker = None
 
 def get_kw_model():
     global kw_model
@@ -334,16 +335,17 @@ def save_feedback_to_huggingface():
 def main():
     load_environment()
 
-    if "chat_ready" not in st.session_state:
-        st.session_state.chat_ready = False
-    if "chat_history" not in st.session_state:
-        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
-    if "last_question" not in st.session_state:
-        st.session_state.last_question = ""
+    # Initialize session state
+    for key, default in {
+        "chat_ready": False,
+        "chat_history": [],
+        "vectorstore": None,
+        "feedback_submitted": False,
+        "last_question": "",
+        "feedback_log": []
+    }.items():
+        if key not in st.session_state:
+            st.session_state[key] = default
 
     st.header("Chat with MODTRAN Documents 📄")
 
@@ -354,50 +356,64 @@ def main():
             st.session_state.chat_ready = True
             st.success("MODTRAN User Manual loaded successfully!")
 
-    # 📝 Show different view based on whether feedback was submitted
+    # UI after feedback submission
     if st.session_state.feedback_submitted:
-        st.success("✅ Thank you for your feedback! Please ask another question.")
-        st.session_state.user_input = ""  # Clear input field
-        st.session_state.feedback_submitted = False
+        st.success("✅ Thank you for your feedback!")
+        st.info("You may now ask another question.")
+        st.markdown(f"📝 Feedbacks collected: **{len(st.session_state.feedback_log)} / 5**")
+        # Reset UI state for next question
+        st.session_state.user_input = ""
         st.session_state.last_question = ""
-    else:
-        user_question = st.text_input("Ask your question:", key="user_input")
-
-        if st.session_state.chat_ready and user_question and user_question != st.session_state.last_question:
-            with st.spinner("Generating answer..."):
-                try:
-                    set_global_vectorstore(st.session_state.vectorstore)
-                    response = handle_user_query(user_question)
-                except Exception as e:
-                    response = f"⚠️ Something went wrong: {e}"
-
-            st.markdown("### Answer:")
-            st.write(response)
-
-            st.session_state.chat_history.append({"user": user_question, "bot": response})
-            st.session_state.last_question = user_question
-
-            with st.form(key=f"feedback_form_{len(st.session_state.chat_history)}"):
-                rating = st.radio(
-                    "Rate this response:",
-                    options=["1", "2", "3", "4", "5"],
-                    key=f"rating_{len(st.session_state.chat_history)}",
-                    horizontal=True
-                )
-                submitted = st.form_submit_button("Submit Rating")
-                if submitted:
-                    feedback = {
-                        "question": user_question,
-                        "response": response,
-                        "rating": rating,
-                        "timestamp": datetime.datetime.now().isoformat()
-                    }
-                    st.session_state.feedback_log.append(feedback)
+        st.session_state.feedback_submitted = False
+
+    user_question = st.text_input("Ask your question:", key="user_input")
+
+    # Question handling logic
+    if (
+        st.session_state.chat_ready and
+        user_question and
+        user_question != st.session_state.last_question
+    ):
+        with st.spinner("Generating answer..."):
+            try:
+                set_global_vectorstore(st.session_state.vectorstore)
+                response = handle_user_query(user_question)
+            except Exception as e:
+                response = f"⚠️ Something went wrong: {e}"
+
+        # Show answer and feedback form
+        st.markdown("### Answer:")
+        st.write(response)
+
+        st.session_state.chat_history.append({"user": user_question, "bot": response})
+        st.session_state.last_question = user_question
+
+        st.markdown(f"📝 Feedbacks collected: **{len(st.session_state.feedback_log)} / 5**")
+
+        with st.form(key=f"feedback_form_{len(st.session_state.chat_history)}"):
+            rating = st.radio(
+                "Rate this response:",
+                options=["1", "2", "3", "4", "5"],
+                key=f"rating_{len(st.session_state.chat_history)}",
+                horizontal=True
+            )
+            submitted = st.form_submit_button("Submit Rating")
+            if submitted:
+                feedback = {
+                    "question": user_question,
+                    "response": response,
+                    "rating": rating,
+                    "timestamp": datetime.datetime.now().isoformat()
+                }
+                st.session_state.feedback_log.append(feedback)
+
+                if len(st.session_state.feedback_log) >= 1:
+                    print("📦 Upload threshold reached — saving feedback to Hugging Face.")
                     save_feedback_to_huggingface()
-                    
-                    # After feedback submit:
-                    st.session_state.feedback_submitted = True
-                    st.rerun()
+
+                st.session_state.feedback_submitted = True
+                st.rerun()
+