fix

main.py

@@ -4,48 +4,58 @@ import numpy as np
 from openai import OpenAI
 import os
 
-
-
 import streamlit as st
 from PIL import Image
 from diffusers import AutoPipelineForText2Image
 import random
+
+client = OpenAI()
 @st.cache_data(ttl=3600)
-def get_prompt_to_guess():
-    prompts = [
-        "Une cascade lumineuse dans une forêt enchantée",
-    "Un coucher de soleil sur une plage déserte",
-    "Un champ de tulipes multicolores à perte de vue",
-    "Un château perché sur une montagne majestueuse",
-    "Une ville futuriste illuminée par des néons",
-    "Une forêt brumeuse où les arbres semblent danser",
-    "Une soirée magique dans le ciel étoilé",
-    "Une bibliothèque remplie de livres flottants",
-    "Un paysage hivernal avec des arbres enneigés",
-    "Une ville suspendue dans les nuages",
-    "Un pont de cristal au-dessus d'une cascade étincelante",
-    "Un champ de coquelicots sous un ciel bleu azur",
-    "Un bateau en papier naviguant sur une rivière magique",
-    "Un jardin secret rempli de fleurs exotiques",
-    "Une île déserte entourée d'une mer turquoise",
-    "Une montgolfière survolant un paysage onirique",
-    "Un champ de lavande embaumant l'air",
-    "Un petit village entouré de montagnes enneigées",
-    "Une forêt tropicale avec des plantes géantes",
-    "Un phare solitaire sur une falaise abrupte",
-    "Un arc-en-ciel se reflétant dans un lac calme",
-    "Une cabane en bois cachée au milieu des arbres",
-    "Un champ de tournesols sous un soleil éclatant",
-    "Une ville médiévale entourée de murailles imposantes",
-    "Un château de glace scintillant dans la nuit",
-    "Un chemin de pierres menant à un endroit mystérieux",
-    "Une rue animée remplie de cafés et de terrasses",
-    "Une cascade gelée dans un paysage d'hiver",
-    "Un jardin japonais paisible avec un petit étang",
-    "Une aurore boréale éblouissante embrassant le ciel étoilé",
-    ]
-    return random.choice(prompts)
-    random_prompt = ["tree", "cat", "dog", "consultant", "artificial intelligence", "beauty", "immeuble", "plage", "cyborg", "futuristic"]
+def get_prompt_to_guess(index):
+    # prompts = [
+    # "Une cascade lumineuse dans une forêt enchantée",
+    # "Un coucher de soleil sur une plage déserte",
+    # "Un champ de tulipes multicolores à perte de vue",
+    # "Un château perché sur une montagne majestueuse",
+    # "Une ville futuriste illuminée par des néons",
+    # "Une forêt brumeuse où les arbres semblent danser",
+    # "Une soirée magique dans le ciel étoilé",
+    # "Une bibliothèque remplie de livres flottants",
+    # "Un paysage hivernal avec des arbres enneigés",
+    # "Une ville suspendue dans les nuages",
+    # "Un pont de cristal au-dessus d'une cascade étincelante",
+    # "Un champ de coquelicots sous un ciel bleu azur",
+    # "Un bateau en papier naviguant sur une rivière magique",
+    # "Un jardin secret rempli de fleurs exotiques",
+    # "Une île déserte entourée d'une mer turquoise",
+    # "Une montgolfière survolant un paysage onirique",
+    # "Un champ de lavande embaumant l'air",
+    # "Un petit village entouré de montagnes enneigées",
+    # "Une forêt tropicale avec des plantes géantes",
+    # "Un phare solitaire sur une falaise abrupte",
+    # "Un arc-en-ciel se reflétant dans un lac calme",
+    # "Une cabane en bois cachée au milieu des arbres",
+    # "Un champ de tournesols sous un soleil éclatant",
+    # "Une ville médiévale entourée de murailles imposantes",
+    # "Un château de glace scintillant dans la nuit",
+    # "Un chemin de pierres menant à un endroit mystérieux",
+    # "Une rue animée remplie de cafés et de terrasses",
+    # "Une cascade gelée dans un paysage d'hiver",
+    # "Un jardin japonais paisible avec un petit étang",
+    # "Une aurore boréale éblouissante embrassant le ciel étoilé",
+    # ]
+    # return random.choice(prompts)
+    random_prompt = ["arbre", 
+                     "écologie",
+                     "chat", 
+                     "chien", 
+                     "consultant", 
+                     "artificial intelligence", 
+                     "beauté", 
+                     "immeuble", 
+                     "plage", 
+                     "cyborg", 
+                     "futuriste"]
     response = client.chat.completions.create(
     model="gpt-3.5-turbo",
     messages=[
@@ -64,57 +74,69 @@ 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()
 
 pipe = get_model()
-prompt = get_prompt_to_guess()
-im_to_guess = generate_image(pipe, prompt)
-h, w = im_to_guess.size
+
+with open("style.css") as f:
+    st.markdown('<style>{}</style>'.format(f.read()), unsafe_allow_html=True)
+
+from text import compare_text, format_text_html
+if "guess_number" not in st.session_state:
+    st.session_state["guess_number"] = 0
 
 st.title("Guess the prompt by Ekimetrics")
-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])
+st.markdown("""
+            Game developed by Jeremy LE from Ekimetrics to test and play with the new SDXL Turbo model from stability.ai\n
+            Rules : \n
+            - guess the prompt (in French, with no fault) to generate the left image with the sdxl turbo model\n
+            - use testing prompt side to help you guess the prompt by testing some\n
+            - If a word is **correct** and **at the right place in sentence**, the word is in :green[green]\n 
+            - If a word is **correct** and **not** at the right place in sentence, the word is in :gray[gray]\n 
+            - If a word is **incorrect**, the word is in :red[red]\n
+            **Disclosure** : this runs on CPU so generation are quite slow (even with sdxl turbo). Generation time took approx 40s. 
+            """)
+
+next_guess = st.button("click here for next guess", use_container_width=True)
+if next_guess:
+    st.session_state["guess_number"] += 1
+    print("getting next pormpt")
+    print(st.session_state["guess_number"])
+prompt = get_prompt_to_guess(st.session_state["guess_number"])
+
+col_1, col_2 = st.columns([1,1])
 with col_1:
     st.header("GUESS THE PROMPT")
