Update main.py

main.py

@@ -97,16 +97,25 @@ async def get_data(input_file: UploadFile = File(...)):
                    text = vision(file_content)
         else:
             raise HTTPException(status_code=400, detail="Unsupported file type")
-        print(text,"=========================================")
+
         # Call Gemini (or another model) to extract required data
         prompt = f"""This is CV data: {text.strip()} 
-         I want only:
-
-         firstname, lastname, contact number, total years of experience, LinkedIn link, experience, skills
-
-         in JSON format only"""
+                IMPORTANT: The output should be a JSON array! Make Sure the JSON is valid.
+                                                                  
+                Example Output:
+                [
+                    "firstname" : firstname,
+                    "lastname" : lastname,
+                    "contact_number" : contact number
+                    "total_years_of_experience" : total years of experience,
+                    "LinkedIn_link" : LinkedIn link,
+                    "experience" : experience,
+                    "skills" : skills
+                ]
+                """
         
         response = model_text.generate_content(prompt)
+        print(response.text)
         data = json.loads(response.text.replace("```json", "").replace("```", ""))
         return {"data": data}