Update main.py

main.py

@@ -6,6 +6,8 @@ from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import aiofiles
 import PyPDF2
+from langchain_openai import ChatOpenAI
+from langchain.schema import HumanMessage
 
 UPLOAD_FOLDER = "uploads"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
@@ -22,6 +24,28 @@ app.add_middleware(
 )
 
 
+llm = ChatOpenAI(
+    model_name="gpt-4o-mini",  # Use a valid model name like "gpt-4o" or "gpt-4-turbo"
+    temperature=0,
+    openai_api_key=os.getenv("OPENAI_API_KEY")
+)
+
+def parse_resume_text(text: str) -> dict:
+    prompt = f"""
+    Extract structured information from this resume text and return the result as a JSON object with the following keys:
+    - basics: {first_name, last_name, gender, emails, phone_numbers, address, total_experience_in_years, profession, summary, skills, has_driving_license}
+    - educations
+    - professional_experiences
+    - trainings_and_certifications
+    - languages
+    - awards
+    - references
+    Resume:
+    {text}
+    """
+    result = llm([HumanMessage(content=prompt)])
+    return result.content
+
 # ✅ Save uploaded file asynchronously
 async def save_file(file: UploadFile) -> str:
     filename = f"{uuid4()}_{file.filename}"
@@ -56,10 +80,13 @@ async def parse_resume(file: UploadFile = File(...)):
         text = extract_text_from_pdf(path)
         print("✅ Text extracted.")
 
+        json_result = parse_resume_text(text)
+        print("✅ JSON Created.")
+        
         os.remove(path)
         print("🧹 File removed.")
 
-        return JSONResponse(content={"text": text}, media_type="application/json")
+        return json_result
     
     except Exception as e:
         import traceback