final UI

frontend/app.py

@@ -76,25 +76,13 @@ with st.sidebar:
 ask_tab, challenge_tab = st.tabs(["Ask Anything", "Challenge Me"])
 
 with ask_tab:
-    st.subheader("Document Chat")
-    if "chat_history" not in st.session_state:
-        st.session_state.chat_history = []
+    st.subheader("Document Q&A")
+    
     if st.session_state.current_file:
-        # Display chat history
-        for msg in st.session_state.chat_history:
-            if msg["role"] == "user":
-                st.markdown(f"**You:** {msg['content']}")
-            else:
-                st.markdown(f"**Assistant:** {msg['content']}")
-
-        # Chat input
-        user_question = st.text_input("Type your question and press Enter:", key="chat_input")
-        send = st.button("Send", key="send_chat")
-        if send and user_question.strip():
-            # Add user message to history
-            st.session_state.chat_history.append({"role": "user", "content": user_question})
-
-            # Send to backend
+        # Chat interface
+        user_question = st.text_input("Ask a question about the document:")
+        
+        if user_question:
             response = requests.post(
                 f"{BACKEND_URL}/chat",
                 json={
@@ -103,11 +91,13 @@ with ask_tab:
                     "model": "gpt-4o-mini"
                 }
             )
+            
             if response.status_code == 200:
                 data = response.json()
-                # Add assistant response to history
-                st.session_state.chat_history.append({"role": "assistant", "content": data["answer"]})
-                st.experimental_rerun()
+                st.divider()
+                st.subheader("Answer")
+                st.write(data["answer"])
+                st.caption(f"Session ID: {data['session_id']}")
             else:
                 st.error("Failed to get response")
     else: