fix

main.py

@@ -1,14 +1,16 @@
-from diffusers import AutoPipelineForText2Image
+
 import torch
 import numpy as np
 from openai import OpenAI
 import os
+
 client = OpenAI()
 
 import streamlit as st
 from PIL import Image
+from diffusers import AutoPipelineForText2Image
 
-@st.cache_data(ttl=3600)
+@st.cache_data(ttl=600)
 def get_prompt_to_guess():
     response = client.chat.completions.create(
     model="gpt-3.5-turbo",
@@ -28,9 +30,18 @@ def get_model():
 def generate_image(_pipe, prompt):
     return _pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0, seed=1).images[0]
 
+if "ask_answer" not in st.session_state:
+    st.session_state["ask_answer"] = False
+
+if "testing" not in st.session_state:
+    st.session_state["testing"] = False
+
 if "submit_guess" not in st.session_state:
     st.session_state["submit_guess"] = False
 
+if "real_ask_answer" not in st.session_state:
+    st.session_state["real_ask_answer"] = False
+
 def check_prompt(prompt, prompt_to_guess):
     return prompt.strip() == prompt_to_guess.strip()
 
@@ -40,13 +51,12 @@ im_to_guess = generate_image(pipe, prompt)
 h, w = im_to_guess.size
 
 st.title("Guess the prompt by Ekimetrics")
-st.text("Rules : guess the prompt (in French) to generate the left image with the sdxl turbo model")
+st.text("Rules : guess the prompt (in French, with no fault) to generate the left image with the sdxl turbo model")
 st.text("Hint : use right side to help you guess the prompt by testing some")
 st.text("Disclosure : this runs on CPU so generation are quite slow (even with sdxl turbo)")
 col_1, col_2 = st.columns([0.5, 0.5])
 with col_1:
     st.header("GUESS THE PROMPT")
-    st.image(im_to_guess)
     guessed_prompt = st.text_area("Input your guess prompt")
     st.session_state["submit_guess"] = st.button("guess the prompt")
     if st.session_state["submit_guess"]:
@@ -54,14 +64,23 @@ with col_1:
             st.text("Good prompt ! test again in 24h !")
         else:
             st.text("wrong prompt !")
+    st.session_state["ask_answer"] = st.button("get the answer")
+    if st.session_state["ask_answer"]:
+        st.text(f"Cheater ! but here is the prompt : \n {prompt}")
+    st.image(im_to_guess)
+    
+        
+if "testing" not in st.session_state:
+    st.session_state["testing"] = False
 
 with col_2:
     st.header("TEST THE PROMPT")
-    testing_prompt = st.text_area("Input your testing prompt")
-    st.session_state["testing"] = st.button("test the prompt")
     if st.session_state["testing"]:
         im = generate_image(pipe, testing_prompt)
         st.session_state["testing"] = False
     else:
         im = np.zeros([h,w,3])
+    testing_prompt = st.text_area("Input your testing prompt")
+    st.session_state["testing"] = st.button("test the prompt")
     st.image(im)
+    

requirements.txt

@@ -3,4 +3,6 @@ transformers
 accelerate
 streamlit
 torch
-Pillow
+Pillow
+openai
+diffusers