Update app.py

app.py

@@ -5,7 +5,7 @@ import docx
 import requests
 import json
 
-# Function to extract text from PDF (unchanged)
+# Function to extract text from PDF
 def extract_text_from_pdf(file):
     pdf_reader = PyPDF2.PdfReader(file)
     text = ""
@@ -13,13 +13,13 @@ def extract_text_from_pdf(file):
         text += page.extract_text()
     return text
 
-# Function to extract text from Word document (unchanged)
+# Function to extract text from Word document
 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)
+# Function to process uploaded file
 def process_uploaded_file(file):
     filename = file.name.lower()  # Case-insensitive check
     if filename.endswith(".pdf"):
@@ -29,16 +29,15 @@ def process_uploaded_file(file):
     else:
         raise ValueError("Unsupported file format. Please upload a PDF or Word document.")
 
-# Function to call Together API (unchanged except for timeout)
+# Function to call Together API
 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 expert in Applicant Tracking Systems (ATS) and resume optimization. 
+        {
+            "role": "system", 
+            "content": """You are an expert in Applicant Tracking Systems (ATS) and resume optimization. 
 Your task is to analyze resumes against job descriptions to provide detailed, actionable feedback 
 that helps candidates pass ATS screening and impress human reviewers.
 
@@ -52,10 +51,10 @@ Follow these guidelines when analyzing resumes:
 7. Be thorough but concise in your analysis
 
 Your analysis must be comprehensive yet practical for the candidate to implement."""
-    },
-    {
-        "role": "user", 
-        "content": f"""
+        },
+        {
+            "role": "user", 
+            "content": f"""
 Analyze the following resume against the job description for ATS compatibility and hiring potential.
 
 RESUME:
@@ -113,8 +112,8 @@ Provide a detailed analysis in EXACTLY the following JSON format:
 
 Ensure all recommendations are specific, actionable, and tailored to both the resume content and job description.
 """
-    }
-]
+        }
+    ]
     
     payload = {
         "model": "mistralai/Mistral-7B-Instruct-v0.3",
@@ -136,30 +135,40 @@ Ensure all recommendations are specific, actionable, and tailored to both the re
         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
+        # The key fix: Safely parse the JSON content
+        try:
+            parsed = json.loads(content)
+            # Check if the JSON structure is valid for our use case
+            # Note: We've updated this validation to match our new JSON structure
+            if "ATS_Compatibility" not in parsed or "Overall_Assessment" not in parsed:
+                return {"error": "API returned unexpected JSON structure"}
+            return parsed
+        except json.JSONDecodeError as e:
+            return {"error": f"Failed to parse API response: {str(e)}", "raw_content": content[:500] + "..." if len(content) > 500 else content}
         
-    except Exception as e:
+    except requests.exceptions.RequestException as e:
         return {"error": f"API request failed: {str(e)}"}
+    except Exception as e:
+        return {"error": f"Unexpected error: {str(e)}"}
 
-# Main function (only added JSON validation)
+# Main function
 def analyze_resume(file, job_description):
     try:
+        if not file:
+            return json.dumps({"error": "Please upload a resume file"}, indent=2)
+        if not job_description:
+            return json.dumps({"error": "Please enter a job description"}, indent=2)
+        
         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
+        # Return as formatted JSON string for better display
+        return json.dumps(result, indent=2)
         
     except Exception as e:
-        return json.dumps({"error": str(e)}, indent=2)
+        return json.dumps({"error": f"Error analyzing resume: {str(e)}"}, indent=2)
 
-# Gradio interface (only changed file component type)
+# Gradio interface
 with gr.Blocks(fill_height=True, title="Smart ATS Resume Analyzer") as demo:
     with gr.Sidebar():
         gr.Markdown("# Smart ATS Resume Analyzer")
@@ -170,7 +179,7 @@ with gr.Blocks(fill_height=True, title="Smart ATS Resume Analyzer") as demo:
             resume_upload = gr.File(
                 label="Upload Resume (PDF or Word)", 
                 file_types=[".pdf", ".docx"],
-                type="filepath"  # Only this line changed
+                type="filepath"
             )
             job_desc = gr.Textbox(label="Job Description", lines=10, placeholder="Paste the job description here...")
             submit_btn = gr.Button("Analyze Resume")