Update app.py

app.py

@@ -27,29 +27,33 @@ LINKS = {
     "karlo": "https://huggingface.co/kakaobrain/karlo-v1-alpha",
 }
 SD_1_5_RESULT = """
+"## The traditional one 🔥! 
 ![img](https://huggingface.co/datasets/OpenGenAI/logos/blob/main/37351293.png)
-"The traditional one 🔥! You mostly resonate with **Stable Diffusion 1-5** released by RunwayML.
+\n You mostly resonate with **Stable Diffusion 1-5** released by RunwayML.
 \n Stable Diffusion 1-5 is the most used open-source text-to-image model offering an amazing speed-to-image-quality trade-off!
-Check out your soulmate [here](https://huggingface.co/runwayml/stable-diffusion-v1-5).
+\n Check out your soulmate [here](https://huggingface.co/runwayml/stable-diffusion-v1-5).
 """
 SD_2_1_RESULT = """
+## The creative one 🎨! 
 ![img](https://huggingface.co/datasets/OpenGenAI/logos/resolve/main/7vmYr2XwVcPtkLzac_jxQ.png)
-"The creative one 🎨! You mostly resonate with **Stable Diffusion 2-1** released by Stability AI.
+\n You mostly resonate with **Stable Diffusion 2-1** released by Stability AI.
 \n Stable Diffusion 2-1 is the latest open-source release of Stable Diffusion and allows to render stunning images of much larger sizes than Stable Diffusion v1.
 Try it out [here](https://huggingface.co/stabilityai/stable-diffusion-2-1).
 """
 IF_V1_0 = """
+## The powerful one ⚡! 
 ![img](https://huggingface.co/datasets/OpenGenAI/logos/blob/main/1662453741854-63170739dc97a974718be2c7.png)
-The powerful one ⚡! You mostly resonate with **IF v1-0** released by DeepFloyd.
+\n You mostly resonate with **IF v1-0** released by DeepFloyd.
 \n IF v1-0 is by far the largest of the open-sourced text-to-image models and is a very powerful image generator.
-Besides being able to generate multiple complex concepts in the same image, IF v1-0 is also extremely good at generating text in images.
-Check out your new best friend [here](https://huggingface.co/DeepFloyd/IF-I-XL-v1.0)
+\n Besides being able to generate multiple complex concepts in the same image, IF v1-0 is also extremely good at generating text in images.
+\n Check out your new best friend [here](https://huggingface.co/DeepFloyd/IF-I-XL-v1.0)
 """
 KARLO = """
+## The precise one 🎯!
 ![img](https://huggingface.co/datasets/OpenGenAI/logos/blob/main/1670220967262-615ed619c807b26d117a49bd.png)
-The precise one 🎯! You mostly resonate with **Karlo** released by KakaoBrain.
+\n You mostly resonate with **Karlo** released by KakaoBrain.
 \n Karlo is based on the same architecture as DALLE-2 and has been trained on the [well curated COYO dataset](https://huggingface.co/datasets/kakaobrain/coyo-700m).
-Play around with it [here]("https://huggingface.co/kakaobrain/karlo-v1-alpha").
+\n Play around with it [here]("https://huggingface.co/kakaobrain/karlo-v1-alpha").
 """
 
 RESULT = {
@@ -199,9 +203,10 @@ def change_view(row_number, dataframe):
 
         dataset.push_to_hub(repo_id, token=os.getenv("HF_TOKEN"))
         return {
-            intro_view: gr.update(visible=True),
+            intro_view: gr.update(visible=False),
             result_view: gr.update(visible=True),
             gallery_view: gr.update(visible=False),
+            start_view: gr.update(visible=True)
             result: RESULT[favorite_model],
         }
     else:
@@ -209,6 +214,7 @@ def change_view(row_number, dataframe):
             intro_view: gr.update(visible=False),
             result_view: gr.update(visible=False),
             gallery_view: gr.update(visible=True),
+            start_view: gr.update(visible=False)
             result: "",
         }
 
