Minor changes to variables

app.py

@@ -7,10 +7,6 @@ import tensorflow as tf
 import torch
 import os
 
-emo_list = []
-gen_list = []
-tempo_list = []
-
 physical_devices = tf.config.experimental.list_physical_devices('GPU')
 if len(physical_devices) > 0:
     tf.config.experimental.set_memory_growth(physical_devices[0], True)
@@ -45,6 +41,9 @@ if 'genre' not in st.session_state:
 if 'beat' not in st.session_state:
     st.session_state.beat = None
 
+if "text_prompt" not in st.session_state:
+    st.session_state.text_prompt = None
+
 emotion_service = load_emo_model()
 genre_service = load_genre_model()
 beat_service = load_beat_model()
@@ -73,16 +72,16 @@ with c3:
     st.text(st.session_state.beat)
 
 if st.session_state.emotion != None and st.session_state.genre != None and st.session_state.beat != None:
-    text_output = None
     if st.button("Generate text description to be fed into stable diffusion"):
         if st.session_state.beat > 100:
             speed = "medium and steady"
         else:
             speed = "slow and calm"
         st.caption("Text description of your music file")
-        text_output = "A scenic image that describes a " + speed + " pace with a feeling of" + st.session_state.emotion + "."
-        st.text(text_output)
-    if text_output:
+        text = "A scenic image that describes a " + speed + " pace with a feeling of" + st.session_state.emotion + "."
+        st.session_state.text_prompt = text
+        st.text(st.session_state.text_prompt)
+    if st.session_state.text_prompt:
        if st.button("Generate image from text description"):
-            image = image_service(text_output)
+            image = image_service(st.session_state.text_prompt)
             st.image(image)