Update app.py

app.py

@@ -279,6 +279,9 @@ def crewai_process(research_topic):
 if 'text_block' not in st.session_state:
     st.session_state['text_block'] = ''
 
+if 'on_topic' not in st.session_state:
+    st.session_state['on_topic'] = ''
+
 st.set_page_config(page_title='Clarifai Story Teller', layout='wide')
 
 
@@ -293,7 +296,7 @@ with tab1:
     
     # Input for the user
     input_topic = st.text_area("Input Topic", height=100, placeholder="Input Topic...")
-    
+    st.session_state['on_topic'] = input_topic
     # Button to run the process
     if st.button("Run"):
         # Run the crewai process
@@ -316,7 +319,7 @@ with tab2:
         cols = st.columns(4)
         with st.spinner('Generating Content...'):
             for i, prompt in enumerate(prompts):
-                image, img_error = generate_image(prompt)
+                image, img_error = generate_image(prompt+st.session_state.on_topic )
                 audio, audio_error = generate_audio(prompt)
     
                 with cols[i % 4]: