Update app.py

app.py

@@ -6,15 +6,17 @@ from reportlab.pdfgen import canvas
 from reportlab.lib.utils import ImageReader
 from transformers import pipeline
 
-# Load FLAN-T5 model from Hugging Face
-summary_generator = pipeline("text2text-generation", model="google/flan-t5-base")
+# Load smaller faster model
+summary_generator = pipeline("text2text-generation", model="google/flan-t5-small")
 
-def generate_resume(name, job_title, email, phone, skills, education, experience, photo):
-    # Generate a passionate summary
-    prompt = f"Write a passionate summary for a person named {name}, applying as a {job_title}, skilled in {skills}, with experience in {experience}."
-    summary = summary_generator(prompt, max_length=200)[0]["generated_text"]
+def generate_resume(name, job_title, email, phone, skills, education, experience, photo, use_ai):
+    # Fast AI summary or fallback
+    if use_ai:
+        prompt = f"Write a passionate professional summary for {name}, applying as a {job_title}, skilled in {skills}, with experience in {experience}."
+        summary = summary_generator(prompt, max_length=200)[0]["generated_text"]
+    else:
+        summary = f"{name} is a passionate {job_title} with skills in {skills}, and experience in {experience}."
 
-    # Create PDF buffer
     buffer = io.BytesIO()
     pdf = canvas.Canvas(buffer, pagesize=A4)
     width, height = A4
@@ -77,7 +79,7 @@ def generate_resume(name, job_title, email, phone, skills, education, experience
     buffer.seek(0)
     return buffer
 
-# Gradio UI
+# Launch Gradio App
 iface = gr.Interface(
     fn=generate_resume,
     inputs=[
@@ -88,11 +90,13 @@ iface = gr.Interface(
         gr.Textbox(label="Skills (comma-separated)"),
         gr.Textbox(label="Education (use ; to separate items)"),
         gr.Textbox(label="Experience (use ; to separate items)"),
-        gr.File(label="Profile Photo (optional)", type="binary")
+        gr.File(label="Profile Photo (optional)", type="binary"),
+        gr.Checkbox(label="Use AI to auto-generate summary?", value=True)
     ],
     outputs=gr.File(label="📄 Download Resume PDF"),
-    title="🧠 AI Resume Builder (ReportLab + FLAN-T5)",
-    description="Generates a beautiful resume with AI-generated summary using FLAN-T5 and exports a working PDF (100% compatible with Hugging Face Spaces)."
+    title="🚀 AI Resume Builder (Fast Edition)",
+    description="Create a smart resume using FLAN-T5-Small for fast AI summary or manual fallback. Works great on Hugging Face Spaces!",
+    live=True
 )
 
 iface.launch()