Update app.py

app.py

@@ -3,21 +3,15 @@ from transformers import pipeline
 
 generator = pipeline("text-generation", model="gpt2")
 
-def improve_description(description):
-    prompt = f"Improve this Google Business Profile description. Make it clear, professional, and SEO-friendly:\n\n{description}\n\nImproved version:"
-    result = generator(prompt, max_length=120, num_return_sequences=1)[0]['generated_text']
+def generate_description(business_name):
+    prompt = f"Write a professional and SEO-friendly Google My Business description for a business named '{business_name}'. Highlight its quality, trust, and services."
+    result = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
     return result
 
-with gr.Blocks() as demo:
-    gr.Markdown("## Webpy - GMB Optimizer")
-    gr.Markdown("Paste your Google Business Profile URL (optional):")
-    url = gr.Textbox(label="Google My Business URL (optional)")
-    
-    gr.Markdown("Then paste your current description below:")
-    input_desc = gr.Textbox(lines=4, label="Current Description", placeholder="e.g. We create affordable websites for small businesses in Colombia.")
-    output = gr.Textbox(label="Improved Description")
-
-    submit = gr.Button("Enhance")
-    submit.click(fn=improve_description, inputs=input_desc, outputs=output)
-
-demo.launch()
+gr.Interface(
+    fn=generate_description,
+    inputs=gr.Textbox(label="Business Name", placeholder="e.g. Webpy"),
+    outputs="text",
+    title="Google My Business Description Generator",
+    description="Enter your business name and get a professional, AI-enhanced description for your GMB profile.",
+).launch()