Update main.py

main.py

@@ -6,6 +6,7 @@ from sentence_transformers import SentenceTransformer
 from thefuzz import process, fuzz
 import random
 import os
+from supabase import create_client, Client
 
 PINECONE_API_KEY = os.environ.get("PINECONE_API_KEY")
 INDEX_NAME = "cine-match"
@@ -26,6 +27,11 @@ if not PINECONE_API_KEY:
         "PINECONE_API_KEY not set. Add it to environment or ml-engine/.env"
     )
 
+SUPABASE_URL = os.environ.get("SUPABASE_URL")
+SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
+
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+
 app = FastAPI()
 
 app.add_middleware(
@@ -53,8 +59,18 @@ class FinalRecommendationRequest(BaseModel):
     selected_titles: list[str]
     genre: str
 
+def log_search(query_text: str, search_type: str):
+    try:
+        supabase.table("logs").insert({
+            "query": query_text,
+            "type": search_type
+        }).execute()
+    except Exception as e:
+        print(f"Error logging search: {e}")
+
 @app.post("/search")
 def semantic_search(req: SearchRequest):
+    log_search(req.query, "search_bar")
     try:
         query_vector = model.encode(req.query).tolist()
         filter_dict = {}
@@ -127,6 +143,7 @@ def get_quiz_items(req: QuizRequest):
 def hybrid_recommend(req: FinalRecommendationRequest):
     joined_titles = ", ".join(req.selected_titles)
     semantic_query = f"{req.mood} {req.genre} similar to {joined_titles}"
+    log_search(semantic_query, "hybrid_wizard")
     query_vector = model.encode(semantic_query).tolist()
     results = index.query(vector=query_vector, top_k=60, include_metadata=True)