email data

app/main.py

@@ -5,7 +5,7 @@ from typing import Optional, Tuple
 
 from fastapi import FastAPI, HTTPException, Response
 from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
 
 # Optional .env for local testing
 from dotenv import load_dotenv
@@ -34,6 +34,8 @@ app.add_middleware(
 
 
 RUNS = Path("runs"); RUNS.mkdir(parents=True, exist_ok=True)
+DATA_DIR = Path("data"); DATA_DIR.mkdir(parents=True, exist_ok=True)
+EMAILS_FILE = DATA_DIR / "emails.txt"
 
 # ---------------- simple 10 RPM rate limiter ----------------
 class RateLimiter:
@@ -383,6 +385,22 @@ def refine_loop(user_prompt: str, max_error_refines: int = 3, do_visual_refine:
 class PromptIn(BaseModel):
     prompt: str
 
+class EmailIn(BaseModel):
+    email: str
+
+    @property
+    def sanitized(self) -> str:
+        return self.email
+
+    @validator("email")
+    def validate_email(cls, value: str) -> str:
+        cleaned = value.strip()
+        if not cleaned:
+            raise ValueError("Email cannot be empty")
+        if not re.match(r"^[^@\s]+@[^@\s]+\.[^@\s]+$", cleaned):
+            raise ValueError("Email is not valid")
+        return cleaned
+
 @app.get("/")
 def health():
     return {"ok": True, "model": MODEL, "has_gemini": bool(API_KEY)}
@@ -408,3 +426,14 @@ def generate_and_render(inp: PromptIn):
         media_type="video/mp4",
         headers={"Content-Disposition": 'inline; filename="result.mp4"'}
     )
+
+@app.post("/store-email")
+def store_email(email: EmailIn):
+    """Append the provided email address to a server-side text file."""
+    sanitized_email = email.sanitized
+    try:
+        with EMAILS_FILE.open("a", encoding="utf-8") as fh:
+            fh.write(f"{sanitized_email}\n")
+    except Exception:
+        raise HTTPException(500, "Failed to save email address")
+    return {"stored": True}