Update main.py

main.py

@@ -8,8 +8,14 @@ import aiohttp
 import time
 import hmac
 import hashlib
+import sqlite3
+import requests
+from datetime import datetime
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
+app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=["*"], allow_headers=["*"])
 
 # ============================================
 # CORS - allow your frontend domains
@@ -119,7 +125,7 @@ async def get_user(request: Request, email: str = None, user_id: str = None):
         return {
             "userId": "admin",
             "name": "Administrator",
-            "email": "admin@domify.com",
+            "email": "domifyacademy@gmail.com",
             "tier": "Master",
             "expiry": (datetime.now() + timedelta(days=365)).isoformat(),
             "certPaid": True,
@@ -256,4 +262,134 @@ async def keep_alive_self():
 async def startup_event():
     print("🚀 Domify Academy Backend Starting...")
     asyncio.create_task(keep_alive_self())
-    print("✅ Keep-alive task started (pings every 5 minutes)")
+    print("✅ Keep-alive task started (pings every 5 minutes)")
+
+
+
+    
+# Google Apps Script Webhook URL (deploy this first)
+APPS_SCRIPT_URL = "https://script.google.com/macros/s/AKfycbwRuM5jLusScp0gzZqOmjO3_V_PXvn72vG4GSZ9P5jNb9kwav6_GVf5g0zWI7CwrXGb/exec"
+# ============================================
+# STORE CERTIFICATE CODE (First call from certificate page)
+# ============================================
+@app.post("/store-certificate-code")
+async def store_certificate_code(request: Request):
+    try:
+        data = await request.json()
+        verification_code = data.get('verification_code')
+        user_name = data.get('user_name')
+        user_email = data.get('user_email')
+        user_id = data.get('user_id')
+        tier = data.get('tier')
+        date = data.get('date')
+        
+        # Save to SQLite (without Drive URL yet)
+        conn = sqlite3.connect('domify_users.db')
+        c = conn.cursor()
+        c.execute("""
+            CREATE TABLE IF NOT EXISTS certificates (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                verification_code TEXT UNIQUE,
+                user_name TEXT,
+                user_email TEXT,
+                user_id TEXT,
+                tier TEXT,
+                date TEXT,
+                drive_url TEXT
+            )
+        """)
+        c.execute("""
+            INSERT OR REPLACE INTO certificates (verification_code, user_name, user_email, user_id, tier, date, drive_url)
+            VALUES (?, ?, ?, ?, ?, ?, ?)
+        """, (verification_code, user_name, user_email, user_id, tier, date, None))
+        conn.commit()
+        conn.close()
+        
+        # Send to Google Sheet (via Apps Script)
+        try:
+            requests.post(APPS_SCRIPT_URL, json={
+                "verification_code": verification_code,
+                "user_name": user_name,
+                "user_email": user_email,
+                "user_id": user_id,
+                "tier": tier,
+                "date": date,
+                "drive_url": ""
+            }, timeout=10)
+            print(f"✅ Sent to Google Sheet: {verification_code}")
+        except Exception as e:
+            print(f"Sheet error: {e}")
+        
+        return {"success": True, "message": "Certificate code stored"}
+        
+    except Exception as e:
+        return {"success": False, "error": str(e)}
+
+# ============================================
+# UPDATE DRIVE URL (Called after PDF uploaded to Drive)
+# ============================================
+@app.post("/update-drive-url")
+async def update_drive_url(request: Request):
+    try:
+        data = await request.json()
+        verification_code = data.get('verification_code')
+        drive_url = data.get('drive_url')
+        
+        # Update SQLite
+        conn = sqlite3.connect('domify_users.db')
+        c = conn.cursor()
+        c.execute("UPDATE certificates SET drive_url = ? WHERE verification_code = ?", (drive_url, verification_code))
+        conn.commit()
+        conn.close()
+        
+        # Also update Google Sheet (optional, send a second webhook)
+        try:
+            # First get the existing row data to update
+            conn = sqlite3.connect('domify_users.db')
+            c = conn.cursor()
+            c.execute("SELECT user_name, user_email, user_id, tier, date FROM certificates WHERE verification_code = ?", (verification_code,))
+            row = c.fetchone()
+            conn.close()
+            
+            if row:
+                requests.post(APPS_SCRIPT_URL, json={
+                    "verification_code": verification_code,
+                    "user_name": row[0],
+                    "user_email": row[1],
+                    "user_id": row[2],
+                    "tier": row[3],
+                    "date": row[4],
+                    "drive_url": drive_url
+                }, timeout=10)
+        except Exception as e:
+            print(f"Sheet update error: {e}")
+        
+        return {"success": True, "message": "Drive URL updated"}
+        
+    except Exception as e:
+        return {"success": False, "error": str(e)}
+
+# ============================================
+# VERIFY CERTIFICATE (Returns Drive URL so image can be displayed)
+# ============================================
+@app.get("/verify-certificate")
+async def verify_certificate(code: str, name: str):
+    conn = sqlite3.connect('domify_users.db')
+    c = conn.cursor()
+    c.execute("""
+        SELECT user_name, tier, date, drive_url FROM certificates 
+        WHERE verification_code = ? AND user_name = ?
+    """, (code.upper(), name))
+    row = c.fetchone()
+    conn.close()
+    
+    if row:
+        return {
+            "valid": True,
+            "name": row[0],
+            "tier": row[1],
+            "date": row[2],
+            "imageUrl": row[3]  # This is the Drive URL to display
+        }
+    else:
+        return {"valid": False, "error": "No matching certificate found"}