Update app.py

app.py

@@ -1,59 +1,117 @@
-messages = [
+import os
+import gradio as gr
+import PyPDF2
+import docx
+import requests
+import json
+
+# Function to extract text from PDF (unchanged)
+def extract_text_from_pdf(file):
+    pdf_reader = PyPDF2.PdfReader(file)
+    text = ""
+    for page in pdf_reader.pages:
+        text += page.extract_text()
+    return text
+
+# Function to extract text from Word document (unchanged)
+def extract_text_from_docx(file):
+    doc = docx.Document(file)
+    text = "\n".join([para.text for para in doc.paragraphs])
+    return text
+
+# Function to process uploaded file (only added case-insensitive check)
+def process_uploaded_file(file):
+    filename = file.name.lower()  # Case-insensitive check
+    if filename.endswith(".pdf"):
+        return extract_text_from_pdf(file)
+    elif filename.endswith(".docx"):
+        return extract_text_from_docx(file)
+    else:
+        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"
+    
+    [
     {
         "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 specific JSON format."
+        "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": """
-        Analyze the provided resume against the job description for ATS compatibility. Assess keywords, formatting, skills, experience relevance, and education. Return the result by filling in the EXACT JSON template below with appropriate values and recommendations. Output ONLY the completed JSON, with no additional text, comments, or explanations. Ensure all fields are populated, even if with empty lists or default scores.
-
-        JSON Template to Fill:
-        {
-          "ATS Parameters": {
-            "Keywords": {
-              "Match": <integer 0-100>,
-              "Recommendation": [<list of 0-5 strings>]
-            },
-            "Formatting": {
-              "Match": <integer 0-100>,
-              "Recommendation": [<list of 0-5 strings>]
-            },
-            "Skills Match": {
-              "Match": <integer 0-100>,
-              "Recommendation": [<list of 0-5 strings>]
-            },
-            "Experience Relevance": {
-              "Match": <integer 0-100>,
-              "Recommendation": [<list of 0-5 strings>]
-            },
-            "Education": {
-              "Match": <integer 0-100>,
-              "Recommendation": [<list of 0-5 strings>]
-            }
-          },
-          "Score": {
-            "Keywords": <integer 0-100>,
-            "Formatting": <integer 0-100>,
-            "Skills Match": <integer 0-100>,
-            "Experience Relevance": <integer 0-100>,
-            "Education": <integer 0-100>,
-            "Overall": <integer 0-100, average of above scores>
-          }
-        }
-
-        Rules:
-        - Replace <integer 0-100> with an integer score between 0 and 100.
-        - Replace <list of 0-5 strings> with a list of 0 to 5 concise string recommendations (e.g., [] if none).
-        - "Overall" is the rounded average of the five "Score" values.
-        - Base analysis on ATS requirements (keyword density, simple formatting, etc.) and the job description.
-        - Output ONLY the filled JSON template, nothing else.
-
-        Resume:
-        {resume_text}
-
-        Job Description:
-        {job_description}
-        """
+        "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}"""
+    }
+]
+
+  
+    
+    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"}
+    }
+    
+    headers = {
+        "Authorization": f"Bearer {TOGETHER_API_KEY}",
+        "Content-Type": "application/json",
     }
-]
+    
+    try:
+        response = requests.post(url, json=payload, headers=headers, timeout=30)
+        response.raise_for_status()
+        result = response.json()
+        content = result.get("choices", [{}])[0].get("message", {}).get("content", "{}")
+        
+        # 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")
+        return parsed
+        
+    except Exception as e:
+        return {"error": f"API request failed: {str(e)}"}
+
+# Main function (only added JSON validation)
+def analyze_resume(file, job_description):
+    try:
+        resume_text = process_uploaded_file(file)
+        result = analyze_with_mistral(resume_text, job_description)
+        
+        # Ensure the output matches your original format
+        if isinstance(result, dict) and "error" in result:
+            return json.dumps(result, indent=2)
+        return result
+        
+    except Exception as e:
+        return json.dumps({"error": str(e)}, indent=2)
+
+# Gradio interface (only changed file component type)
+with gr.Blocks(fill_height=True, title="Smart ATS Resume Analyzer") as demo:
+    with gr.Sidebar():
+        gr.Markdown("# Smart ATS Resume Analyzer")
+        gr.Markdown("Upload your resume (PDF/Word) and enter a job description to get an ATS compatibility score.")
+    
+    with gr.Row():
+        with gr.Column(scale=1):
+            resume_upload = gr.File(
+                label="Upload Resume (PDF or Word)", 
+                file_types=[".pdf", ".docx"],
+                type="filepath"  # Only this line changed
+            )
+            job_desc = gr.Textbox(label="Job Description", lines=10, placeholder="Paste the job description here...")
+            submit_btn = gr.Button("Analyze Resume")
+        with gr.Column(scale=2):
+            output = gr.JSON(label="ATS Analysis Result")
+    
+    submit_btn.click(
+        fn=analyze_resume,
+        inputs=[resume_upload, job_desc],
+        outputs=output
+    )
+
+demo.launch()