Update app.py

app.py

@@ -41,15 +41,15 @@ def generate_script(user_prompt: str, model_id: str, token: str, duration: int):
         )
 
         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)
+        result = llama_pipeline(combined_prompt, max_new_tokens=500, do_sample=True, temperature=0.9)
 
-        # Ensure the output contains the expected structure
+        # Check if the output contains "Music Suggestion:"
         generated_text = result[0]["generated_text"]
         if "Music Suggestion:" in generated_text:
             script, music_suggestion = generated_text.split("Music Suggestion:", 1)
             return script.strip(), music_suggestion.strip()
         else:
-            # Fallback if the output format isn't as expected
+            # Return the full text as the script and indicate no music suggestion was found
             return generated_text.strip(), "No specific music suggestion found."
     except Exception as e:
         return f"Error generating script: {e}", None