Update app.py

app.py

@@ -12,7 +12,7 @@ HEADERS = {
     "Content-Type": "application/json"
 }
 
-# === FILTER SETUP ===
+# === BLOCK INAPPROPRIATE CONTENT ===
 BLOCKED_KEYWORDS = [
     "sex", "drugs", "violence", "suicide", "death", "kill", "murder", "gun",
     "abuse", "politics", "terror", "crime", "war", "religion", "racism",
@@ -30,17 +30,12 @@ You are a scriptwriter for an educational podcast aimed at school students in gr
 
 Create a short and engaging podcast script on the topic: "{topic}".
 
-Two hosts, Host 1 and Host 2, should have a back-and-forth conversation about the 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 complex jargon and make it suitable for students aged 13–18.
 
-Use the format:
-Host 1: ...
-Host 2: ...
-...
-
-Make sure the conversation stays focused, clear, and ends in under 800 words.
-Avoid going off-topic or repeating points.
+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.
 """
 
     data = {
@@ -62,13 +57,17 @@ Avoid going off-topic or repeating points.
 
 # === FORMAT SCRIPT CLEANLY FOR DISPLAY ===
 def format_script(script: str) -> str:
-    import re
-
     # Remove markdown bolding like **text**
     script = re.sub(r'\*\*(.*?)\*\*', r'\1', script)
 
-    # Remove intro/outro music lines or similar direction text
-    script = re.sub(r'(?i)^.*music.*$', '', script, flags=re.MULTILINE)
+    # 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 = re.sub(r'\b[Aa]lex:', 'Ali:', script)
+    script = re.sub(r'\b[Mm]aya:', 'Talha:', script)
 
     lines = script.strip().split("\n")
     formatted = ""
@@ -77,46 +76,62 @@ def format_script(script: str) -> str:
         line = line.strip()
         if not line:
             continue
-        if line.startswith("Host 1:"):
-            content = line.replace("Host 1:", "").strip()
-            formatted += f"<div class='host1'> <b>Host 1:</b> {content}</div>\n"
-        elif line.startswith("Host 2:"):
-            content = line.replace("Host 2:", "").strip()
-            formatted += f"<div class='host2'> <b>Host 2:</b> {content}</div>\n"
+        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"
         else:
             formatted += f"<div class='narration'>{line}</div>\n"
 
     return formatted
 
-# === FINAL APP FUNCTION ===
+# === 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)
 
-# === GLOSSY CSS STYLING ===
+# === DARK GLOSSY CSS ===
 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;
     margin: 16px 0;
-    box-shadow: 0 4px 10px rgba(0,0,0,0.1);
+    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, #d0e8ff, #e6f3ff);
-# }
+.host1 {
+    background: linear-gradient(135deg, #4b6cb7, #182848);  /* Dark blue gradient */
+}
 
-# .host2 {
-#     background: linear-gradient(135deg, #ffe6f0, #fce8ff);
-# }
+.host2 {
+    background: linear-gradient(135deg, #2c3e50, #4ca1af);  /* Slate gray gradient */
+}
 
 .narration {
     font-style: italic;
-    color: #555;
+    color: #cccccc;
     margin: 10px 0;
     font-size: 15px;
 }
@@ -124,14 +139,13 @@ custom_css = """
 
 # === GRADIO INTERFACE ===
 with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("## 🎓 Educational Podcast Generator for Students")
-    gr.Markdown("Enter an educational topic, and get a podcast-style conversation between two friendly hosts!")
+    gr.Markdown("## 🎙️ Educational Podcast Generator with Ali & Talha", elem_id="title")
+    gr.Markdown("Enter an educational topic, and enjoy a friendly podcast conversation between Ali and Talha.")
 
     topic_input = gr.Textbox(label="Enter educational topic")
-    generate_btn = gr.Button("🎙️ Generate Podcast")
+    generate_btn = gr.Button("🎧 Generate Podcast")
     output_box = gr.HTML()
 
     generate_btn.click(fn=generate, inputs=topic_input, outputs=output_box)
 
-# === LAUNCH ===
 demo.launch()