Update main.py

main.py

@@ -10,7 +10,13 @@ from fastapi.middleware.cors import CORSMiddleware
 from PyPDF2 import PdfReader
 
 
+import google.generativeai as genai
+import json
+
 secret = os.environ["key"]
+genai.configure(api_key=secret)
+model_vision = genai.GenerativeModel('gemini-pro-vision')
+model_text = genai.GenerativeModel('gemini-pro')
 
 app = FastAPI()
 
@@ -33,8 +39,17 @@ async def get_data(pdf: UploadFile = File(...)):
         for page in pdf_reader.pages:
             text += page.extract_text()
 
-        # Return extracted text
-        return {"text": text.strip()}
+        # call gemini to get required data extracted text
+        prompt = f"""this is cv data : {text.strip()} 
+         i want only 
+
+         fisrtname,lastname,contact number,total years of experince,linkdn link,experinece,skils 
+
+         in json format only"""
+        
+        response = model_text.generate_content(prompt)
+        data = json.loads(response.text.replace("```json","").replace("```",""))
+        return {"data":data}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error processing PDF: {str(e)}")