Update app.py

app.py

@@ -34,12 +34,12 @@ if "messages" not in st.session_state:
 
 init_alr = False
 
-def init_model():
+def init_model(user_input):
     os.environ["API_KEY"] = "AIzaSyCADQk7KaP90mSKslhtEC3qRjyeUVxBkvI"
     genai.configure(api_key=os.environ.get("API_KEY"))
     model = genai.GenerativeModel('gemini-pro')
     
-    userInput = "Give me an afrofunk beat along with a cool melody and catchy drums, along with an opera style"
+    userInput = user_input #"Give me an afrofunk beat along with a cool melody and catchy drums, along with an opera style"
     prompt = userInput + ". Output ONLY the specific music genre that the user would like that fits their requirements, with no excess words. Simply the genre, NO APOSTROPHES. If required, add more than one genre to best fit the user's prompt by adding a comma and space and a another genre. Desired Output: <all_lowercase_comma_separated_genres>"
     response = model.generate_content(prompt)
     genre = response.text
@@ -65,7 +65,7 @@ if prompt := st.chat_input():
         if init_alr == False:
             init_alr = True
             
-            genre = init_model()
+            genre = init_model(prompt)
             #Call query generator with text_input
             #query = prompty(date_input, prompt)
             #result = qa({"query": query})