-    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"]:
+    guessed_prompt = st.text_input("Input your guess prompt")
+    submit_guess = st.button("guess the prompt", use_container_width=True, type="primary")
+    if submit_guess:
         if check_prompt(guessed_prompt, prompt):
-            st.text("Good prompt ! test again in 24h !")
+            st.markdown("Good prompt ! test again in 1h or click on next guess!")
         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)
+            st.markdown("wrong prompt !")
+        compare_dict = compare_text(guessed_prompt, prompt)
+        st.markdown(format_text_html(compare_dict))
+    get_answer = st.button("get the answer", use_container_width=True)
+    if get_answer:
+        st.markdown(f"Cheater ! but here is the prompt : \n**{prompt}**")
     
         
-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")
-    if st.session_state["testing"]:
+    testing_prompt = st.text_input("Input your testing prompt")
+    test_prompt = st.button("test prompt",use_container_width=True, type="primary")
+
+with col_1:
+    im_to_guess = generate_image(pipe, prompt)
+    h, w = im_to_guess.size
+    st.image(im_to_guess)
+
+with col_2:
+    if test_prompt:
         im = generate_image(pipe, testing_prompt)
         st.session_state["testing"] = False
     else:
         im = np.zeros([h,w,3])
-    st.session_state["testing"] = st.button("test the prompt")
     st.image(im)
-    
+        

style.css

@@ -0,0 +1,4 @@
+.fullScreenFrame > div {
+    display: flex;
+    justify-content: center;
+}

text.py

@@ -0,0 +1,19 @@
+def compare_text(text_to_guess, text_to_match):
+    words_to_match = text_to_match.lower().split(" ")
+    words_to_guess = text_to_guess.strip().split(" ")
+    return [
+        [word, word.lower() in words_to_match, i < len(words_to_match) and word.lower() == words_to_match[i]] for i, word in enumerate(words_to_guess) 
+    ]
+
+def format_one_word(word, in_sentence, correct_place):
+    if in_sentence and correct_place:
+        return f"**:green[{word}]**"
+    elif in_sentence:
+        return f"**:gray[{word}]**"
+    else:
+        return f"**:red[{word}]**"
+    
+def format_text_html(compare_text_dict):
+    return " ".join([
+        format_one_word(word, in_sentence, correct_place)  for word, in_sentence, correct_place in compare_text_dict
+        ])