Spaces:
Build error
Build error
| import os | |
| import requests | |
| import gradio as gr | |
| import re | |
| # === API SETUP === | |
| GROQ_API_KEY = "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" # Bạn nên dùng biến môi trường | |
| GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions" | |
| HEADERS = { | |
| "Authorization": f"Bearer {GROQ_API_KEY}", | |
| "Content-Type": "application/json" | |
| } | |
| # === BLOCKED TOPICS === | |
| BLOCKED_KEYWORDS = [ | |
| "sex", "drugs", "violence", "suicide", "death", "kill", "murder", "gun", | |
| "abuse", "politics", "terror", "crime", "war", "religion", "racism", | |
| "nude", "adult", "nsfw" | |
| ] | |
| def is_safe_topic(topic: str) -> bool: | |
| topic_lower = topic.lower() | |
| return not any(bad_word in topic_lower for bad_word in BLOCKED_KEYWORDS) | |
| # === PODCAST GENERATION === | |
| def generate_educational_podcast(topic: str, language: str, model: str = "llama3-70b-8192") -> str: | |
| prompt = f""" | |
| You are a scriptwriter for an educational podcast for students in grades 8 to 12. | |
| Create a short and engaging podcast script in **{language}** on the topic: "{topic}". | |
| The two hosts are named Ali and Talha. Ali starts the conversation. | |
| They should have a back-and-forth conversation about the topic. | |
| The tone should be friendly, easy to understand, and informative. | |
| Avoid music or sound effects. Keep the total length under 800 words. | |
| Only write their dialogue and a few narration lines if needed. Do NOT use Host 1/2 or generic names. | |
| """ | |
| data = { | |
| "model": model, | |
| "messages": [ | |
| {"role": "system", "content": "You are a creative and age-appropriate educational podcast writer."}, | |
| {"role": "user", "content": prompt.strip()} | |
| ], | |
| "temperature": 0.75, | |
| "max_tokens": 1024 | |
| } | |
| response = requests.post(GROQ_API_URL, headers=HEADERS, json=data) | |
| if response.status_code == 200: | |
| return response.json()["choices"][0]["message"]["content"].strip() | |
| else: | |
| return f"<b>Error {response.status_code}</b>: {response.text}" | |
| # === SCRIPT FORMATTING === | |
| def format_script(script: str) -> str: | |
| script = re.sub(r'\*\*(.*?)\*\*', r'\1', script) | |
| script = re.sub(r'(?i)^.*(music|sound effect).*$','', script, flags=re.MULTILINE) | |
| script = script.replace("Host 1:", "Ali:").replace("Host 2:", "Talha:") | |
| script = re.sub(r'\b[Aa]lex:', 'Ali:', script) | |
| script = re.sub(r'\b[Mm]aya:', 'Talha:', script) | |
| lines = script.strip().split("\n") | |
| formatted = "" | |
| for line in lines: | |
| line = line.strip() | |
| if not line: | |
| continue | |
| if line.startswith("Ali:"): | |
| content = line.replace("Ali:", "").strip() | |
| formatted += f"<div class='host1'>🎙️ <b>Ali:</b> {content}</div>\n" | |
| elif line.startswith("Talha:"): | |
| content = line.replace("Talha:", "").strip() | |
| formatted += f"<div class='host2'>🎧 <b>Talha:</b> {content}</div>\n" | |
| return formatted | |
| # === GRADIO UI === | |
| custom_css = """ | |
| .host1, .host2 { | |
| border-radius: 16px; | |
| padding: 14px; | |
| margin: 16px 0; | |
| box-shadow: 0 4px 10px rgba(255,255,255,0.2); | |
| font-size: 16px; | |
| line-height: 1.6; | |
| color: #ffffff; | |
| } | |
| .host1 { | |
| background: linear-gradient(135deg, #4b6cb7, #182848); | |
| } | |
| .host2 { | |
| background: linear-gradient(135deg, #2c3e50, #4ca1af); | |
| } | |
| """ | |
| def generate(topic, language): | |
| if not is_safe_topic(topic): | |
| return "<b style='color:red;'>⚠️ Restricted content. Please enter an appropriate educational topic.</b>" | |
| raw_script = generate_educational_podcast(topic, language) | |
| return format_script(raw_script) | |
| with gr.Blocks(css=custom_css) as demo: | |
| gr.Markdown("## 🎙️ Educational Podcast Generator for Students") | |
| gr.Markdown("Enter a topic and choose your language to generate a friendly podcast between Ali and Talha.") | |
| topic_input = gr.Textbox(label="Enter educational topic") | |
| lang_input = gr.Dropdown( | |
| choices=["English", "Vietnamese", "Spanish", "French", "Japanese"], | |
| value="English", | |
| label="Select output language" | |
| ) | |
| generate_btn = gr.Button("🎧 Generate Podcast") | |
| output_box = gr.HTML() | |
| generate_btn.click(fn=generate, inputs=[topic_input, lang_input], outputs=output_box) | |
| demo.launch() | |