Updated app.py to version 7

app.py

@@ -75,7 +75,8 @@ with tab1:
             st.subheader("❓ Ask AI About Your Data")
             question = st.text_input(
                 "What would you like to know about this data?",
-                placeholder="e.g., What is the average value in column X?"
+                placeholder="e.g., What is the average value in column X?",
+                key="data_question"
             )
             
             if question:
@@ -123,39 +124,40 @@ with tab2:
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
     
-    # Chat input
-    user_input = st.chat_input("Type your message here...")
+    # Chat input - MUST be outside tabs, so we use text_input instead
+    user_input = st.text_input(
+        "Type your message:",
+        placeholder="Ask me anything...",
+        key="chat_input"
+    )
     
     if user_input:
         # Add user message to history
         st.session_state.messages.append({"role": "user", "content": user_input})
         
-        # Display user message
-        with st.chat_message("user"):
-            st.markdown(user_input)
-        
         # Generate AI response
-        with st.chat_message("assistant"):
-            with st.spinner("⏳ Generating response..."):
-                try:
-                    prompt = f"User: {user_input}\n\nAssistant:"
-                    
-                    response = client.text_generation(
-                        prompt,
-                        max_new_tokens=300,
-                        temperature=0.7,
-                    )
-                    
-                    assistant_message = response.strip()
-                    st.markdown(assistant_message)
-                    
-                    # Add assistant message to history
-                    st.session_state.messages.append({
-                        "role": "assistant",
-                        "content": assistant_message
-                    })
-                except Exception as e:
-                    st.error(f"Error generating response: {e}")
+        with st.spinner("⏳ Generating response..."):
+            try:
+                prompt = f"User: {user_input}\n\nAssistant:"
+                
+                response = client.text_generation(
+                    prompt,
+                    max_new_tokens=300,
+                    temperature=0.7,
+                )
+                
+                assistant_message = response.strip()
+                
+                # Add assistant message to history
+                st.session_state.messages.append({
+                    "role": "assistant",
+                    "content": assistant_message
+                })
+                
+                # Rerun to display the new messages
+                st.rerun()
+            except Exception as e:
+                st.error(f"Error generating response: {e}")
 
 # ============================================================================
 # TAB 3: About