Update app.py

app.py

@@ -1,35 +1,53 @@
 from transformers import pipeline
+from docx import Document
 import gradio as gr
 
 generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b", device=-1)
 
-def generate_cv(name, title, education, experience, skills):
+def generate_cv_docx(name, title, education, experience, skills):
     prompt = f"""
-Create a clean, professional CV in plain text format:
+You are a CV assistant. Generate a clean, structured CV in this format:
 
-Name: {name}
-Career Title: {title}
+Full Name: {name}
+Title: {title}
 Education: {education}
 Experience: {experience}
 Skills: {skills}
-
-CV:
 """
-    output = generator(prompt, max_length=512, temperature=0.7, do_sample=True)[0]['generated_text']
-    return output[len(prompt):]  # remove prompt part
+    result = generator(prompt, max_length=512, do_sample=True, temperature=0.7)[0]['generated_text']
+    
+    # Generate Word doc
+    doc = Document()
+    doc.add_heading(name, level=1)
+    doc.add_paragraph(title, style='Intense Quote')
+
+    doc.add_heading("Education", level=2)
+    doc.add_paragraph(education)
+
+    doc.add_heading("Experience", level=2)
+    doc.add_paragraph(experience)
+
+    doc.add_heading("Skills", level=2)
+    doc.add_paragraph(skills)
+
+    # Save document
+    filepath = "generated_cv.docx"
+    doc.save(filepath)
+
+    return filepath
 
-demo = gr.Interface(
-    fn=generate_cv,
+interface = gr.Interface(
+    fn=generate_cv_docx,
     inputs=[
         gr.Textbox(label="Full Name"),
-        gr.Textbox(label="Job Title / Career Goal"),
+        gr.Textbox(label="Job Title"),
         gr.Textbox(label="Education"),
         gr.Textbox(label="Work Experience"),
         gr.Textbox(label="Skills")
     ],
-    outputs="text",
-    title="Free CV Builder (No API)",
-    description="Uses Falcon-RW-1B — an open-source AI model that runs on CPU for free."
+    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."
 )
 
-demo.launch()
+interface.launch()