Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import random | |
| import requests | |
| import os | |
| # ==== CONFIG ==== | |
| GROQ_API_KEY = os.getenv("GROQ_API_KEY") # Set this in Hugging Face secrets | |
| GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions" | |
| GROQ_MODEL = "llama3-8b-8192" # Or use llama3-70b-8192 | |
| # === Dummy topic generator === | |
| def generate_trending_topics(niche): | |
| sample_topics = { | |
| "finance": ["How to Save $10,000 in 6 Months", "Why Rich People Avoid Taxes", "The Psychology of Spending", "Is Real Estate Still Worth It?", "Side Hustles That Made Millionaires"], | |
| "fitness": ["5-Minute Fat Burning Workout", "What Happens If You Never Skip Leg Day", "Foods That Kill Your Gains", "Morning Routine of Athletes", "How to Build Abs Fast"], | |
| "horror": ["Real Haunted Places in America", "The Unsolved Mystery of Room 1046", "Creepiest Urban Legends by State", "What Happens in Sleep Paralysis", "Demonic Possessions or Mental Illness?"], | |
| } | |
| return sample_topics.get(niche.lower(), [f"Viral Topic #{i+1} for {niche}" for i in range(5)]) | |
| # === Script generator using Groq === | |
| def generate_script(title): | |
| import json | |
| headers = { | |
| "Authorization": f"Bearer {GROQ_API_KEY}", | |
| "Content-Type": "application/json" | |
| } | |
| prompt = f""" | |
| You are a viral YouTube script writer. Generate a high-engagement short-form YouTube script (around 150–170 words) based on this video idea: "{title}". | |
| Use simple, modern English. Include a strong hook in the first line to grab attention. Avoid long intros or boring facts. Write in a dramatic, emotional, or suspenseful tone depending on the topic. No outro needed. | |
| """ | |
| payload = { | |
| "model": GROQ_MODEL, | |
| "messages": [ | |
| {"role": "system", "content": "You are a helpful assistant."}, | |
| {"role": "user", "content": prompt.strip()} | |
| ], | |
| "temperature": 0.7 | |
| } | |
| try: | |
| response = requests.post(GROQ_API_URL, headers=headers, json=payload) | |
| response.raise_for_status() | |
| return response.json()['choices'][0]['message']['content'] | |
| except Exception as e: | |
| return f"Error generating script: {e}" | |
| # === Full pipeline === | |
| def generate_output(niche): | |
| topics = generate_trending_topics(niche) | |
| results = [] | |
| for topic in topics: | |
| script = generate_script(topic) | |
| results.append(f"📌 **{topic}**\n\n{script}") | |
| return results | |
| # === Gradio Interface === | |
| with gr.Blocks() as demo: | |
| gr.Markdown("# 🎬 YouTube Topic & Script Generator\nEnter a niche to generate trending topics and scripts.") | |
| niche_input = gr.Textbox(label="Enter Your Niche (e.g. Finance, Fitness, Horror)") | |
| generate_btn = gr.Button("Generate") | |
| output_boxes = [gr.Markdown() for _ in range(5)] # Predefine 5 outputs | |
| def on_click(niche): | |
| scripts = generate_output(niche) | |
| return scripts | |
| generate_btn.click(on_click, inputs=[niche_input], outputs=output_boxes) | |
| # Run the app | |
| demo.launch() | |