Update main.py

main.py

@@ -10,6 +10,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from utils import read_file, fetch_job_description, optimize_resume_api
 import logging
 import os
+from fastapi.responses import JSONResponse
 API_SECRET = os.getenv("API_SECRET")
 
 # Set up logging
@@ -22,10 +23,6 @@ app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_credentials=True,
 
 security = HTTPBearer()
 
-class OptimizationResponse(BaseModel):
-    optimizedResume: str
-    metadata: dict
-
 class ErrorResponse(BaseModel):
     detail: str
 
@@ -35,6 +32,13 @@ def verify_token(credentials: HTTPAuthorizationCredentials = Depends(security)):
         raise HTTPException(status_code=401, detail="Invalid authentication token")
     return token
 
+import re
+
+def extract_xml_content(text, tag):
+    pattern = f"<{tag}>(.*?)</{tag}>"
+    matches = re.findall(pattern, text, re.DOTALL)
+    return matches
+
 @app.post("/api/v1/optimize-resume", response_model=OptimizationResponse, responses={400: {"model": ErrorResponse}, 401: {"model": ErrorResponse}, 500: {"model": ErrorResponse}})
 async def optimize_resume(
     resume: Optional[UploadFile] = File(None),
@@ -63,16 +67,22 @@ async def optimize_resume(
             job_description_content = fetch_job_description(jobDescriptionUrl)
 
         # Perform optimization
-        optimized_resume = optimize_resume_api(resume_content, job_description_content)
-
-        metadata = {
-            "original_resume":resume_content,
-            "original_jd": job_description_content}
-
-        return OptimizationResponse(
-            optimizedResume=optimized_resume,
-            metadata=metadata
-        )
+        output = optimize_resume_api(resume_content, job_description_content)
+        optimized_resume = extract_xml_content(output, "optimized_resume")
+        changes_made = extract_xml_content(output, "changes_made")
+        
+        # Prepare the response
+        response_data = {
+            "optimizedResume": optimized_resume[0] if optimized_resume else "",
+            "changesMade": changes_made[0] if changes_made else "",
+            "metadata": {
+                "original_resume": resume_content,
+                "original_jd": job_description_content
+            }
+        }
+
+        return JSONResponse(content=response_data)
+        
     except ValueError as ve:
         logger.error(f"ValueError occurred: {str(ve)}", exc_info=True)
         raise HTTPException(status_code=400, detail=str(ve))