Update app.py

app.py

@@ -30,52 +30,81 @@ def process_uploaded_file(file):
         raise ValueError("Unsupported file format. Please upload a PDF or Word document.")
 
 # Function to call Together API (unchanged except for timeout)
-def analyze_with_mistral(resume_text, job_description):
-    TOGETHER_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
-    url = "https://api.together.xyz/v1/chat/completions"
-    
-    messages= [
-    {
-        "role": "system",
-        "content": "You are an AI expert in ATS resume analysis. Your task is to analyze a resume against a job description for ATS compatibility and return the result in          a structured JSON format. You must strictly follow the JSON schema provided and output ONLY JSON."
-    },
-    {
-        "role": "user",
-        "content": f"""Analyze the provided resume against the job description for ATS compatibility. Evaluate **keywords, formatting, skills, experience relevance, and education** based on ATS best practices.\n\nStrict Instructions:\n- **Return ONLY valid JSON** that matches the exact schema below. NO explanations, comments, or extra text.\n- **Ensure all fields are populated** (use empty lists or defaults if no suggestions apply).\n- **Use integer values (0-100)** for scores.\n- **Ensure 'Overall' is the rounded average** of all score fields.\n\n### **JSON Schema:**\n```json\n{{\n  \"ATS Parameters\": {{\n    \"Keywords\": {{\n      \"Match\": <integer 0-100>,\n      \"Recommendation\": [<list of 0-5 strings>]\n    }},\n    \"Formatting\": {{\n      \"Match\": <integer 0-100>,\n      \"Recommendation\": [<list of 0-5 strings>]\n    }},\n    \"Skills Match\": {{\n      \"Match\": <integer 0-100>,\n      \"Recommendation\": [<list of 0-5 strings>]\n    }},\n    \"Experience Relevance\": {{\n      \"Match\": <integer 0-100>,\n      \"Recommendation\": [<list of 0-5 strings>]\n    }},\n    \"Education\": {{\n      \"Match\": <integer 0-100>,\n      \"Recommendation\": [<list of 0-5 strings>]\n    }}\n  }},\n  \"Score\": {{\n    \"Keywords\": <integer 0-100>,\n    \"Formatting\": <integer 0-100>,\n    \"Skills Match\": <integer 0-100>,\n    \"Experience Relevance\": <integer 0-100>,\n    \"Education\": <integer 0-100>,\n    \"Overall\": <integer 0-100, rounded average of all scores>\n  }}\n}}\n```\n\n### **Processing Rules:**\n- **Score Calculation:** Base your evaluation on **ATS resume best practices** (keyword density, formatting simplicity, skills relevance, etc.).\n- **Recommendations:** Provide actionable suggestions to improve compatibility.\n- **Strict JSON Format:** No preamble, no explanations—return ONLY the completed JSON.\n\nResume:\n{resume_text}\n\nJob Description:\n{job_description}"""
-
-    }
-]
 
   
+def analyze_with_huggingface(resume_text, job_description):
+    HF_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
+    API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3"
     
-    payload = {
-        "model": "mistralai/Mistral-7B-Instruct-v0.3",
-        "messages": messages,
-        "max_tokens": 1500,
-        "temperature": 0.7,
-        "top_p": 0.9,
-        "response_format": {"type": "json_object"}
-    }
+    # Truncate inputs to stay within token limits
+    resume_text = resume_text[:3000]  # Limit resume to 3000 chars
+    job_description = job_description[:1000]  # Limit JD to 1000 chars
+    
+    prompt = f"""<s>[INST] <<SYS>>
+You are an AI expert in ATS resume analysis. Return ONLY valid JSON matching this schema:
+{{
+  "ATS Parameters": {{
+    "Keywords": {{"Match": <int>, "Recommendation": [<str>]}},
+    "Formatting": {{"Match": <int>, "Recommendation": [<str>]}},
+    "Skills Match": {{"Match": <int>, "Recommendation": [<str>]}},
+    "Experience Relevance": {{"Match": <int>, "Recommendation": [<str>]}},
+    "Education": {{"Match": <int>, "Recommendation": [<str>]}}
+  }},
+  "Score": {{
+    "Keywords": <int>,
+    "Formatting": <int>,
+    "Skills Match": <int>,
+    "Experience Relevance": <int>,
+    "Education": <int>,
+    "Overall": <int>
+  }}
+}}
+<</SYS>>
+
+Analyze this resume against the job description for ATS compatibility:
+Resume: {resume_text}
+Job Description: {job_description}
+Return ONLY the JSON output with no additional text or explanations.[/INST]"""
     
     headers = {
-        "Authorization": f"Bearer {TOGETHER_API_KEY}",
+        "Authorization": f"Bearer {HF_API_KEY}",
         "Content-Type": "application/json",
     }
     
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": 800,
+            "temperature": 0.7,
+            "do_sample": True,
+            "return_full_text": False
+        }
+    }
+    
     try:
-        response = requests.post(url, json=payload, headers=headers, timeout=30)
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
         response.raise_for_status()
+        
+        # Extract JSON from response
         result = response.json()
-        content = result.get("choices", [{}])[0].get("message", {}).get("content", "{}")
+        if isinstance(result, list):
+            content = result[0].get("generated_text", "{}")
+        else:
+            content = result.get("generated_text", "{}")
+        
+        # Clean the output (sometimes models add extra text)
+        json_start = content.find('{')
+        json_end = content.rfind('}') + 1
+        json_str = content[json_start:json_end]
         
-        # Parse and validate the JSON structure
-        parsed = json.loads(content)
-        if "ATS Parameters" not in parsed or "Score" not in parsed:
-            raise ValueError("API returned invalid format")
+        parsed = json.loads(json_str)
+        if not all(k in parsed for k in ["ATS Parameters", "Score"]):
+            raise ValueError("Invalid JSON structure")
+            
         return parsed
         
     except Exception as e:
-        return {"error": f"API request failed: {str(e)}"}
+        return {"error": f"Hugging Face API error: {str(e)}"}
 
 # Main function (only added JSON validation)
 def analyze_resume(file, job_description):