Update app.py

app.py

@@ -150,10 +150,22 @@ Immediately STOP after that. It should be EXACTLY in this format:
 "A grand orchestral arrangement with thunderous percussion, epic brass fanfares, and soaring strings, creating a cinematic atmosphere fit for a heroic battle"
 """
 
+mustango_sys = f"""
+You are an AI whose job is to help users create their own music which its genre will reflect the character or scene from an image described by users.
+In particular, you need to respond succintly with few musical words, in a friendly tone, write a musical prompt for a music generation model, incmudond chords progression.
+
+For example, if a user says, "a painting of three old women having tea party", provide immediately a musical prompt corresponding to the image description. 
+Immediately STOP after that. It should be EXACTLY in this format:
+"The song is an instrumental. The song is in medium tempo with a classical guitar playing a lilting melody in accompaniment style. The song is emotional and romantic. The song is a romantic instrumental song. The chord sequence is Gm, F6, Ebm. The time signature is 4/4. This song is in Adagio. The key of this song is G minor."
+"""
+
 @spaces.GPU(enable_queue=True)
-def get_musical_prompt(user_prompt):
-    
-    agent_maker_sys = standard_sys
+def get_musical_prompt(user_prompt, chosen_model):
+
+    if chosen_model == "Mustango" :
+        agent_maker_sys = mustango_sys
+    else :
+        agent_maker_sys = standard_sys
 
     instruction = f"""
 <|system|>
@@ -174,7 +186,7 @@ def infer(image_in, chosen_model):
     user_prompt = get_caption(image_in)
     
     gr.Info("Building a musical prompt according to the image caption ...")
-    musical_prompt = get_musical_prompt(user_prompt)
+    musical_prompt = get_musical_prompt(user_prompt, chosen_model)
 
     if chosen_model == "MAGNet" :
         gr.Info("Now calling MAGNet for music...")