Update app.py

app.py

@@ -2,52 +2,76 @@ from transformers import pipeline
 from docx import Document
 import gradio as gr
 
+# Load lightweight model for basic text polishing (optional, not used heavily)
 generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b", device=-1)
 
-def generate_cv_docx(name, title, education, experience, skills):
-    prompt = f"""
-You are a CV assistant. Generate a clean, structured CV in this format:
-
-Full Name: {name}
-Title: {title}
-Education: {education}
-Experience: {experience}
-Skills: {skills}
-"""
-    result = generator(prompt, max_length=512, do_sample=True, temperature=0.7)[0]['generated_text']
+def generate_cv_docx(name, title, location, email, linkedin, github, portfolio,
+                     summary, tech_skills, soft_skills, education,
+                     languages, experience, projects, certifications):
     
-    # Generate Word doc
+    # Create document
     doc = Document()
+    
+    # Header
     doc.add_heading(name, level=1)
-    doc.add_paragraph(title, style='Intense Quote')
+    doc.add_paragraph(f"{title}")
+    doc.add_paragraph(f"{location} | {email}")
+    doc.add_paragraph(f"LinkedIn: {linkedin} | GitHub: {github} | Portfolio: {portfolio}")
+
+    # Sections
+    doc.add_heading("Professional Summary", level=2)
+    doc.add_paragraph(summary)
+
+    doc.add_heading("Technical Skills", level=2)
+    doc.add_paragraph(tech_skills)
+
+    doc.add_heading("Soft Skills", level=2)
+    doc.add_paragraph(soft_skills)
 
     doc.add_heading("Education", level=2)
     doc.add_paragraph(education)
 
+    doc.add_heading("Languages", level=2)
+    doc.add_paragraph(languages)
+
     doc.add_heading("Experience", level=2)
     doc.add_paragraph(experience)
 
-    doc.add_heading("Skills", level=2)
-    doc.add_paragraph(skills)
+    doc.add_heading("Key Projects", level=2)
+    doc.add_paragraph(projects)
+
+    doc.add_heading("Certifications", level=2)
+    doc.add_paragraph(certifications)
 
-    # Save document
+    # Save the file
     filepath = "generated_cv.docx"
     doc.save(filepath)
 
     return filepath
 
+# Gradio UI
 interface = gr.Interface(
     fn=generate_cv_docx,
     inputs=[
         gr.Textbox(label="Full Name"),
         gr.Textbox(label="Job Title"),
-        gr.Textbox(label="Education"),
-        gr.Textbox(label="Work Experience"),
-        gr.Textbox(label="Skills")
+        gr.Textbox(label="Location"),
+        gr.Textbox(label="Email"),
+        gr.Textbox(label="LinkedIn URL"),
+        gr.Textbox(label="GitHub URL"),
+        gr.Textbox(label="Portfolio URL"),
+        gr.Textbox(label="Professional Summary", lines=4, placeholder="Write 2–4 lines about your profile..."),
+        gr.Textbox(label="Technical Skills (Languages, Libraries, Tools)", lines=3),
+        gr.Textbox(label="Soft Skills", lines=2),
+        gr.Textbox(label="Education", lines=3),
+        gr.Textbox(label="Languages Known", lines=2),
+        gr.Textbox(label="Experience (Job Title, Company, Dates, Bullets)", lines=4),
+        gr.Textbox(label="Key Projects", lines=4),
+        gr.Textbox(label="Certifications", lines=3)
     ],
     outputs=gr.File(label="Download Your CV (.docx)"),
     title="Free CV Builder (Download as Word)",
-    description="This tool creates a structured Word CV using free AI and lets you download it."
+    description="🧠 Build your CV using a clean form and download it as a Word document — powered by open-source AI, no payment required!"
 )
 
 interface.launch()