Update app.py

app.py

@@ -333,9 +333,15 @@ def save_feedback_to_huggingface():
 def clear_user_input():
     st.session_state["user_input"] = ""
     
+from datetime import datetime
+
+def clear_user_input():
+    st.session_state.user_input = ""
+
 def main():
     load_environment()
 
+    # Initialize session state
     if "chat_ready" not in st.session_state:
         st.session_state.chat_ready = False
     if "chat_history" not in st.session_state:
@@ -351,6 +357,39 @@ def main():
 
     st.header("Chat with MODTRAN Documents 📄")
 
+    # Add custom CSS for chat bubbles
+    st.markdown("""
+        <style>
+            .user-bubble {
+                background-color: #f0f0f0;
+                padding: 1rem;
+                margin: 0.5rem 0;
+                border-radius: 1rem;
+                color: #333;
+            }
+            .bot-bubble {
+                background-color: #e8f4fd;
+                padding: 1rem;
+                margin: 0.5rem 0;
+                border-radius: 1rem;
+                color: #111;
+            }
+            .rating-line {
+                color: #ffaa00;
+                font-weight: bold;
+                margin-top: -0.5rem;
+                margin-bottom: 1rem;
+            }
+            .feedback-counter {
+                color: #444;
+                font-size: 0.9rem;
+                margin-top: 1rem;
+                font-style: italic;
+            }
+        </style>
+    """, unsafe_allow_html=True)
+
+    # Preload documents and agent
     if not st.session_state.chat_ready:
         with st.spinner("Loading MODTRAN document..."):
             preload_modtran_document()
@@ -360,14 +399,14 @@ def main():
 
     # 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']}")
+        st.markdown(f'<div class="user-bubble"><strong>You:</strong> {exchange["user"]}</div>', unsafe_allow_html=True)
+        st.markdown(f'<div class="bot-bubble"><strong>MODTRAN Bot:</strong> {exchange["bot"]}</div>', unsafe_allow_html=True)
 
         if "rating" in exchange:
-            st.markdown(f"⭐️ You rated this: {exchange['rating']}/5")
+            st.markdown(f'<div class="rating-line">⭐️ You rated this: {exchange["rating"]}/5</div>', unsafe_allow_html=True)
 
         elif i == len(st.session_state.chat_history) - 1:
-            # Show rating form for latest if not rated
+            # Show feedback form for most recent response
             with st.form(key=f"feedback_form_{i}"):
                 rating = st.radio(
                     "Rate this response:",
@@ -395,13 +434,12 @@ def main():
                     st.session_state.feedback_submitted = True
                     st.rerun()
 
-    # Feedback counter
-    st.markdown(f"📝 Feedbacks collected: **{len(st.session_state.feedback_log)} / 5**")
+    # Show feedback counter
+    #st.markdown(f'<div class="feedback-counter">📄 Feedbacks collected: <strong>{len(st.session_state.feedback_log)} / 5</strong></div>', unsafe_allow_html=True)
 
-    # Input box for new question
+    # Ask next 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:
@@ -410,18 +448,18 @@ def main():
             except Exception as e:
                 response = f"⚠️ Something went wrong: {e}"
 
-        # Append to chat history
+        # Save chat
         st.session_state.chat_history.append({
             "user": user_question,
             "bot": response
         })
 
-        # Save the last answered question
         st.session_state.last_answered_question = user_question
-
+        st.session_state.user_input = ""
         st.rerun()
 
 
+
 if __name__ == "__main__":
     load_environment()
     main()