@@ -216,9 +222,10 @@ def change_view(row_number, dataframe):
 TITLE = "# What's Your AI Gen Personality 🧬?"
 
 DESCRIPTION = """
-*In this interactive game you are shown image descriptions along side 4 AI generated images. Select the image that best fits the image description.
-Answer **10** questions to find out what AI generator most resonates with you. If none of the images fit the prompt, choose the image that comes closest 
-in your opinion.* \n\n
+*In this interactive game you are shown image descriptions along side 4 AI generated images.
+\n Select the image that best fits the image description.
+\n Answer **10** questions to find out what AI generator most resonates with you. 
+\n Your submissions contribute to [**Open Parti Prompts Leaderboard**](https://huggingface.co/spaces/OpenGenAI/parti-prompts-leaderboard) ❤️
 """
 EXPLANATION = """\n\n
 ## What to do 📖 \n\n
@@ -227,23 +234,23 @@ EXPLANATION = """\n\n
 2. A prompt and 4 different images are displayed
 3. Select your favorite image
 4. After 10 rounds your favorite diffusion model is displayed \n\n
-
-*The prompts you are shown originate from the [Parti Prompts](https://huggingface.co/datasets/nateraw/parti-prompts) dataset. 
+"""
+NOTE = """
+The prompts you are shown originate from the [Parti Prompts](https://huggingface.co/datasets/nateraw/parti-prompts) dataset. 
 Parti Prompts is designed to test text-to-image AI models on 1600+ prompts of varying difficulty and categories. 
 The images you are shown have been pre-generated with 4 state-of-the-art open-sourced text-to-image models. \n\n
 You answers will be used to contribute to the official [**Open Parti Prompts Leaderboard**](https://huggingface.co/spaces/OpenGenAI/parti-prompts-leaderboard). \n
 By playing this game, you are greatly helping the community to have a better, **human-preference-aligned** metric to compare text-to-image models ❤️. \n\n
-Every month, the generated images will be updated with possibly improved models.*
+Every month, the generated images will be updated with possibly improved models.
 """
 
 GALLERY_COLUMN_NUM = len(SUBMISSIONS)
 
 with gr.Blocks() as demo:
+    gr.Markdown(TITLE)
     with gr.Column(visible=True) as intro_view:
-        gr.Markdown(TITLE)
         gr.Markdown(DESCRIPTION)
         gr.Markdown(EXPLANATION)
-        start_button = gr.Button("Start").style(full_width=False)
 
     headers = ["prompt", "result", "id", "Challenge", "Category", "Note"] + [
         f"choice_{i}" for i in range(len(SUBMISSIONS))
@@ -272,6 +279,10 @@ with gr.Blocks() as demo:
         )
         gr.Markdown("Click on start to play again!")
 
+    with gr.Column(visible=True) as start_view:
+        start_button = gr.Button("Start").style(full_width=True)
+        gr.Markdown(NOTE)
+
     with gr.Column(visible=False):
         selected_image = gr.Number(label="Selected index", value=-1, precision=0)
 
@@ -307,7 +318,7 @@ with gr.Blocks() as demo:
     ).then(
         fn=change_view,
         inputs=[row_number, dataframe],
-        outputs=[intro_view, result_view, gallery_view, result]
+        outputs=[intro_view, result_view, gallery_view, start_view, result]
     ).then(
         fn=process, inputs=[dataframe], outputs=[image_1, image_2, image_3, image_4, prompt, counter]
     )
@@ -325,7 +336,7 @@ with gr.Blocks() as demo:
         ).then(
             fn=change_view,
             inputs=[row_number, dataframe],
-            outputs=[intro_view, result_view, gallery_view, result]
+            outputs=[intro_view, result_view, gallery_view, start_view, result]
         ).then(
             fn=process,
             inputs=[dataframe, row_number],