Update app.py

app.py

@@ -1012,6 +1012,10 @@ class HuggingFaceChatBot:
 # FIXED MAIN APPLICATION WITH PROPER CHAT INPUT PLACEMENT
 # ===============================================================================
 
+# ===============================================================================
+# FIXED APPLICATION WITH UNIQUE WIDGET KEYS
+# ===============================================================================
+
 def create_huggingface_app():
     """Main Streamlit application optimized for Hugging Face Spaces"""
     
@@ -1121,7 +1125,7 @@ def create_huggingface_app():
     ])
     
     # -------------------------------------------------------------------------
-    # TAB 1: UPLOAD & PROCESS
+    # TAB 1: UPLOAD & PROCESS - FIXED WITH UNIQUE KEY
     # -------------------------------------------------------------------------
     
     with tab1:
@@ -1154,14 +1158,15 @@ def create_huggingface_app():
             </div>
             """, unsafe_allow_html=True)
         
-        # File upload interface
+        # File upload interface with UNIQUE KEY
         st.markdown("### 📁 Upload Your Invoices")
         
         uploaded_files = st.file_uploader(
             "Choose invoice files (PDF, TXT supported)",
             type=['pdf', 'txt'],
             accept_multiple_files=True,
-            help=f"Maximum file size: {HF_CONFIG['max_file_size_mb']}MB per file"
+            help=f"Maximum file size: {HF_CONFIG['max_file_size_mb']}MB per file",
+            key="main_file_uploader"  # UNIQUE KEY ADDED
         )
         
         if uploaded_files:
@@ -1172,7 +1177,7 @@ def create_huggingface_app():
             
             st.info(f"📊 {len(uploaded_files)} files selected")
             
-            if st.button("🚀 Process Files", type="primary", use_container_width=True):
+            if st.button("🚀 Process Files", type="primary", use_container_width=True, key="process_files_btn"):
                 progress_bar = st.progress(0)
                 status_container = st.container()
                 results_container = st.container()
@@ -1192,7 +1197,7 @@ def create_huggingface_app():
                     with results_container:
                         if result.invoice_number:
                             successful += 1
-                            with st.expander(f"✅ {uploaded_file.name}", expanded=False):
+                            with st.expander(f"✅ {uploaded_file.name}", expanded=False, key=f"result_expander_{i}"):
                                 col1, col2 = st.columns(2)
                                 with col1:
                                     st.write(f"**Invoice #:** {result.invoice_number}")
@@ -1214,7 +1219,7 @@ def create_huggingface_app():
                     st.balloons()
     
     # -------------------------------------------------------------------------
-    # TAB 2: AI CHAT - FIXED LAYOUT
+    # TAB 2: AI CHAT - FIXED LAYOUT WITH UNIQUE KEYS
     # -------------------------------------------------------------------------
     
     with tab2:
@@ -1225,7 +1230,7 @@ def create_huggingface_app():
             st.markdown("### 💬 Chat History")
             chat_container = st.container()
             with chat_container:
-                for message in st.session_state.chat_history:
+                for i, message in enumerate(st.session_state.chat_history):
                     with st.chat_message(message["role"]):
                         st.markdown(message["content"])
         
@@ -1244,7 +1249,7 @@ def create_huggingface_app():
                     "Find invoices with high amounts"
                 ]
                 for i, query in enumerate(queries):
-                    if st.button(query, key=f"query_{i}", use_container_width=True):
+                    if st.button(query, key=f"basic_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()})
@@ -1260,7 +1265,7 @@ def create_huggingface_app():
                         "Find maintenance contracts"
                     ]
                     for i, query in enumerate(semantic_queries):
-                        if st.button(query, key=f"semantic_{i}", use_container_width=True):
+                        if st.button(query, key=f"semantic_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()})
@@ -1277,11 +1282,11 @@ def create_huggingface_app():
             user_input = st.text_input(
                 "Type your question here:",
                 placeholder="e.g., 'show me total spending' or 'find technology purchases'",
-                key="chat_text_input"
+                key="chat_text_input_tab"  # UNIQUE KEY ADDED
             )
         
         with col2:
-            ask_button = st.button("🚀 Ask", type="primary", use_container_width=True)
+            ask_button = st.button("🚀 Ask", type="primary", use_container_width=True, key="ask_button_tab")
         
         # Process the input
         if ask_button and user_input:
@@ -1307,12 +1312,12 @@ def create_huggingface_app():
         
         # Clear chat button
         if st.session_state.chat_history:
-            if st.button("🗑️ Clear Chat History", use_container_width=True):
+            if st.button("🗑️ Clear Chat History", use_container_width=True, key="clear_chat_btn"):
                 st.session_state.chat_history = []
                 st.rerun()
     
     # -------------------------------------------------------------------------
-    # TAB 3: ANALYTICS
+    # TAB 3: ANALYTICS - WITH UNIQUE KEYS
     # -------------------------------------------------------------------------
     
     with tab3:
@@ -1360,7 +1365,7 @@ def create_huggingface_app():
                     title="Invoice Amount Distribution",
                     labels={'amount': 'Amount (₹)', 'count': 'Number of Invoices'}
                 )
-                st.plotly_chart(fig_hist, use_container_width=True)
+                st.plotly_chart(fig_hist, use_container_width=True, key="amount_histogram")
                 
                 # Top suppliers
                 if df['supplier_name'].notna().any():
@@ -1373,7 +1378,7 @@ def create_huggingface_app():
                         title="Top 10 Suppliers by Total Amount",
                         labels={'x': 'Total Amount (₹)', 'y': 'Supplier'}
                     )
-                    st.plotly_chart(fig_suppliers, use_container_width=True)
+                    st.plotly_chart(fig_suppliers, use_container_width=True, key="suppliers_chart")
                 
                 # Confidence analysis
                 fig_confidence = px.histogram(
@@ -1382,13 +1387,13 @@ def create_huggingface_app():
                     title="Extraction Confidence Distribution",
                     labels={'confidence': 'Confidence Score', 'count': 'Number of Invoices'}
                 )
-                st.plotly_chart(fig_confidence, use_container_width=True)
+                st.plotly_chart(fig_confidence, use_container_width=True, key="confidence_chart")
         
         except Exception as e:
             st.error(f"Analytics error: {e}")
     
     # -------------------------------------------------------------------------
-    # TAB 4: DATA EXPLORER
+    # TAB 4: DATA EXPLORER - WITH UNIQUE KEYS
     # -------------------------------------------------------------------------
     
     with tab4:
@@ -1419,19 +1424,19 @@ def create_huggingface_app():
             
             df = pd.DataFrame(df_data)
             
-            # Filters
+            # Filters with unique keys
             col1, col2, col3 = st.columns(3)
             
             with col1:
                 suppliers = ['All'] + sorted(df['Supplier'].dropna().unique().tolist())
-                selected_supplier = st.selectbox("Filter by Supplier", suppliers)
+                selected_supplier = st.selectbox("Filter by Supplier", suppliers, key="supplier_filter")
             
             with col2:
                 methods = ['All'] + sorted(df['Method'].dropna().unique().tolist())
-                selected_method = st.selectbox("Filter by Method", methods)
+                selected_method = st.selectbox("Filter by Method", methods, key="method_filter")
             
             with col3:
-                min_amount = st.number_input("Min Amount", min_value=0.0, value=0.0)
+                min_amount = st.number_input("Min Amount", min_value=0.0, value=0.0, key="amount_filter")
             
             # Apply filters
             filtered_df = df.copy()
@@ -1449,24 +1454,26 @@ def create_huggingface_app():
                 column_config={
                     "Amount": st.column_config.NumberColumn("Amount", format="₹%.2f"),
                     "Confidence": st.column_config.ProgressColumn("Confidence", min_value=0, max_value=1)
-                }
+                },
+                key="data_explorer_table"  # UNIQUE KEY ADDED
             )
             
-            # Export options
+            # Export options with unique keys
             col1, col2 = st.columns(2)
             
             with col1:
-                if st.button("📥 Export CSV", use_container_width=True):
+                if st.button("📥 Export CSV", use_container_width=True, key="export_csv_btn"):
                     csv_data = filtered_df.to_csv(index=False)
                     st.download_button(
                         "Download CSV",
                         csv_data,
                         f"invoices_{datetime.now().strftime('%Y%m%d_%H%M')}.csv",
-                        "text/csv"
+                        "text/csv",
+                        key="download_csv_btn"
                     )
             
             with col2:
-                if st.button("📄 Export JSON", use_container_width=True):
+                if st.button("📄 Export JSON", use_container_width=True, key="export_json_btn"):
                     filtered_invoices = [inv for inv in invoices 
                                        if inv.get('invoice_number') in filtered_df['Invoice Number'].values]
                     
@@ -1480,14 +1487,15 @@ def create_huggingface_app():
                         "Download JSON",
                         json.dumps(export_data, indent=2),
                         f"invoices_{datetime.now().strftime('%Y%m%d_%H%M')}.json",
-                        "application/json"
+                        "application/json",
+                        key="download_json_btn"
                     )
         
         except Exception as e:
             st.error(f"Data explorer error: {e}")
     
     # -------------------------------------------------------------------------
-    # GLOBAL CHAT INPUT - OUTSIDE TABS
+    # GLOBAL CHAT INPUT - OUTSIDE TABS WITH UNIQUE KEY
     # -------------------------------------------------------------------------
     
     # Add some spacing
@@ -1496,8 +1504,8 @@ def create_huggingface_app():
     # 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...")
+    # Use chat_input outside of tabs with unique key - this will work
+    global_user_query = st.chat_input("Ask about your invoices from anywhere...", key="global_chat_input")
     
     if global_user_query:
         # Add user message
@@ -1537,6 +1545,27 @@ def create_huggingface_app():
 # 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()
+
+# ===============================================================================
+# MAIN APPLICATION ENTRY POINT
+# ===============================================================================
+
 def main():
     """Main entry point for Hugging Face Spaces"""
     try: