Update core/analyze.py

core/analyze.py

@@ -2,7 +2,7 @@ import os
 import time
 import json
 import logging
-from openai import OpenAI
+from groq import Groq
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -11,19 +11,15 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger(__name__)
 
-# Configure OpenAI Client
-api_key = os.getenv("OPENROUTER_API_KEY")
-MODEL_NAME = os.getenv("OPENROUTER_MODEL", "arcee-ai/trinity-large-preview:free")
+# Configure Groq Client
+api_key = os.getenv("GROQ_API_KEY")
+MODEL_NAME = os.getenv("GROQ_MODEL", "llama-3.1-8b-instant")
 
-
-client = OpenAI(
-    base_url="https://openrouter.ai/api/v1",
-    api_key=api_key
-)
+client = Groq(api_key=api_key)
 
 
 def analyze_transcript(transcript):
-    """Analyze transcript using OpenRouter via Env Key."""
+    """Analyze transcript using Groq API."""
 
     prompt = f"""
     You are an expert video editor and viral content strategist. 
@@ -68,7 +64,7 @@ def analyze_transcript(transcript):
 
     max_retries = 3
     base_delay = 5
-    content = None  # FIX: initialize content to avoid UnboundLocalError
+    content = None
 
     for attempt in range(max_retries):
         try:
@@ -78,10 +74,6 @@ def analyze_transcript(transcript):
                     {"role": "system", "content": "You are a helpful assistant that outputs only valid JSON."},
                     {"role": "user", "content": prompt}
                 ],
-                extra_headers={
-                    "HTTP-Referer": "https://github.com/Start-To-End-AI",
-                    "X-Title": "Video Clipper AI",
-                },
                 temperature=0.7,
             )
 
@@ -102,7 +94,7 @@ def analyze_transcript(transcript):
             return {"content": content}
 
         except Exception as e:
-            print(f"❌ Error in OpenRouter analysis: {e}")
+            print(f"❌ Error in Groq analysis: {e}")
             if attempt < max_retries - 1:
                 wait_time = base_delay * (2 ** attempt)
                 print(f"⚠️ Retrying task in {wait_time}s...")
@@ -114,14 +106,11 @@ def analyze_transcript(transcript):
     return {"content": '{"segments": []}'}
 
 
-# Smart chunking system for long transcripts
 def smart_chunk_transcript(transcript, max_tokens=4000):
     """
     Split transcript into coherent chunks at sentence boundaries
     while preserving context and meaning.
     """
-    import json
-    # Simple sentence-based chunking
     sentences = transcript.replace('\n', ' ').split('. ')
     chunks = []
     current_chunk = []