Update app.py

app.py

@@ -11,7 +11,10 @@ udio_auth_token = os.getenv("UDIO_API_KEY")
 udio_wrapper = UdioWrapper(udio_auth_token)
 
 
-def run_agent(role, goal, backstory, task_description, expected_output):
+def run_agent(pw,role, goal, backstory, task_description, expected_output):
+    if pw != os.getenv("PW"):
+        raise gr.Error("Invalid password. Please try again.")
+        
     # Create agent with user-specified parameters
     general_agent = Agent(
         role=role,
@@ -43,7 +46,7 @@ def run_agent(role, goal, backstory, task_description, expected_output):
     #song_path = song[0]['song_path']
 
     #return [result,song_path]
-    return song
+    return result
 
 def make_song(task_description, result):
     song = udio_wrapper.create_song(prompt=task_description,custom_lyrics=result)
@@ -53,6 +56,7 @@ def make_song(task_description, result):
 with gr.Blocks() as demo:
     gr.Markdown("# Agent and Task Configuration")
 
+    pw = gr.Textbox(label="Password", type="password", placeholder="Enter the password")
     role = gr.Textbox(label="Role of the Agent", placeholder="Enter the role of your agent")
     goal = gr.Textbox(label="Goal of the Agent", placeholder="Enter the goal of your agent")
     backstory = gr.Textbox(label="Backstory of the Agent", placeholder="Enter the backstory of your agent")
@@ -68,7 +72,7 @@ with gr.Blocks() as demo:
 
     submit_button.click(
         fn=run_agent,
-        inputs=[role, goal, backstory, task_description, expected_output],
+        inputs=[pw,role, goal, backstory, task_description, expected_output],
         #outputs=[result,player]
         outputs=result
     )