Update app.py

app.py

@@ -340,6 +340,11 @@ def clear_user_input():
 
 from datetime import datetime
 
+def clear_user_input():
+    st.session_state.user_input = ""
+
+from datetime import datetime
+
 def clear_user_input():
     st.session_state.user_input = ""
 
@@ -360,23 +365,15 @@ def main():
     if "last_answered_question" not in st.session_state:
         st.session_state.last_answered_question = ""
 
-    st.header("Chat with MODTRAN Documents 📄")
+    st.header("Chat with MODTRAN Document 📄")
 
-    # Inject CSS for chat-style alignment
+    # Inject custom CSS for chat bubbles
     st.markdown("""
         <style>
             .chat-container {
                 display: flex;
                 flex-direction: column;
-                gap: 0.5rem;
-            }
-            .user-bubble {
-                align-self: flex-end;
-                background-color: rgba(30, 30, 30, 0.85);
-                color: white;
-                padding: 1rem;
-                border-radius: 1rem 0.5rem 1rem 1rem;
-                max-width: 80%;
+                gap: 0.75rem;
             }
             .bot-bubble {
                 align-self: flex-start;
@@ -386,6 +383,14 @@ def main():
                 border-radius: 0.5rem 1rem 1rem 1rem;
                 max-width: 80%;
             }
+            .user-bubble {
+                align-self: flex-end;
+                background-color: rgba(0, 0, 0, 0.75);
+                color: white;
+                padding: 1rem;
+                border-radius: 1rem 0.5rem 1rem 1rem;
+                max-width: 80%;
+            }
             .rating-line {
                 font-weight: bold;
                 color: #ffaa00;
@@ -402,7 +407,7 @@ def main():
         </style>
     """, unsafe_allow_html=True)
 
-    # Load chatbot agent and vectorstore if not ready
+    # Load vectorstore and chatbot agent
     if not st.session_state.chat_ready:
         with st.spinner("Loading MODTRAN document..."):
             preload_modtran_document()
@@ -410,18 +415,20 @@ def main():
             st.session_state.chat_ready = True
             st.success("MODTRAN User Manual loaded successfully!")
 
-    # 🟦 Chat History Output
+    # Render all previous Q&A in chat format
     st.markdown('<div class="chat-container">', unsafe_allow_html=True)
 
     for i, exchange in enumerate(st.session_state.chat_history):
-        st.markdown(f'<div class="user-bubble"><strong>You:</strong> {exchange["user"]}</div>', unsafe_allow_html=True)
+        # MODTRAN Bot's answer (left side)
         st.markdown(f'<div class="bot-bubble"><strong>MODTRAN Bot:</strong> {exchange["bot"]}</div>', unsafe_allow_html=True)
+        # User's question (right side)
+        st.markdown(f'<div class="user-bubble"><strong>You:</strong> {exchange["user"]}</div>', unsafe_allow_html=True)
 
-        # Show rating if available
+        # If already rated
         if "rating" in exchange:
             st.markdown(f'<div class="rating-line">⭐️ You rated this: {exchange["rating"]}/5</div>', unsafe_allow_html=True)
 
-        # Show form for last unanswered item
+        # If not rated yet and it's the last message, show form
         elif i == len(st.session_state.chat_history) - 1:
             with st.form(key=f"feedback_form_{i}"):
                 rating = st.radio(
@@ -440,9 +447,7 @@ def main():
                         "timestamp": datetime.now().isoformat()
                     })
 
-                    print(f"📌 Feedbacks collected so far: {len(st.session_state.feedback_log)}")
-
-                    if len(st.session_state.feedback_log) >= 5:
+                    if len(st.session_state.feedback_log) >= 2:
                         print("📦 Upload threshold reached — saving feedback to Hugging Face.")
                         save_feedback_to_huggingface()
 
@@ -452,10 +457,10 @@ def main():
 
     st.markdown('</div>', unsafe_allow_html=True)
 
-    # Feedback status
-    st.markdown(f'<div class="feedback-counter">📄 Feedbacks collected: <strong>{len(st.session_state.feedback_log)} / 5</strong></div>', unsafe_allow_html=True)
-
+    # Show current feedback progress
+    #st.markdown(f'<div class="feedback-counter">📄 Feedbacks collected: <strong>{len(st.session_state.feedback_log)} / 5</strong></div>', unsafe_allow_html=True)
 
+    # Input for next question
     user_question = st.text_input("Ask your next question:", key="user_input")
 
     if user_question and user_question != st.session_state.last_answered_question:
@@ -466,6 +471,7 @@ def main():
             except Exception as e:
                 response = f"⚠️ Something went wrong: {e}"
 
+        # Save new Q&A
         st.session_state.chat_history.append({
             "user": user_question,
             "bot": response
@@ -474,6 +480,7 @@ def main():
         st.session_state.last_answered_question = user_question
         st.rerun()
 
+
 if __name__ == "__main__":
     load_environment()
     main()