Update app.py

app.py

@@ -22,7 +22,10 @@ if st.button("Generate Story"):
         ]
     }).json()
 
-    # Display generated story to user
-    generated_text = response["data"]
-    st.write("Generated Story:")
-    st.write(generated_text)
+    # Check if response contains the expected key
+    if "data" in response:
+        generated_text = response["data"]
+        st.write("Generated Story:")
+        st.write(generated_text)
+    else:
+        st.write("Error: Unexpected response from API.")