import gradio as gr
import random

# SaaS Idea Categories
categories = [
    "AI-Powered Marketing", "Health & Fitness", "E-commerce Automation", "Remote Work Tools", "Finance & Investing",
    "EdTech & Learning", "Legal & Compliance", "HR & Recruitment", "Cybersecurity", "SaaS for Small Businesses"
]

# AI-Based SaaS Idea Generator
def generate_saas_idea(category, custom_niche, num_ideas):
    ideas = {
        "AI-Powered Marketing": [
            "An AI tool that generates personalized email marketing campaigns.",
            "A SaaS that automates ad targeting and optimization using AI."
        ],
        "Health & Fitness": [
            "A virtual AI coach that tailors workouts based on real-time feedback.",
            "A nutrition tracking SaaS that generates meal plans using AI."
        ],
        "E-commerce Automation": [
            "AI-based product recommendation engine for online stores.",
            "A tool that automates social media content creation for e-commerce brands."
        ],
        "Remote Work Tools": [
            "An AI-powered meeting summarizer with action item tracking.",
            "A virtual co-working space with AI productivity tracking."
        ],
        "Finance & Investing": [
            "AI-driven stock market trend predictor for retail investors.",
            "A budgeting SaaS that automatically categorizes expenses using AI."
        ],
        "EdTech & Learning": [
            "An AI tutor that provides instant feedback on coding exercises.",
            "A language learning SaaS that adapts to user mistakes dynamically."
        ],
        "Legal & Compliance": [
            "An AI-based contract analyzer for small businesses.",
            "A GDPR compliance monitoring tool for websites."
        ],
        "HR & Recruitment": [
            "An AI-driven resume screening tool that ranks candidates.",
            "A SaaS for automated job description optimization."
        ],
        "Cybersecurity": [
            "A SaaS that detects phishing emails using AI.",
            "An AI-powered password strength checker and breach alert system."
        ],
        "SaaS for Small Businesses": [
            "An AI-powered invoice and expense tracking system.",
            "A chatbot that handles customer support queries automatically."
        ]
    }
    
    selected_ideas = random.sample(ideas.get(category, ["No ideas available."]), min(num_ideas, len(ideas.get(category, []))))
    
    if custom_niche:
        selected_ideas.append(f"An AI-powered SaaS solution for {custom_niche}.")
    
    return "\n\n".join(selected_ideas)

# Gradio UI
iface = gr.Interface(
    fn=generate_saas_idea,
    inputs=[
        gr.Dropdown(categories, label="Select a Category"),
        gr.Textbox(label="Enter Custom Niche (Optional)"),
        gr.Slider(minimum=1, maximum=5, step=1, value=2, label="Number of Ideas")
    ],
    outputs="text",
    title="LaunchPad AI - SaaS Idea Generator",
    description="Get AI-generated SaaS ideas! Choose a category, enter a niche (optional), and select the number of ideas to generate.",
)

# Launch the App
iface.launch()