Update app.py

app.py

@@ -1,70 +1,54 @@
 import gradio as gr
-import pandas as pd
-from transformers import pipeline
-from email.mime.text import MIMEText
-import smtplib
-import io
-
-# Load Hugging Face FLAN-T5 Model
-email_generator = pipeline("text2text-generation", model="google/flan-t5-base")
-
-# Email Sending Function
-def send_email(to_email, subject, body, from_email, from_password):
-    msg = MIMEText(body, "plain")
-    msg["Subject"] = subject
-    msg["From"] = from_email
-    msg["To"] = to_email
-
-    try:
-        server = smtplib.SMTP_SSL("smtp.gmail.com", 465)
-        server.login(from_email, from_password)
-        server.sendmail(from_email, [to_email], msg.as_string())
-        server.quit()
-        return f"✅ Email sent to {to_email}"
-    except Exception as e:
-        return f"❌ Failed to send to {to_email}: {str(e)}"
-
-# Main Processing Function
-def process_and_send(csv_file, sender_email, sender_pass):
-    try:
-        # Decode CSV file from bytes to string
-        df = pd.read_csv(io.StringIO(csv_file.decode("utf-8")))
-    except Exception as e:
-        return f"❌ Failed to read CSV: {e}"
-
-    logs = []
-
-    for index, row in df.iterrows():
-        name = row["Name"]
-        email = row["Email"]
-        status = row["Status"]
-        internship_title = row["Internship_Title"]
-        company_name = row["Company_Name"]
-
-        prompt = f"Write a short and professional email to {name} who applied for the {internship_title} internship at {company_name}. Their application status is: {status}."
-
-        try:
-            mail_body = email_generator(prompt, max_length=200)[0]["generated_text"]
-            subject = f"Regarding your application for {internship_title} at {company_name}"
-            result = send_email(email, subject, mail_body, sender_email, sender_pass)
-        except Exception as e:
-            result = f"❌ Error for {email}: {str(e)}"
-
-        logs.append(result)
-
-    return "\n".join(logs)
-
-# Gradio UI
-iface = gr.Interface(
-    fn=process_and_send,
-    inputs=[
-        gr.File(label="📂 Upload CSV of Applicants", file_types=[".csv"]),
-        gr.Text(label="📧 Your Email (Gmail)", placeholder="example@gmail.com"),
-        gr.Text(label="🔐 Your Email Password or App Password", type="password"),
-    ],
-    outputs="text",
-    title="AI-Powered Mailing System (No API Key Needed)",
-    description="Upload your applicant CSV and send personalized emails using Hugging Face FLAN-T5 model.",
-)
-
-iface.launch()
+import base64
+import pdfkit
+from jinja2 import Template
+
+# Load HTML template
+with open("template.html", "r", encoding="utf-8") as f:
+    template_html = f.read()
+
+def generate_resume(name, job_title, email, phone, summary, skills, education, experience, photo):
+    photo_base64 = ""
+    if photo is not None:
+        photo_base64 = base64.b64encode(photo.read()).decode('utf-8')
+
+    data = {
+        "name": name,
+        "job_title": job_title,
+        "email": email,
+        "phone": phone,
+        "summary": summary,
+        "skills": [s.strip() for s in skills.split(",")],
+        "education": [e.strip() for e in education.split(";")],
+        "experience": [x.strip() for x in experience.split(";")],
+        "photo_base64": photo_base64
+    }
+
+    template = Template(template_html)
+    rendered_html = template.render(**data)
+
+    with open("resume_output.html", "w", encoding="utf-8") as f:
+        f.write(rendered_html)
+
+    pdfkit.from_file("resume_output.html", "resume.pdf")
+    return "resume.pdf"
+
+inputs = [
+    gr.Text(label="Full Name"),
+    gr.Text(label="Job Title"),
+    gr.Text(label="Email"),
+    gr.Text(label="Phone"),
+    gr.Textbox(label="Professional Summary", lines=4),
+    gr.Textbox(label="Skills (comma-separated)"),
+    gr.Textbox(label="Education (separate with ; )"),
+    gr.Textbox(label="Experience (separate with ; )"),
+    gr.File(label="Upload Profile Photo (Optional)", type="binary")
+]
+
+gr.Interface(
+    fn=generate_resume,
+    inputs=inputs,
+    outputs=gr.File(label="📄 Download Resume PDF"),
+    title="🧠 AI Resume Builder",
+    description="Enter your details and generate a polished PDF resume instantly."
+).launch()