Update main.py

main.py

@@ -13,7 +13,7 @@ from models import (
 from reddit_client import RedditClient
 from analyzer import call_groq_for_insight
 from serp_client import get_search_volume, search_competitors, search_funding_news
-from supabase_client import supabase
+from supabase_client import supabase_auth, supabase_db
 from config import GROQ_MODEL, GROQ_API_KEY, SUPABASE_JWT_SECRET
 
 # Groq client is imported in analyzer
@@ -22,7 +22,7 @@ reddit_client = RedditClient()
 
 app = FastAPI(title="Founder Intelligence Engine API", version="2.0")
 
-# CORS – allow all origins for development; restrict in production
+# CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -42,19 +42,19 @@ async def root():
 @app.post("/auth/signup", response_model=UserProfile)
 async def signup(user: UserSignup):
     try:
-        # 1. Create auth user
-        response = supabase.auth.sign_up({
+        # 1. Create auth user (using anon client)
+        response = supabase_auth.auth.sign_up({
             "email": user.email,
             "password": user.password,
             "options": {"data": {"full_name": user.full_name}}
         })
         user_data = response.user
 
-        # 2. Check if profile already exists (e.g., from a previous attempt)
-        existing = supabase.table("profiles").select("*").eq("id", user_data.id).execute()
+        # 2. Check if profile already exists (using service client to bypass RLS)
+        existing = supabase_db.table("profiles").select("*").eq("id", user_data.id).execute()
         if not existing.data:
             # Create profile record
-            supabase.table("profiles").insert({
+            supabase_db.table("profiles").insert({
                 "id": user_data.id,
                 "email": user_data.email,
                 "full_name": user.full_name,
@@ -73,13 +73,12 @@ async def signup(user: UserSignup):
         )
     except Exception as e:
         # If the user already exists in auth but not in profiles, we can still return a helpful message
-        # But Supabase will raise an exception for duplicate email.
         raise HTTPException(status_code=400, detail=str(e))
 
 @app.post("/auth/login")
 async def login(creds: UserLogin):
     try:
-        response = supabase.auth.sign_in_with_password({
+        response = supabase_auth.auth.sign_in_with_password({
             "email": creds.email,
             "password": creds.password
         })
@@ -89,34 +88,40 @@ async def login(creds: UserLogin):
 
 @app.get("/auth/me", response_model=UserProfile)
 async def get_me(user=Depends(get_current_user)):
+    # user already contains profile info from the token validation
+    # but we can fetch fresh profile from DB if needed
+    profile = supabase_db.table("profiles").select("*").eq("id", user["id"]).execute()
+    if not profile.data:
+        raise HTTPException(status_code=404, detail="Profile not found")
+    p = profile.data[0]
     return UserProfile(
         id=user["id"],
         email=user["email"],
-        full_name=user["profile"]["full_name"],
-        subscription_tier=user["profile"]["subscription_tier"],
-        subscription_status=user["profile"]["subscription_status"],
-        created_at=user["profile"]["created_at"]
+        full_name=p["full_name"],
+        subscription_tier=p["subscription_tier"],
+        subscription_status=p["subscription_status"],
+        created_at=p["created_at"]
     )
 
 # ====================== Subscription Plans ======================
 @app.get("/plans", response_model=List[SubscriptionPlan])
 async def list_plans():
-    plans = supabase.table("plans").select("*").execute()
+    plans = supabase_db.table("plans").select("*").execute()
     return [SubscriptionPlan(**p) for p in plans.data]
 
 # ====================== Helper: Check subscription limits ======================
 def check_subscription_limit(user_id: str, requested_posts: int):
     """Raise 403 if user exceeds plan limits."""
-    profile = supabase.table("profiles").select("subscription_tier").eq("id", user_id).execute()
+    profile = supabase_db.table("profiles").select("subscription_tier").eq("id", user_id).execute()
     tier = profile.data[0]["subscription_tier"] if profile.data else "free"
     # Fetch plan features
-    plan = supabase.table("plans").select("features").eq("name", tier.capitalize()).execute()
+    plan = supabase_db.table("plans").select("features").eq("name", tier.capitalize()).execute()
     if plan.data:
         features = plan.data[0]["features"]
         posts_limit = features.get("posts_per_month", 10)  # default free limit
         # Count posts this month for this user
         start_of_month = datetime.now().replace(day=1, hour=0, minute=0, second=0).isoformat()
-        count_res = supabase.table("insights").select("id", count="exact").eq("user_id", user_id).gte("created_at", start_of_month).execute()
+        count_res = supabase_db.table("insights").select("id", count="exact").eq("user_id", user_id).gte("created_at", start_of_month).execute()
         current_count = count_res.count if hasattr(count_res, 'count') else 0
         if current_count + requested_posts > posts_limit:
             raise HTTPException(status_code=403, detail=f"Monthly post limit of {posts_limit} reached. Upgrade your plan.")
@@ -152,8 +157,8 @@ async def analyze(request: AnalysisRequest, user=Depends(get_current_user)):
                 "insight_data": insight,
                 "opportunity_score": insight["opportunity_score"]
             }
-            # Insert into Supabase
-            supabase.table("insights").insert(result).execute()
+            # Insert into Supabase using service client
+            supabase_db.table("insights").insert(result).execute()
             all_results.append({
                 "subreddit": sub,
                 "post_title": post.get("title"),
@@ -166,7 +171,7 @@ async def analyze(request: AnalysisRequest, user=Depends(get_current_user)):
 
 @app.get("/insights", response_model=List[InsightResponse])
 async def get_insights(user=Depends(get_current_user)):
-    res = supabase.table("insights").select("*").eq("user_id", user["id"]).order("created_at", desc=True).execute()
+    res = supabase_db.table("insights").select("*").eq("user_id", user["id"]).order("created_at", desc=True).execute()
     insights = []
     for row in res.data:
         insights.append(InsightResponse(
@@ -183,15 +188,13 @@ async def get_insights(user=Depends(get_current_user)):
 # ====================== Query / RAG ======================
 @app.post("/query", response_model=QueryResponse)
 async def query_insights(request: QueryRequest, user=Depends(get_current_user)):
-    # Simple keyword search over user's insights (full text search on insight_data)
-    insights = supabase.table("insights").select("*").eq("user_id", user["id"]).execute()
+    insights = supabase_db.table("insights").select("*").eq("user_id", user["id"]).execute()
     relevant = []
     for row in insights.data:
         data = row["insight_data"]
         text = f"{data.get('burning_pain','')} {data.get('startup_idea','')} {data.get('customer_type','')}"
         if request.query.lower() in text.lower():
             relevant.append(row)
-    # If no exact matches, return top N by opportunity_score
     if not relevant:
         relevant = sorted(insights.data, key=lambda x: x["opportunity_score"], reverse=True)[:request.n_results]
     else:
@@ -200,9 +203,7 @@ async def query_insights(request: QueryRequest, user=Depends(get_current_user)):
     if not relevant:
         raise HTTPException(status_code=404, detail="No relevant insights found")
 
-    # Build context
     context = "\n\n".join([f"From r/{r['subreddit']}: {r['insight_data'].get('burning_pain','')} {r['insight_data'].get('startup_idea','')}" for r in relevant])
-    # Call Groq to answer
     from groq import Groq
     groq = Groq(api_key=GROQ_API_KEY)
     try:
@@ -259,7 +260,7 @@ Insight: {request.insight}
 # ====================== Trends ======================
 @app.get("/trends", response_model=TrendResponse)
 async def trends(subreddit: str = None, user=Depends(get_current_user)):
-    query = supabase.table("insights").select("timestamp", "subreddit").eq("user_id", user["id"])
+    query = supabase_db.table("insights").select("timestamp", "subreddit").eq("user_id", user["id"])
     if subreddit:
         query = query.eq("subreddit", subreddit)
     res = query.execute()