Update app.py

app.py

@@ -4,7 +4,7 @@ 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_KEY = "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"  # Nên dùng biến môi trường trong thực tế
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
 
 HEADERS = {
@@ -23,28 +23,30 @@ 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:
+# === REFLECTIVE SCRIPT GENERATION ===
+def generate_reflective_script(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.
+You are a reflective and thoughtful writer.
 
-Create a short and engaging podcast script in **{language}** on the topic: "{topic}".
+Write a short, emotional, and thought-provoking script in **{language}** about 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.
+The tone should be reflective, poetic, and emotional — suitable for a short narration video.
+Avoid giving facts or academic details. Instead, focus on emotions, metaphors, inner voice, and human feelings.
+Use beautiful language, natural rhythm, and pauses.
 
-Only write their dialogue and a few narration lines if needed. Do NOT use Host 1/2 or generic names.
+The output should be between 200 to 400 words, formatted as a poetic monologue or a subtle inner conversation.
+
+Do NOT include sound effects, scene descriptions, or storytelling instructions.
+Just output the reflective script.
 """
 
     data = {
         "model": model,
         "messages": [
-            {"role": "system", "content": "You are a creative and age-appropriate educational podcast writer."},
+            {"role": "system", "content": "You are a poetic, emotional writer that creates powerful scripts for videos."},
             {"role": "user", "content": prompt.strip()}
         ],
-        "temperature": 0.75,
+        "temperature": 0.8,
         "max_tokens": 1024
     }
 
@@ -56,62 +58,46 @@ Only write their dialogue and a few narration lines if needed. Do NOT use Host 1
 
 # === 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")
+    paragraphs = 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"
+    for para in paragraphs:
+        para = para.strip()
+        if para:
+            formatted += f"<div class='narration'>{para}</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);
+.narration {
+    background: linear-gradient(to right, #434343, #000000);
+    color: #f5f5f5;
+    padding: 20px;
+    margin: 18px 0;
+    border-radius: 12px;
+    font-size: 18px;
+    line-height: 1.8;
+    box-shadow: 0 4px 12px rgba(0,0,0,0.5);
+    font-family: 'Georgia', serif;
 }
 """
 
 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 "<b style='color:red;'>⚠️ Restricted content. Please enter a safe, inspirational topic.</b>"
+    raw_script = generate_reflective_script(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.")
+    gr.Markdown("## 🌌 Reflective Video Script Generator")
+    gr.Markdown("Enter a thoughtful topic and choose your language to generate a poetic and emotional script for video narration.")
 
-    topic_input = gr.Textbox(label="Enter educational topic")
+    topic_input = gr.Textbox(label="Enter a reflective topic (e.g., loneliness, hope, youth...)")
     lang_input = gr.Dropdown(
         choices=["English", "Vietnamese", "Spanish", "French", "Japanese"],
         value="English",
         label="Select output language"
     )
-    generate_btn = gr.Button("🎧 Generate Podcast")
+    generate_btn = gr.Button("📝 Generate Script")
     output_box = gr.HTML()
 
     generate_btn.click(fn=generate, inputs=[topic_input, lang_input], outputs=output_box)