Update app.py

app.py

@@ -1008,6 +1008,10 @@ class HuggingFaceChatBot:
 # STREAMLIT APPLICATION FOR HUGGING FACE
 # ===============================================================================
 
+# ===============================================================================
+# FIXED MAIN APPLICATION WITH PROPER CHAT INPUT PLACEMENT
+# ===============================================================================
+
 def create_huggingface_app():
     """Main Streamlit application optimized for Hugging Face Spaces"""
     
@@ -1032,6 +1036,14 @@ def create_huggingface_app():
     .status-ok { color: #28a745; font-weight: bold; }
     .status-warning { color: #ffc107; font-weight: bold; }
     .status-error { color: #dc3545; font-weight: bold; }
+    .chat-container {
+        border: 1px solid #ddd;
+        border-radius: 10px;
+        padding: 1rem;
+        margin: 1rem 0;
+        max-height: 400px;
+        overflow-y: auto;
+    }
     </style>
     """, unsafe_allow_html=True)
     
@@ -1202,45 +1214,29 @@ def create_huggingface_app():
                     st.balloons()
     
     # -------------------------------------------------------------------------
-    # TAB 2: AI CHAT
+    # TAB 2: AI CHAT - FIXED LAYOUT
     # -------------------------------------------------------------------------
     
     with tab2:
         st.header("💬 AI Chat Interface")
         
-        # Chat interface
-        user_query = st.chat_input("Ask about your invoices... (e.g., 'show me total spending')")
-        
-        if user_query:
-            # Add user message
-            st.session_state.chat_history.append({
-                "role": "user", 
-                "content": user_query,
-                "timestamp": datetime.now()
-            })
-            
-            # Get AI response
-            with st.spinner("🤖 AI is analyzing..."):
-                response = st.session_state.hf_chatbot.query_database(user_query)
-                
-                st.session_state.chat_history.append({
-                    "role": "assistant",
-                    "content": response,
-                    "timestamp": datetime.now()
-                })
-        
-        # Display chat history
-        for message in st.session_state.chat_history:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
+        # Display chat history first
+        if st.session_state.chat_history:
+            st.markdown("### 💬 Chat History")
+            chat_container = st.container()
+            with chat_container:
+                for message in st.session_state.chat_history:
+                    with st.chat_message(message["role"]):
+                        st.markdown(message["content"])
         
-        # Suggested queries
+        # Suggested queries when no chat history
         if not st.session_state.chat_history:
             st.markdown("### 💡 Try These Queries")
             
             col1, col2 = st.columns(2)
             
             with col1:
+                st.markdown("**📊 Basic Queries:**")
                 queries = [
                     "Show me a summary of all invoices",
                     "How much have we spent in total?",
@@ -1248,13 +1244,14 @@ def create_huggingface_app():
                     "Find invoices with high amounts"
                 ]
                 for i, query in enumerate(queries):
-                    if st.button(query, key=f"query_{i}"):
+                    if st.button(query, key=f"query_{i}", use_container_width=True):
                         st.session_state.chat_history.append({"role": "user", "content": query, "timestamp": datetime.now()})
                         response = st.session_state.hf_chatbot.query_database(query)
                         st.session_state.chat_history.append({"role": "assistant", "content": response, "timestamp": datetime.now()})
                         st.rerun()
             
             with col2:
+                st.markdown("**🔍 Semantic Queries:**")
                 if st.session_state.hf_processor.vector_store:
                     semantic_queries = [
                         "Find technology equipment purchases",
@@ -1263,11 +1260,56 @@ def create_huggingface_app():
                         "Find maintenance contracts"
                     ]
                     for i, query in enumerate(semantic_queries):
-                        if st.button(query, key=f"semantic_{i}"):
+                        if st.button(query, key=f"semantic_{i}", use_container_width=True):
                             st.session_state.chat_history.append({"role": "user", "content": query, "timestamp": datetime.now()})
                             response = st.session_state.hf_chatbot.query_database(query)
                             st.session_state.chat_history.append({"role": "assistant", "content": response, "timestamp": datetime.now()})
                             st.rerun()
+                else:
+                    st.info("Semantic search not available. Upload some invoices first!")
+        
+        # Alternative input method using text input and button
+        st.markdown("### ✍️ Ask a Question")
+        
+        col1, col2 = st.columns([4, 1])
+        
+        with col1:
+            user_input = st.text_input(
+                "Type your question here:",
+                placeholder="e.g., 'show me total spending' or 'find technology purchases'",
+                key="chat_text_input"
+            )
+        
+        with col2:
+            ask_button = st.button("🚀 Ask", type="primary", use_container_width=True)
+        
+        # Process the input
+        if ask_button and user_input:
+            # Add user message
+            st.session_state.chat_history.append({
+                "role": "user", 
+                "content": user_input,
+                "timestamp": datetime.now()
+            })
+            
+            # Get AI response
+            with st.spinner("🤖 AI is analyzing..."):
+                response = st.session_state.hf_chatbot.query_database(user_input)
+                
+                st.session_state.chat_history.append({
+                    "role": "assistant",
+                    "content": response,
+                    "timestamp": datetime.now()
+                })
+            
+            # Clear the input and rerun to show new messages
+            st.rerun()
+        
+        # Clear chat button
+        if st.session_state.chat_history:
+            if st.button("🗑️ Clear Chat History", use_container_width=True):
+                st.session_state.chat_history = []
+                st.rerun()
     
     # -------------------------------------------------------------------------
     # TAB 3: ANALYTICS
@@ -1444,6 +1486,44 @@ def create_huggingface_app():
         except Exception as e:
             st.error(f"Data explorer error: {e}")
     
+    # -------------------------------------------------------------------------
+    # GLOBAL CHAT INPUT - OUTSIDE TABS
+    # -------------------------------------------------------------------------
+    
+    # Add some spacing
+    st.markdown("---")
+    
+    # Global chat input that works from any tab
+    st.markdown("### 💬 Quick Chat (Available from any tab)")
+    
+    # Use chat_input outside of tabs - this will work
+    global_user_query = st.chat_input("Ask about your invoices from anywhere...")
+    
+    if global_user_query:
+        # Add user message
+        st.session_state.chat_history.append({
+            "role": "user", 
+            "content": global_user_query,
+            "timestamp": datetime.now()
+        })
+        
+        # Get AI response
+        with st.spinner("🤖 AI is analyzing..."):
+            response = st.session_state.hf_chatbot.query_database(global_user_query)
+            
+            st.session_state.chat_history.append({
+                "role": "assistant",
+                "content": response,
+                "timestamp": datetime.now()
+            })
+        
+        # Show the response immediately
+        with st.chat_message("assistant"):
+            st.markdown(response)
+        
+        # Suggest switching to chat tab if not already there
+        st.info("💡 Switch to the 'AI Chat' tab to see full conversation history!")
+    
     # Footer
     st.markdown("---")
     st.markdown("""
@@ -1453,6 +1533,27 @@ def create_huggingface_app():
     </div>
     """, unsafe_allow_html=True)
 
+# ===============================================================================
+# MAIN APPLICATION ENTRY POINT
+# ===============================================================================
+
+def main():
+    """Main entry point for Hugging Face Spaces"""
+    try:
+        # Display Hugging Face info if running on HF Spaces
+        if IS_HF_SPACE:
+            st.sidebar.info("🤗 Running on Hugging Face Spaces")
+        
+        # Create and run the app
+        create_huggingface_app()
+        
+    except Exception as e:
+        st.error(f"Application error: {e}")
+        st.info("Please refresh the page or contact support if the error persists.")
+
+if __name__ == "__main__":
+    main()
+
 # ===============================================================================
 # HUGGING FACE REQUIREMENTS AND CONFIGURATION
 # ===============================================================================