Update app.py

app.py

@@ -30,9 +30,14 @@ prompt = st.text_area("Enter the beginning of your plot...")
 clicked = st.button("Generate my movie")
 
 if clicked:
-    generated_plot = model.generate_plot(prompt)[0]
-    st.write("Assistant:")
-    st.markdown(generated_plot)
+    st.write("Clicked!")
+    generated_plot = model.generate_plot(prompt)
+    st.write("Generated plot:", generated_plot)
+    if generated_plot:
+        st.write("Assistant:")
+        st.markdown(generated_plot[0])
+    else:
+        st.write("No plot generated.")
 
 with open('./style.css') as f:
     css = f.read()
@@ -40,5 +45,3 @@ with open('./style.css') as f:
 
 
 
-
-