Update app.py

app.py

@@ -298,7 +298,8 @@ def handle_user_query(query):
 def save_feedback_to_huggingface():
     try:
         if not st.session_state.feedback_log:
-            return  # No feedback yet
+            print("⚠️ No feedbacks collected yet.")
+            return
 
         feedback_df = pd.DataFrame(st.session_state.feedback_log)
         now = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
@@ -310,7 +311,10 @@ def save_feedback_to_huggingface():
 
             token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACEHUB_API_TOKEN")
             if not token:
-                raise ValueError("HF token not found.")
+                raise ValueError("❌ Hugging Face token not found!")
+
+            print(f"📤 Attempting upload to repo: ZarinT/chatbot-feedback as {filename}")
+            print("📝 Feedback data:", feedback_df)
 
             api = HfApi(token=token)
             api.upload_file(
@@ -320,11 +324,12 @@ def save_feedback_to_huggingface():
                 repo_type="dataset"
             )
 
+        print("✅ Feedback uploaded successfully.")
         st.session_state.feedback_log.clear()
-        print("Feedback uploaded successfully.")
 
     except Exception as e:
-        print("Error uploading feedback:", e)
+        print("❌ Feedback upload failed:", e)
+
     
 def main():
     load_environment()
@@ -349,56 +354,51 @@ def main():
             st.session_state.chat_ready = True
             st.success("MODTRAN User Manual loaded successfully!")
 
-    user_question = st.text_input("Ask your question:", key="user_input")
-
-    if (
-        st.session_state.chat_ready and
-        user_question and
-        not st.session_state.feedback_submitted 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  # ✅ Mark as answered
-
-        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:
-                    save_feedback_to_huggingface()
-
-                
-                st.session_state.feedback_submitted = True
-                st.session_state.user_input = ""  
-                st.rerun()
-
-    # Reset flag if new question is typed
-    if st.session_state.feedback_submitted and not user_question:
+    # 📝 Show different view based on whether feedback was submitted
+    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.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)
+                    save_feedback_to_huggingface()
+                    
+                    # After feedback submit:
+                    st.session_state.feedback_submitted = True
+                    st.rerun()
+