Update app/services/extraction.py

app/services/extraction.py

@@ -6,7 +6,7 @@ from dotenv import load_dotenv
 from difflib import SequenceMatcher
 
 load_dotenv() 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+genai.configure(api_key=os.getenv("EXTRACTION_API_KEY"))
 
 # Define categories from dataset
 RECIPE_CATEGORIES = [
@@ -359,18 +359,26 @@ Output:
     ]
 
     # Send the prompt to OpenAI API
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages,
-        temperature=0,
-        max_tokens=150,
-        top_p=1,
-        frequency_penalty=0,
-        presence_penalty=0,
-    )
+    prompt = ""
+    for message in messages:
+        if message["role"] == "system":
+            prompt += message["content"] + "\n\n"
+        else:
+            prompt += message["content"]
 
+    # Configure the Gemini model
+    model = genai.GenerativeModel('gemini-2.5-flash')
+    
+    # Generate response
+    response = model.generate_content(prompt, 
+                                     generation_config=genai.types.GenerationConfig(
+                                         temperature=0,
+                                         max_output_tokens=150,
+                                         top_p=1
+                                     ))
+    
     # Process the response
-    output_text = response['choices'][0]['message']['content'].strip()
+    output_text = response.text.strip()
     
     try:
         result = json.loads(output_text)