Update app.py

app.py

@@ -39,8 +39,8 @@ Resume:
 
 Only list the missing keywords as bullet points.
 """
-    result = summarizer(prompt, max_new_tokens=300, do_sample=True)[0]['generated_text']
-    return result.strip()
+    result = summarizer(prompt, max_new_tokens=300, do_sample=True, return_full_text=False)[0]
+    return result.get('generated_text', result.get('summary_text', str(result))).strip()
 
 # 🔍 Prompt for dynamic section classification and feedback
 def build_dynamic_prompt(job_desc, resume_text, analyze_with_jd):
@@ -67,13 +67,14 @@ def analyze_resume(job_desc, resume_text, analyze_with_jd):
     user_prompt = build_dynamic_prompt(job_desc, resume_text, analyze_with_jd)
 
     try:
-        response = summarizer(user_prompt, max_new_tokens=512, do_sample=True)[0]['generated_text']
+        result = summarizer(user_prompt, max_new_tokens=512, do_sample=True, return_full_text=False)[0]
+        response_text = result.get('generated_text', result.get('summary_text', str(result))).strip()
         if analyze_with_jd and job_desc:
             matched, _ = compare_keywords(resume_text, job_desc)
             highlighted_resume = highlight_keywords(resume_text, matched)
             llm_missing_keywords = extract_missing_keywords_with_llm(job_desc, resume_text)
-            return f"### 🔍 Resume with Highlighted Matches\n\n{highlighted_resume}\n\n---\n**✅ Matched Keywords:** {', '.join(sorted(matched)) or 'None'}\n\n**❌ LLM-Inferred Missing Keywords:**\n{llm_missing_keywords}\n\n---\n{response.strip()}"
-        return response.strip()
+            return f"### 🔍 Resume with Highlighted Matches\n\n{highlighted_resume}\n\n---\n**✅ Matched Keywords:** {', '.join(sorted(matched)) or 'None'}\n\n**❌ LLM-Inferred Missing Keywords:**\n{llm_missing_keywords}\n\n---\n{response_text}"
+        return response_text
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
@@ -85,13 +86,13 @@ def create_ui():
                 analyze_checkbox = gr.Checkbox(label="Analyze with Job Description", value=True)
                 job_desc = gr.Textbox(label="Job Description", lines=6, placeholder="Paste job description here...")
                 resume_text = gr.Textbox(label="Resume Text", lines=16, placeholder="Paste resume content here...")
+                submit_btn = gr.Button("Analyze Resume")
             with gr.Column():
                 output_analysis = gr.Markdown(label="Resume Analysis and Suggestions")
 
-        resume_text.change(fn=analyze_resume, inputs=[job_desc, resume_text, analyze_checkbox], outputs=output_analysis)
-        job_desc.change(fn=analyze_resume, inputs=[job_desc, resume_text, analyze_checkbox], outputs=output_analysis)
+        submit_btn.click(fn=analyze_resume, inputs=[job_desc, resume_text, analyze_checkbox], outputs=output_analysis)
 
     return demo
 
 if __name__ == '__main__':
-    create_ui().launch()
+    create_ui().launch()