Update parser_logic.py

parser_logic.py

@@ -1,6 +1,5 @@
 import os
 import json
-import re
 import logging
 import fitz  # PyMuPDF
 from google import genai
@@ -14,11 +13,11 @@ load_dotenv()
 
 api_key = os.getenv("GEMINI_API_KEY")
 if not api_key:
-    # Fallback to check if key is empty
     raise ValueError("GEMINI_API_KEY is missing from Secrets.")
 
 client = genai.Client(api_key=api_key)
 
+
 def extract_text_from_stream(file_bytes: bytes) -> str:
     text = ""
     try:
@@ -30,22 +29,32 @@ def extract_text_from_stream(file_bytes: bytes) -> str:
         raise ValueError("Failed to extract text from PDF.")
     return text
 
+
 def parse_resume_with_ai(resume_text: str) -> dict:
     prompt = """
-    Extract the following data from this resume text:
-    - name, email, phone, skills (list), and summary.
-    Return strictly valid JSON.
-    """ 
-    
+    Extract the following information from the resume text below.
+    Return STRICTLY valid JSON with these fields:
+
+    {
+      "name": "",
+      "email": "",
+      "phone": "",
+      "skills": [],
+      "summary": ""
+    }
+    """
+
     try:
         response = client.models.generate_content(
-            model="gemini-1.5-flash",
+            model="gemini-pro",  # ✅ FIXED MODEL
             contents=prompt + "\n\n" + resume_text[:10000],
             config=types.GenerateContentConfig(
                 response_mime_type="application/json"
             )
         )
+
         return json.loads(response.text)
+
     except Exception as e:
         logger.error(f"AI Processing Error: {e}")
-        return {"error": f"AI Processing failed: {str(e)}"}
+        return {"error": f"AI Processing failed: {str(e)}"}