COMPLETE REWRITE: Remove session state dependency for HF Spaces compatibility

app.py

@@ -14,12 +14,6 @@ st.set_page_config(
     layout="wide"
 )
 
-# Initialize session state FIRST - before any other code
-if 'processed_data' not in st.session_state:
-    st.session_state.processed_data = None
-if 'email_generator' not in st.session_state:
-    st.session_state.email_generator = None
-
 def init_database():
     """Initialize SQLite database for caching"""
     conn = sqlite3.connect('leads.db')
@@ -57,18 +51,15 @@ def init_database():
     conn.commit()
     conn.close()
 
+@st.cache_resource
 def load_email_generator():
     """Load the email generator with caching"""
-    if st.session_state.email_generator is None:
-        with st.spinner("🤖 Loading AI model (first time may take a few minutes)..."):
-            try:
-                st.session_state.email_generator = EmailGenerator()
-                st.success("✅ AI model loaded successfully!")
-            except Exception as e:
-                st.error(f"❌ Failed to load AI model: {str(e)}")
-                st.info("💡 The model will be downloaded automatically on first run. Please ensure you have a stable internet connection.")
-                return None
-    return st.session_state.email_generator
+    try:
+        return EmailGenerator()
+    except Exception as e:
+        st.error(f"❌ Failed to load AI model: {str(e)}")
+        st.info("💡 The model will be downloaded automatically on first run. Please ensure you have a stable internet connection.")
+        return None
 
 def process_csv_data(df, tone, creativity, num_variations):
     """Process CSV data and generate emails"""
@@ -186,25 +177,17 @@ def display_results(results, tone, creativity):
                 
                 # Subject line
                 st.write("**Subject:**")
-                subject_container = st.container()
-                with subject_container:
-                    st.code(best['subject_line'], language=None)
-                    if st.button(f"📋 Copy Subject", key=f"copy_subject_{idx}"):
-                        st.success("Subject copied to clipboard!")
+                st.code(best['subject_line'], language=None)
                 
                 # Email body
                 st.write("**Email Body:**")
-                email_container = st.container()
-                with email_container:
-                    st.text_area(
-                        "Email Content",
-                        best['email_body'],
-                        height=300,
-                        key=f"email_body_{idx}",
-                        label_visibility="collapsed"
-                    )
-                    if st.button(f"📋 Copy Email", key=f"copy_email_{idx}"):
-                        st.success("Email copied to clipboard!")
+                st.text_area(
+                    "Email Content",
+                    best['email_body'],
+                    height=300,
+                    key=f"email_body_{idx}",
+                    label_visibility="collapsed"
+                )
             
             with col3:
                 st.write("**🎯 Generation Settings:**")
@@ -221,10 +204,7 @@ def display_results(results, tone, creativity):
                     st.write(f"**🔄 Variations ({len(result['variations'])}):**")
                     for i, var in enumerate(result['variations']):
                         quality_color = "🌟" if var['quality_score'] >= 8 else "⚡" if var['quality_score'] >= 6 else "⚠️"
-                        if st.button(f"{quality_color} Variation {i+1} ({var['quality_score']:.1f})", key=f"var_{idx}_{i}"):
-                            # Show this variation
-                            st.info(f"**Subject:** {var['subject_line']}")
-                            st.text_area("Body", var['email_body'], height=200, key=f"var_body_{idx}_{i}")
+                        st.write(f"{quality_color} Variation {i+1}: {var['quality_score']:.1f}/10")
 
 def export_to_csv(results):
     """Export results to CSV format"""
@@ -292,10 +272,6 @@ def main():
         # Model info
         st.subheader("🤖 AI Model")
         st.info("**Vicuna-7B GGUF**\n\nOptimized for quality cold email generation")
-        
-        if st.button("🔄 Reload Model"):
-            st.session_state.email_generator = None
-            st.experimental_rerun()
     
     # File upload
     st.subheader("📁 Upload Your Lead List")
@@ -328,6 +304,9 @@ def main():
                 "text/csv"
             )
     
+    # Initialize results variable
+    results = None
+    
     if uploaded_file is not None:
         try:
             # Load and validate CSV
@@ -375,8 +354,28 @@ def main():
                     results = process_csv_data(df, tone, creativity, num_variations)
                 
                 if results:
-                    st.session_state.processed_data = results
                     st.success("✅ Email generation complete!")
+                    
+                    # Display results immediately
+                    st.markdown("---")
+                    display_results(results, tone, creativity)
+                    
+                    # Export functionality
+                    st.markdown("---")
+                    st.subheader("📤 Export Results")
+                    
+                    export_df = export_to_csv(results)
+                    if export_df is not None:
+                        csv = export_df.to_csv(index=False)
+                        st.download_button(
+                            "📥 Download Results as CSV",
+                            csv,
+                            f"cold_emails_export_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
+                            "text/csv",
+                            help="Download all generated emails in CSV format"
+                        )
+                        
+                        st.info(f"📊 Ready to export {len(export_df)} successful email generations")
                 else:
                     st.error("❌ Failed to process leads. Please try again.")
         
@@ -384,28 +383,6 @@ def main():
             st.error(f"❌ Error reading CSV file: {str(e)}")
             st.info("Please ensure your CSV file has the correct format and encoding.")
     
-    # Display results if available - This is where the error was happening
-    if st.session_state.processed_data is not None:
-        st.markdown("---")
-        display_results(st.session_state.processed_data, tone, creativity)
-        
-        # Export functionality
-        st.markdown("---")
-        st.subheader("📤 Export Results")
-        
-        export_df = export_to_csv(st.session_state.processed_data)
-        if export_df is not None:
-            csv = export_df.to_csv(index=False)
-            st.download_button(
-                "📥 Download Results as CSV",
-                csv,
-                f"cold_emails_export_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
-                "text/csv",
-                help="Download all generated emails in CSV format"
-            )
-            
-            st.info(f"📊 Ready to export {len(export_df)} successful email generations")
-    
     # Footer
     st.markdown("---")
     st.markdown(