Update app.py

app.py

@@ -3,33 +3,35 @@
 from transformers import pipeline
 import gradio as gr
 
-# Load FLAN-T5-LARGE pipeline for better outputs
+# Load FLAN-T5-LARGE pipeline
 generator = pipeline("text2text-generation", model="google/flan-t5-large")
 
 def generate_brand_identity(startup_idea):
+    # Make each prompt practical and targeted
+
     # Brand Names
-    prompt_name = f"Suggest three unique and creative brand names for a startup about {startup_idea}. Separate them with commas."
-    brand_names = generator(prompt_name, max_length=80, num_return_sequences=1)[0]["generated_text"]
+    prompt_name = f"Generate 3 unique and creative brand names for a startup that offers: {startup_idea}. The names should be catchy and relevant. Separate with commas."
+    brand_names = generator(prompt_name, max_length=60, num_return_sequences=1)[0]["generated_text"]
 
     # Tagline
-    tagline_prompt = f"Write a catchy, short tagline for a startup about {startup_idea}."
+    tagline_prompt = f"Write a 1-line catchy tagline for a startup that offers: {startup_idea}. Focus on value, not just flair."
     tagline = generator(tagline_prompt, max_length=30, num_return_sequences=1)[0]['generated_text']
 
     # Brand Personality Description
-    personality_prompt = f"Describe in one sentence the brand personality for a startup about {startup_idea}."
+    personality_prompt = f"Describe the brand personality for a startup that offers: {startup_idea}. Use descriptive language."
     personality = generator(personality_prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
 
     # Suggested Brand Colours
-    colours_prompt = f"Suggest primary and secondary brand colours with hex codes for a startup about {startup_idea}."
+    colours_prompt = f"Suggest two brand colour themes (primary and secondary) with HEX codes for a startup that offers: {startup_idea}." 
     colours = generator(colours_prompt, max_length=40, num_return_sequences=1)[0]['generated_text']
 
     # Target Audience Description
-    audience_prompt = f"Describe the ideal target audience for a startup about {startup_idea}."
+    audience_prompt = f"Identify the most likely target audience for a startup that offers: {startup_idea}. Include age, profession, or lifestyle."
     audience = generator(audience_prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
 
     # Slogan Options
-    slogan_prompt = f"Suggest a short, catchy slogan for a startup about {startup_idea}."
-    slogan = generator(slogan_prompt, max_length=20, num_return_sequences=1)[0]['generated_text']
+    slogan_prompt = f"Generate 2 short, memorable slogans for a startup that offers: {startup_idea}. Keep them unique and practical."
+    slogan = generator(slogan_prompt, max_length=40, num_return_sequences=1)[0]['generated_text']
 
     return (
         brand_names.strip(),
@@ -40,7 +42,7 @@ def generate_brand_identity(startup_idea):
         slogan.strip()
     )
 
-# Gradio interface with updated outputs and CSS styling
+# Gradio interface with improved text clarity and appearance
 css = """
 h1 {
     text-align: center;
@@ -73,7 +75,7 @@ iface = gr.Interface(
         gr.Textbox(label="Slogan Options")
     ],
     title="🌟 Brand Identity Generator",
-    description="Generate creative brand names, taglines, brand personality descriptions, colour palettes, target audiences, and slogans for your startup idea using Gen AI (FLAN-T5-LARGE).",
+    description="Generate practical and creative brand names, taglines, brand personality descriptions, colour palettes, target audiences, and slogans for your startup idea using FLAN-T5-LARGE.",
     css=css,
     theme="soft"
 )