Update app.py

app.py

@@ -9,6 +9,13 @@ import json
 from pathlib import Path
 from typing import Dict, List, Optional
 
+# Define the storage directory in the Hugging Face Space
+STORAGE_DIR = os.environ.get('STORAGE_DIR', 'data')
+
+def ensure_storage_dir():
+    """Create storage directory if it doesn't exist"""
+    os.makedirs(STORAGE_DIR, exist_ok=True)
+
 def initialize_email_client():
     sendgrid_key = os.environ.get("SENDGRID_API_KEY")
     sender_email = os.environ.get("SENDER_EMAIL")
@@ -24,33 +31,63 @@ def initialize_email_client():
         return None
 
 class UserProfile:
-    def __init__(self, storage_path: str = "profiles.xlsx"):
-        self.storage_path = storage_path
+    def __init__(self):
+        ensure_storage_dir()
+        self.storage_path = os.path.join(STORAGE_DIR, 'profiles.json')
         self._ensure_storage_exists()
     
     def _ensure_storage_exists(self):
-        if not Path(self.storage_path).exists():
-            df = pd.DataFrame(columns=['user_id', 'name', 'industry', 'background', 'target_audience'])
-            df.to_excel(self.storage_path, index=False)
+        if not os.path.exists(self.storage_path):
+            with open(self.storage_path, 'w') as f:
+                json.dump({"profiles": []}, f)
     
     def save_profile(self, profile_data: Dict) -> bool:
         try:
-            df = pd.read_excel(self.storage_path)
-            profile_data['user_id'] = len(df) + 1
-            df = pd.concat([df, pd.DataFrame([profile_data])], ignore_index=True)
-            df.to_excel(self.storage_path, index=False)
+            with open(self.storage_path, 'r') as f:
+                data = json.load(f)
+            
+            # Update existing profile or add new one
+            profiles = data.get("profiles", [])
+            profile_exists = False
+            for i, profile in enumerate(profiles):
+                if profile.get("name") == profile_data["name"]:
+                    profiles[i] = profile_data
+                    profile_exists = True
+                    break
+            
+            if not profile_exists:
+                profiles.append(profile_data)
+            
+            data["profiles"] = profiles
+            
+            with open(self.storage_path, 'w') as f:
+                json.dump(data, f)
             return True
         except Exception as e:
             print(f"Error saving profile: {e}")
             return False
+    
+    def get_profile(self, name: str) -> Optional[Dict]:
+        try:
+            with open(self.storage_path, 'r') as f:
+                data = json.load(f)
+            
+            for profile in data.get("profiles", []):
+                if profile.get("name") == name:
+                    return profile
+            return None
+        except Exception as e:
+            print(f"Error getting profile: {e}")
+            return None
 
 class EmailTemplate:
-    def __init__(self, templates_path: str = "templates.json"):
-        self.templates_path = templates_path
+    def __init__(self):
+        ensure_storage_dir()
+        self.templates_path = os.path.join(STORAGE_DIR, 'templates.json')
         self._ensure_templates_exist()
     
     def _ensure_templates_exist(self):
-        if not Path(self.templates_path).exists():
+        if not os.path.exists(self.templates_path):
             default_templates = {
                 "sales_pitch": {
                     "subject": "Innovative Solutions for {company}",
@@ -76,6 +113,11 @@ class EmailGenie:
         self.sender_email = os.environ.get("SENDER_EMAIL")
         self.user_profile = UserProfile()
         self.email_template = EmailTemplate()
+        self.current_profile = None
+    
+    def set_current_profile(self, name: str):
+        self.current_profile = self.user_profile.get_profile(name)
+        return self.current_profile is not None
     
     def structure_info(self, info: str) -> str:
         prompt = f"Please structure and summarize the following information about a person or company: {info}"
@@ -86,13 +128,17 @@ class EmailGenie:
         return chat_completion.choices[0].message.content
 
     def generate_email(self, template_name: str, context: Dict) -> str:
+        if not self.current_profile:
+            return "Please set up your profile first."
+        
         template = self.email_template.get_template(template_name)
         structured_info = self.structure_info(context.get('recipient_info', ''))
         
         prompt = f"""
         Using the following template and information, create a personalized email:
         Template: {template}
-        Sender: {context['sender_name']}
+        Sender: {self.current_profile['name']} ({self.current_profile['industry']})
+        Sender Background: {self.current_profile['background']}
         Recipient: {context['recipient']}
         Subject: {context['email_subject']}
         Recipient Information: {structured_info}
@@ -105,9 +151,14 @@ class EmailGenie:
         return chat_completion.choices[0].message.content
 
     def preview_email(self, email_content: str) -> str:
-        return f"=== Email Preview ===\n\n{email_content}"
+        if not self.current_profile:
+            return "Please set up your profile first."
+        return f"=== Email Preview ===\nFrom: {self.current_profile['name']}\n\n{email_content}"
     
     def send_email(self, to_email: str, subject: str, content: str) -> tuple[bool, str]:
+        if not self.current_profile:
+            return False, "Please set up your profile first."
+        
         if not self.sendgrid_client or not self.sender_email:
             return False, "Email sending is not configured"
             
@@ -135,8 +186,13 @@ def create_interface():
             background_input = gr.Textbox(label="Professional Background")
             target_input = gr.Textbox(label="Target Audience")
             save_profile_btn = gr.Button("Save Profile")
+            profile_status = gr.Textbox(label="Status", interactive=False)
         
         with gr.Tab("Generate Email"):
+            profile_name = gr.Textbox(label="Your Profile Name")
+            load_profile_btn = gr.Button("Load Profile")
+            profile_load_status = gr.Textbox(label="Profile Status", interactive=False)
+            
             template_dropdown = gr.Dropdown(
                 choices=["sales_pitch", "job_application"],
                 label="Select Template"
@@ -150,7 +206,7 @@ def create_interface():
         with gr.Tab("Send Email"):
             recipient_email = gr.Textbox(label="Recipient Email")
             send_btn = gr.Button("Send Email")
-            status_output = gr.Textbox(label="Status")
+            status_output = gr.Textbox(label="Status", interactive=False)
         
         def save_profile(name, industry, background, target):
             profile_data = {
@@ -160,11 +216,19 @@ def create_interface():
                 "target_audience": target
             }
             success = emailgenie.user_profile.save_profile(profile_data)
+            if success:
+                emailgenie.set_current_profile(name)
             return "Profile saved successfully!" if success else "Error saving profile"
         
+        def load_profile(name):
+            success = emailgenie.set_current_profile(name)
+            return "Profile loaded successfully!" if success else "Profile not found"
+        
         def generate_email_handler(template, subject, recipient, info):
+            if not emailgenie.current_profile:
+                return "Please load your profile first"
+            
             context = {
-                "sender_name": "Your Name",  # Could be fetched from profile
                 "email_subject": subject,
                 "recipient": recipient,
                 "recipient_info": info
@@ -179,7 +243,13 @@ def create_interface():
         save_profile_btn.click(
             save_profile,
             inputs=[name_input, industry_input, background_input, target_input],
-            outputs=gr.Textbox(label="Status")
+            outputs=profile_status
+        )
+        
+        load_profile_btn.click(
+            load_profile,
+            inputs=[profile_name],
+            outputs=profile_load_status
         )
         
         generate_btn.click(