Update app.py

app.py

@@ -4,7 +4,7 @@ import io
 import re
 import json
 import os
-import gc  # Added back for garbage collection
+import gc
 from huggingface_hub import login
 from dotenv import load_dotenv
 
@@ -73,20 +73,31 @@ def analyze_resume(pdf_file, job_desc=None, inference_fn=None):
     resume_text = extract_text_from_pdf(pdf_file)
     scores, total_score = calculate_scores(resume_text, job_desc)
     
-    prompt = f"""Analyze this resume and return JSON with:
-    2 key strengths (reference these scores: {scores}),
-    3 specific improvements,
-    2 missing skills (if job description provided).
-    Return ONLY valid JSON without markdown:"""
+    # Stricter prompt to ensure JSON output
+    prompt = f"""Given these scores: {scores}, return a valid JSON object with:
+    - "strengths": list of 2 key strengths referencing the scores,
+    - "improvements": list of 3 specific improvements,
+    - "missing_skills": list of 2 missing skills (use job description if provided: {job_desc or "None"}).
+    Output ONLY a valid JSON string, no extra text or markdown."""
     
     try:
-        # Use Together AI inference function
+        # Call Together AI inference
         result = inference_fn(prompt)
+        # Debug: Log the raw result
+        print(f"Raw inference result: {result}")
+        
+        if not result or result.strip() == "":
+            return {"error": "Empty response from inference API"}
+        
+        # Parse the response as JSON
+        parsed_result = json.loads(result)
         return {
             "score": {"total": total_score, "breakdown": scores},
-            "analysis": json.loads(result),
+            "analysis": parsed_result,
             "raw_text": resume_text[:500]
         }
+    except json.JSONDecodeError as e:
+        return {"error": f"Invalid JSON response: {result}"}
     except Exception as e:
         return {"error": str(e)}
 
@@ -97,7 +108,7 @@ with gr.Blocks(theme=gr.themes.Soft(), fill_height=True) as demo:
         gr.Markdown("Powered by mistralai/Mistral-7B-Instruct-v0.3 via Together AI API. Sign in to use.")
         button = gr.LoginButton("Sign in")
     
-    # Load Mistral-7B from Together AI (removed _js)
+    # Load Mistral-7B from Together AI
     inference = gr.load(
         "models/mistralai/Mistral-7B-Instruct-v0.3",
         accept_token=button,
@@ -120,4 +131,4 @@ with gr.Blocks(theme=gr.themes.Soft(), fill_height=True) as demo:
         queue=True
     )
 
-demo.launch(share=True)  # Added share=True for public link
+demo.launch(share=True)