Update app.py

app.py

@@ -35,26 +35,30 @@ def generate_script(user_prompt: str, model_id: str, token: str, duration: int):
         llama_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
         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."
+            "You are an expert radio imaging producer specializing in sound design and music.\n"
+            "---\n"
+            f"Based on the user's concept and the selected duration of {duration} seconds, craft a concise, engaging promo script.\n"
+            "---\n"
+            "Ensure the script fits within the time limit and suggest a matching music style that complements the theme."
         )
 
         combined_prompt = f"{system_prompt}\nUser concept: {user_prompt}\nRefined script and music suggestion:"
         result = llama_pipeline(combined_prompt, max_new_tokens=500, do_sample=True, temperature=0.9)
 
-        # Extract the script and music suggestion
+        # Parsing the output
         generated_text = result[0]["generated_text"]
-        if "Music:" in generated_text:
-            parts = generated_text.split("Music:", 1)
-            script = parts[0].strip()
-            music_suggestion = parts[1].strip()
-            return script, music_suggestion
-        else:
-            return generated_text.strip(), "No specific music suggestion found."
+        if "Refined script and music suggestion:" in generated_text:
+            parts = generated_text.split("Refined script and music suggestion:", 1)[-1].strip()
+            if "Music Style:" in parts:
+                script, music_suggestion = parts.split("Music Style:", 1)
+                return script.strip(), music_suggestion.strip()
+            else:
+                return parts.strip(), "No specific music suggestion found."
+        return "Error: Could not parse the script.", None
     except Exception as e:
         return f"Error generating script: {e}", None
 
+
 # ---------------------------------------------------------------------
 # Voice-Over Generation Function
 # ---------------------------------------------------------------------