Update app.py

app.py

@@ -35,14 +35,80 @@ def initialize_email_client():
         return None
 
 class UserProfile:
-    # [Previous UserProfile code remains the same]
-    # ... [UserProfile implementation remains unchanged]
-    pass
-
+    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 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:
+            with open(self.storage_path, 'r') as f:
+                data = json.load(f)
+            
+            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:
-    # [Previous EmailTemplate code remains the same]
-    # ... [EmailTemplate implementation remains unchanged]
-    pass
+    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 os.path.exists(self.templates_path):
+            default_templates = {
+                "sales_pitch": {
+                    "subject": "Innovative Solutions for {company}",
+                    "body": "Dear {name},\n\nI hope this email finds you well..."
+                },
+                "job_application": {
+                    "subject": "Experienced {role} Application",
+                    "body": "Dear {hiring_manager},\n\nI am writing to express my interest..."
+                }
+            }
+            with open(self.templates_path, 'w') as f:
+                json.dump(default_templates, f)
+    
+    def get_template(self, template_name: str) -> Dict:
+        with open(self.templates_path, 'r') as f:
+            templates = json.load(f)
+        return templates.get(template_name, {})
+
 
 class EmailGenie:
     def __init__(self):