Update app.py

app.py

@@ -38,20 +38,23 @@ def generate_script(user_prompt: str, model_id: str, token: str, duration: int):
 
         system_prompt = (
             f"You are an expert radio imaging producer specializing in sound design and music. "
-            f"Based on the user's concept and the selected duration of {duration} seconds, craft a concise, engaging promo script. "
-            f"Ensure the script fits within the time limit and suggest a matching music style that complements the theme."
+            f"Based on the user's concept and the selected duration of {duration} seconds, generate: "
+            f"1. A concise voice-over script. "
+            f"2. Suggestions for sound design. "
+            f"3. Music styles or track recommendations."
         )
 
-        combined_prompt = f"{system_prompt}\nUser concept: {user_prompt}\nRefined script and music suggestion:"
-        result = llama_pipeline(combined_prompt, max_new_tokens=200, do_sample=True, temperature=0.9)
+        combined_prompt = f"{system_prompt}\nUser concept: {user_prompt}\nOutput:"
+        result = llama_pipeline(combined_prompt, max_new_tokens=300, do_sample=True, temperature=0.8)
 
-        generated_text = result[0]["generated_text"].split("Refined script and music suggestion:")[-1].strip()
-        if "Music Suggestion:" in generated_text:
-            script, music_suggestion = generated_text.split("Music Suggestion:")
-            return script.strip(), music_suggestion.strip()
-        return generated_text, "No specific music suggestion found."
+        generated_text = result[0]["generated_text"].split("Output:")[-1].strip()
+        if "Suggestions:" in generated_text and "Music Recommendations:" in generated_text:
+            script, rest = generated_text.split("Suggestions:", 1)
+            sound_design, music_suggestion = rest.split("Music Recommendations:", 1)
+            return script.strip(), sound_design.strip(), music_suggestion.strip()
+        return "Error: Could not parse output.", "", ""
     except Exception as e:
-        return f"Error generating script: {e}", None
+        return f"Error generating script: {e}", "", ""
 
 # ---------------------------------------------------------------------
 # Voice-Over Generation Function
@@ -122,13 +125,14 @@ def blend_audio(voice_path: str, music_path: str, ducking: bool):
 # ---------------------------------------------------------------------
 # Gradio Interface
 # ---------------------------------------------------------------------
-with gr.Blocks() as demo:
+ith gr.Blocks() as demo:
     gr.Markdown("""
-        # 🎧 AI Promo Studio with Step-by-Step Script, Voice, Music, and Mixing 🚀  
-        Generate and mix radio promos effortlessly with AI tools!
+        # 🎧 AI Promo Studio 🚀
+        Generate scripts, sound design, and music suggestions with ease.
     """)
 
     with gr.Tabs():
+        # Step 1: Generate Script
         with gr.Tab("Step 1: Generate Script"):
             with gr.Row():
                 user_prompt = gr.Textbox(label="Promo Idea", placeholder="E.g., A 30-second promo for a morning show.")
@@ -136,13 +140,14 @@ with gr.Blocks() as demo:
                 duration = gr.Slider(label="Duration (seconds)", minimum=15, maximum=60, step=15, value=30)
 
             generate_script_button = gr.Button("Generate Script")
-            script_output = gr.Textbox(label="Generated Script")
-            music_suggestion_output = gr.Textbox(label="Music Suggestion")
+            script_output = gr.Textbox(label="Generated Voice-Over Script", lines=5)
+            sound_design_output = gr.Textbox(label="Sound Design Ideas", lines=3)
+            music_suggestion_output = gr.Textbox(label="Music Suggestions", lines=3)
 
             generate_script_button.click(
                 fn=lambda user_prompt, model_id, duration: generate_script(user_prompt, model_id, hf_token, duration),
                 inputs=[user_prompt, llama_model_id, duration],
-                outputs=[script_output, music_suggestion_output],
+                outputs=[script_output, sound_design_output, music_suggestion_output],
             )
 
         with gr.Tab("Step 2: Generate Voice"):