Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import re
 
 # === API SETUP ===
-GROQ_API_KEY = "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"
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Bạn nên dùng biến môi trường
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
 
 HEADERS = {
@@ -12,7 +12,7 @@ HEADERS = {
     "Content-Type": "application/json"
 }
 
-# === BLOCK INAPPROPRIATE CONTENT ===
+# === BLOCKED TOPICS ===
 BLOCKED_KEYWORDS = [
     "sex", "drugs", "violence", "suicide", "death", "kill", "murder", "gun",
     "abuse", "politics", "terror", "crime", "war", "religion", "racism",
@@ -24,18 +24,18 @@ def is_safe_topic(topic: str) -> bool:
     return not any(bad_word in topic_lower for bad_word in BLOCKED_KEYWORDS)
 
 # === PODCAST GENERATION ===
-def generate_educational_podcast(topic: str, model: str = "llama3-70b-8192") -> str:
+def generate_educational_podcast(topic: str, language: str, model: str = "llama3-70b-8192") -> str:
     prompt = f"""
-You are a scriptwriter for an educational podcast aimed at school students in grades 8 to 12.
+You are a scriptwriter for an educational podcast for students in grades 8 to 12.
 
-Create a short and engaging podcast script 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. 
+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 any generic names.
+
+Only write their dialogue and a few narration lines if needed. Do NOT use Host 1/2 or generic names.
 """
 
     data = {
@@ -49,67 +49,35 @@ Only write their dialogue and a few narration lines if needed. Do NOT use Host 1
     }
 
     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}"
 
-# === FORMAT SCRIPT CLEANLY FOR DISPLAY ===
+# === SCRIPT FORMATTING ===
 def format_script(script: str) -> str:
-    # Remove markdown bolding like **text**
     script = re.sub(r'\*\*(.*?)\*\*', r'\1', script)
-
-    # Remove music and direction lines
     script = re.sub(r'(?i)^.*(music|sound effect).*$','', script, flags=re.MULTILINE)
-
-    # Normalize host names to Ali and Talha
-    script = script.replace("Host 1:", "Ali:")
-    script = script.replace("Host 2:", "Talha:")
+    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
-        # Only keep lines that are spoken by Ali or Talha
         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
 
-# === FINAL FUNCTION ===
-def generate(topic):
-    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)
-    return format_script(raw_script)
-
-# === DARK GLOSSY CSS ===
+# === GRADIO UI ===
 custom_css = """
-# body {
-#     background-color: darkslateblue !important;
-#     color: #ffffff;
-# }
-
-# .gradio-container {
-#     background-color: darkslateblue !important;
-# }
-
-# textarea, input, .gr-button {
-#     background-color: darkslategray !important;
-#     color: #ffffff !important;
-#     border: 1px solid #aaa !important;
-# }
-
 .host1, .host2 {
     border-radius: 16px;
     padding: 14px;
@@ -119,32 +87,33 @@ custom_css = """
     line-height: 1.6;
     color: #ffffff;
 }
-
 .host1 {
-    background: linear-gradient(135deg, #4b6cb7, #182848);  /* Dark blue gradient */
+    background: linear-gradient(135deg, #4b6cb7, #182848);
 }
-
 .host2 {
-    background: linear-gradient(135deg, #2c3e50, #4ca1af);  /* Slate gray gradient */
-}
-
-.narration {
-    font-style: italic;
-    color: #cccccc;
-    margin: 10px 0;
-    font-size: 15px;
+    background: linear-gradient(135deg, #2c3e50, #4ca1af);
 }
 """
 
-# === GRADIO INTERFACE ===
+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", elem_id="title")
-    gr.Markdown("Enter an educational topic, and enjoy a friendly podcast conversation between Ali and Talha.")
+    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, outputs=output_box)
+    generate_btn.click(fn=generate, inputs=[topic_input, lang_input], outputs=output_box)
 
 demo.launch()