m3hrdadfi commited on
Commit
9cdf303
1 Parent(s): b3fccf7

Ignore session_state bug for now

Browse files
Files changed (1) hide show
  1. app.py +15 -4
app.py CHANGED
@@ -224,8 +224,11 @@ def main():
224
  initial_sidebar_state="expanded"
225
  )
226
  generator = load_text_generator()
227
- if 'get_random_frame' not in st.session_state:
228
- st.session_state.get_random_frame = generator.frames[0]
 
 
 
229
 
230
  local_css("asset/css/style.css")
231
 
@@ -269,7 +272,11 @@ def main():
269
  unsafe_allow_html=True
270
  )
271
  if recipe_button:
272
- st.session_state.get_random_frame = generator.frames[random.randint(0, len(generator.frames)) - 1]
 
 
 
 
273
  entered_items.markdown("**Generate recipe for:** " + items)
274
  with st.spinner("Generating recipe..."):
275
 
@@ -292,7 +299,11 @@ def main():
292
 
293
  with r1:
294
  # st.write(st.session_state.get_random_frame)
295
- recipe_post = generator.generate_frame(generated_recipe, st.session_state.get_random_frame)
 
 
 
 
296
  st.image(
297
  recipe_post,
298
  # width=500,
224
  initial_sidebar_state="expanded"
225
  )
226
  generator = load_text_generator()
227
+ if hasattr(st, "session_state"):
228
+ if 'get_random_frame' not in st.session_state:
229
+ st.session_state.get_random_frame = generator.frames[0]
230
+ else:
231
+ get_random_frame = generator.frames[0]
232
 
233
  local_css("asset/css/style.css")
234
 
272
  unsafe_allow_html=True
273
  )
274
  if recipe_button:
275
+ if hasattr(st, "session_state"):
276
+ st.session_state.get_random_frame = generator.frames[random.randint(0, len(generator.frames)) - 1]
277
+ else:
278
+ get_random_frame = generator.frames[random.randint(0, len(generator.frames)) - 1]
279
+
280
  entered_items.markdown("**Generate recipe for:** " + items)
281
  with st.spinner("Generating recipe..."):
282
 
299
 
300
  with r1:
301
  # st.write(st.session_state.get_random_frame)
302
+ if hasattr(st, "session_state"):
303
+ recipe_post = generator.generate_frame(generated_recipe, st.session_state.get_random_frame)
304
+ else:
305
+ recipe_post = generator.generate_frame(generated_recipe, get_random_frame)
306
+
307
  st.image(
308
  recipe_post,
309
  # width=500,