Update app.py

app.py

@@ -1,82 +1,79 @@
-# app.py
-
+import streamlit as st
 from transformers import pipeline
-import gradio as gr
-
-# Load FLAN-T5-LARGE pipeline
-generator = pipeline("text-generation", model="gpt2-large")
-
-def generate_brand_identity(startup_idea):
-    # Brand Names
-    prompt_name = f"Give three short, creative, and unique brand name ideas for a startup that is {startup_idea}. Avoid repetition and ensure they are realistic for a business."
-    brand_names = generator(prompt_name, max_length=60, num_return_sequences=1)[0]["generated_text"]
-
-    # Tagline
-    tagline_prompt = f"Generate a catchy and original tagline for a startup focused on {startup_idea}. Keep it under 10 words."
-    tagline = generator(tagline_prompt, max_length=30, num_return_sequences=1)[0]['generated_text']
+import torch
 
-    # Brand Personality
-    personality_prompt = f"Describe the brand personality of a startup doing {startup_idea} in one powerful sentence. Use descriptive adjectives, not vague words."
-    personality = generator(personality_prompt, max_length=40, num_return_sequences=1)[0]['generated_text']
+# Set Streamlit page configuration
+st.set_page_config(page_title="Brand Identity Generator", layout="wide")
 
-    # Brand Colours
-    colours_prompt = f"Suggest a primary and secondary brand colour with HEX codes for a startup about {startup_idea}. Format: Color Name - HEX"
-    colours = generator(colours_prompt, max_length=40, num_return_sequences=1)[0]['generated_text']
+# Load FLAN-T5 model pipeline
+@st.cache_resource
+def load_model():
+    return pipeline("text2text-generation", model="google/flan-t5-large")
 
-    # Target Audience
-    audience_prompt = f"Describe the ideal target audience (age, profession, need) for a startup doing {startup_idea}."
-    audience = generator(audience_prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
+generator = load_model()
 
-    # Slogan
-    slogan_prompt = f"Suggest a memorable and short slogan (max 7 words) for a startup about {startup_idea}. Make it inspirational or witty."
-    slogan = generator(slogan_prompt, max_length=20, num_return_sequences=1)[0]['generated_text']
+# Custom Styling (simulate CSS)
+st.markdown(
+    """
+    <style>
+    body {
+        background-color: #f4f8fc;
+        font-family: 'Segoe UI', sans-serif;
+        color: #003366;
+    }
+    .stTextInput > div > div > input {
+        background-color: #e6f0fa;
+        color: #003366;
+        font-size: 16px;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
 
-    return (
-        brand_names.strip(),
-        tagline.strip(),
-        personality.strip(),
-        colours.strip(),
-        audience.strip(),
-        slogan.strip()
-    )
+# Page Title and Description
+st.markdown(
+    "<h1 style='text-align: center; color: #4A90E2;'>🌟 Brand Identity Generator</h1>",
+    unsafe_allow_html=True
+)
 
-# Gradio interface with improved text clarity and appearance
-css = """
-h1 {
-    text-align: center;
-    color: #4A90E2;
-    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
-}
+st.markdown(
+    """
+    Generate **brand names**, **taglines**, **personality descriptions**, **colour palettes**, **target audiences**, and **slogans**
+    for your startup idea using the powerful **FLAN-T5-LARGE** model.
+    """
+)
 
-textarea {
-    background-color: #e6f0fa !important;
-    color: #003366 !important;
-    font-size: 16px !important;
-}
+# Form Input for Startup Idea
+with st.form("input_form"):
+    st.subheader("📌 Describe Your Startup")
+    startup_idea = st.text_input("Startup Idea", placeholder="e.g. an eco-friendly fish delivery platform for urban areas")
+    submitted = st.form_submit_button("Generate")
 
-footer {
-    text-align: center;
-    padding: 10px;
-    color: #888;
-}
-"""
+# On form submission
+if submitted:
+    if not startup_idea.strip():
+        st.warning("Please enter a valid startup idea.")
+    else:
+        st.markdown("---")
+        st.subheader("🔍 Generated Brand Identity")
 
-iface = gr.Interface(
-    fn=generate_brand_identity,
-    inputs=gr.Textbox(lines=2, placeholder="Enter your startup idea here..."),
-    outputs=[
-        gr.Textbox(label="Brand Name Suggestions"),
-        gr.Textbox(label="Tagline"),
-        gr.Textbox(label="Brand Personality Description"),
-        gr.Textbox(label="Suggested Brand Colours"),
-        gr.Textbox(label="Target Audience"),
-        gr.Textbox(label="Slogan Options")
-    ],
-    title="🌟 Brand Identity Generator",
-    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"
-)
+        prompts = {
+            "Brand Name Suggestions": f"Suggest 5 original, non-generic brand names for a startup based on this idea: {startup_idea}. Make sure they are short, unique, and not repetitive.",
+            "Tagline": f"Write a compelling and creative tagline for this startup. Avoid repeating the brand name. Idea: {startup_idea}",
+            "Brand Personality Description": f"Describe in one sentence the personality of a brand that is built around the following idea: {startup_idea}. Make it lively and creative.",
+            "Suggested Brand Colours": f"Suggest two primary and two secondary brand colours (with hex codes) that reflect the startup idea: {startup_idea}.",
+            "Target Audience": f"Describe in 1–2 lines the ideal target audience for this startup idea: {startup_idea}.",
+            "Slogan Options": f"Suggest three short and catchy slogans that capture the spirit of this startup: {startup_idea}. Avoid repeating phrases."
+        }
 
-if __name__ == "__main__":
-    iface.launch()
+        # Generate and display each section
+        for section, prompt in prompts.items():
+            try:
+                with st.spinner(f"Generating {section}..."):
+                    result = generator(prompt, max_length=80, num_return_sequences=1)[0]["generated_text"]
+                    st.markdown(f"### {section}")
+                    st.success(result)
+            except Exception as e:
+                st.markdown(f"### {section}")
+                st.error("Error generating content. Please check your connection or try again later.")