Update app.py

app.py

@@ -1,24 +1,72 @@
 import gradio as gr
 from transformers import pipeline
+from fpdf import FPDF
+import tempfile
+import os
 
-# Load the model pipeline
+# Load the text-generation pipeline using GPT-2
 generator = pipeline("text-generation", model="gpt2")
 
-
-# Define the function to generate proposal text
+# Format the output as a business-style proposal
 def generate_proposal(client_description):
-    prompt = f"Generate a professional project proposal based on the following client request:\n\n{client_description}\n\nProposal:"
-    result = generator(prompt, max_length=512, do_sample=True, temperature=0.7)
-    return result[0]['generated_text']
+    # Generate raw text
+    raw = generator(f"{client_description}", max_length=300, do_sample=True, temperature=0.7)[0]['generated_text']
+
+    # Extract only the generated part (remove the prompt from output)
+    generated_only = raw.replace(client_description, "").strip()
+
+    # Build formatted text
+    formatted = f"""
+    PROJECT PROPOSAL
+
+    Client Requirement:
+    {client_description}
+
+    Proposed Solution:
+    {generated_only}
+
+    Sincerely,
+    AI Proposal Generator
+    """
+
+    # Create a PDF
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.set_font("Arial", size=12)
+    for line in formatted.split("\n"):
+        pdf.multi_cell(0, 10, line)
+
+    # Save to a temporary file
+    pdf_file = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    pdf.output(pdf_file.name)
+
+    # Email suggestion
+    email_body = f"""
+Subject: Business Proposal for Your Project
+
+Dear Client,
+
+Please find attached a professional proposal based on your provided requirements. We're confident this solution meets your needs and are open to feedback or further discussion.
+
+Best regards,  
+Your Name  
+AI Proposal Generator  
+"""
+
+    return formatted, email_body, pdf_file.name
 
-# Build the Gradio interface
+# Gradio UI
 iface = gr.Interface(
     fn=generate_proposal,
     inputs=gr.Textbox(label="Client Requirement", placeholder="Describe the project or client needs here...", lines=5),
-    outputs=gr.Textbox(label="Generated Proposal", lines=15),
+    outputs=[
+        gr.Textbox(label="Formatted Proposal", lines=15),
+        gr.Textbox(label="Email Template", lines=10),
+        gr.File(label="Download Proposal as PDF")
+    ],
     title="AI Proposal Generator",
-    description="This app uses Mistral-7B to generate business proposals based on client input."
+    description="Enter a client request and receive a professional proposal, an email draft, and a downloadable PDF."
 )
 
-# Launch the app
 iface.launch()