Update app_dialogue.py

app_dialogue.py

@@ -15,6 +15,7 @@ from gradio_client.client import DEFAULT_TEMP_DIR
 from text_generation import Client
 from transformers import AutoProcessor
 
+SECRET_TOKEN = os.getenv('SECRET_TOKEN', 'default_secret')
 
 MODELS = [
     # "HuggingFaceM4/idefics-9b-instruct",
@@ -335,6 +336,7 @@ textbox = gr.Textbox(
 )
 with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
     gr.HTML("""<h1 align="center">🐶 IDEFICS Playground</h1>""")
+    secret_token = gr.Textbox(label="Secret token")
     with gr.Row(variant="panel"):
         with gr.Column(scale=1):
             gr.Image(IDEFICS_LOGO, elem_id="banner-image", show_label=False, show_download_button=False)
@@ -484,6 +486,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
             )
 
     def model_inference(
+        secret_token,
         model_selector,
         user_prompt_str,
         chat_history,
@@ -494,6 +497,9 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
         repetition_penalty,
         top_p,
     ):
+        if secret_token != SECRET_TOKEN:
+            raise gr.Error(f'Invalid secret token. Please fork the original space if you want to use it for yourself.')
+
         if user_prompt_str.strip() == "" and image is None:
             return "", None, chat_history
 
@@ -639,6 +645,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
     textbox.submit(
         fn=model_inference,
         inputs=[
+            secret_token,
             model_selector,
             textbox,
             chatbot,
@@ -654,6 +661,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
     submit_btn.click(
         fn=model_inference,
         inputs=[
+            secret_token,
             model_selector,
             textbox,
             chatbot,
@@ -683,6 +691,7 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
     regenerate_btn.click(fn=remove_last_turn, inputs=chatbot, outputs=[chatbot, textbox]).then(
         fn=model_inference,
         inputs=[
+            secret_token,
             model_selector,
             textbox,
             chatbot,
@@ -704,176 +713,6 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
     submit_btn.click(lambda : gr.update(label='📁 Upload image', interactive=True), [], upload_btn)
     textbox.submit(lambda : gr.update(label='📁 Upload image', interactive=True), [], upload_btn)
     clear_btn.click(lambda : gr.update(label='📁 Upload image', interactive=True), [], upload_btn)
-    
-    # Using Flagging for saving dope and problematic examples
-    # Dope examples flagging
-    # dope_callback.setup(
-    #     [
-    #         model_selector,
-    #         textbox,
-    #         chatbot,
-    #         imagebox,
-    #         decoding_strategy,
-    #         temperature,
-    #         max_new_tokens,
-    #         repetition_penalty,
-    #         top_p,
-    #     ],
-    #     "gradio_dope_data_points",
-    # )
-    # dope_bttn.click(
-    #     lambda *args: dope_callback.flag(args),
-    #     [
-    #         model_selector,
-    #         textbox,
-    #         chatbot,
-    #         imagebox,
-    #         decoding_strategy,
-    #         temperature,
-    #         max_new_tokens,
-    #         repetition_penalty,
-    #         top_p,
-    #     ],
-    #     None,
-    #     preprocess=False,
-    # )
-    # # Problematic examples flagging
-    # problematic_callback.setup(
-    #     [
-    #         model_selector,
-    #         textbox,
-    #         chatbot,
-    #         imagebox,
-    #         decoding_strategy,
-    #         temperature,
-    #         max_new_tokens,
-    #         repetition_penalty,
-    #         top_p,
-    #     ],
-    #     "gradio_problematic_data_points",
-    # )
-    # problematic_bttn.click(
-    #     lambda *args: problematic_callback.flag(args),
-    #     [
-    #         model_selector,
-    #         textbox,
-    #         chatbot,
-    #         imagebox,
-    #         decoding_strategy,
-    #         temperature,
-    #         max_new_tokens,
-    #         repetition_penalty,
-    #         top_p,
-    #     ],
-    #     None,
-    #     preprocess=False,
-    # )
-
-    # gr.Markdown("""## How to use?
-
-    #     There are two ways to provide image inputs:
-    #     - Using the image box on the left panel
-    #     - Using the inline syntax: `text<fake_token_around_image><image:URL_IMAGE><fake_token_around_image>text`
-
-    #     The second syntax allows inputting an arbitrary number of images.""")
-
-    examples_path = os.path.dirname(__file__)
-    gr.Examples(
-        examples=[
-            [
-                (
-                    "Which famous person does the person in the image look like? Could you craft an engaging narrative"
-                    " featuring this character from the image as the main protagonist?"
-                ),
-                f"{examples_path}/example_images/obama-harry-potter.jpg",
-            ],
-            [
-                "Can you describe the image? Do you think it's real?",
-                f"{examples_path}/example_images/rabbit_force.png",
-            ],
-            ["Explain this meme to me.", f"{examples_path}/example_images/meme_french.jpg"],
-            ["Give me a short and easy recipe for this dish.", f"{examples_path}/example_images/recipe_burger.webp"],
-            [
-                "I want to go somewhere similar to the one in the photo. Give me destinations and travel tips.",
-                f"{examples_path}/example_images/travel_tips.jpg",
-            ],
-            [
-                "Can you name the characters in the image and give their French names?",
-                f"{examples_path}/example_images/gaulois.png",
-            ],
-            # ["Describe this image in detail.", f"{examples_path}/example_images/plant_bulb.webp"],
-            ["Write a complete sales ad for this product.", f"{examples_path}/example_images/product_ad.jpg"],
-            [
-                (
-                    "As an art critic AI assistant, could you describe this painting in details and make a thorough"
-                    " critic?"
-                ),
-                f"{examples_path}/example_images/art_critic.png",
-            ],
-            [
-                "Can you tell me a very short story based on this image?",
-                f"{examples_path}/example_images/chicken_on_money.png",
-            ],
-            ["Write 3 funny meme texts about this image.", f"{examples_path}/example_images/elon_smoking.jpg"],
-            [
-                "Who is in this picture? Why do people find it surprising?",
-                f"{examples_path}/example_images/pope_doudoune.webp",
-            ],
-            # ["<fake_token_around_image><image:https://assets.stickpng.com/images/6308b83261b3e2a522f01467.png><fake_token_around_image>Make a poem about the company in the image<fake_token_around_image><image:https://miro.medium.com/v2/resize:fit:1400/0*jvDu2oQreHn63-fJ><fake_token_around_image>organizing the Woodstock of AI event,<fake_token_around_image><image:https://nationaltoday.com/wp-content/uploads/2019/12/national-llama-day-1200x834.jpg><fake_token_around_image>and the fact they brought those to the event.", None],
-            ["What are the armed baguettes guarding?", f"{examples_path}/example_images/baguettes_guarding_paris.png"],
-            # ["Can you describe the image?", f"{examples_path}/example_images/bear_costume.png"],
-            ["What is this animal and why is it unusual?", f"{examples_path}/example_images/blue_dog.png"],
-            [
-                "What is this object and do you think it is horrifying?",
-                f"{examples_path}/example_images/can_horror.png",
-            ],
-            [
-                (
-                    "What is this sketch for? How would you make an argument to prove this sketch was made by Picasso"
-                    " himself?"
-                ),
-                f"{examples_path}/example_images/cat_sketch.png",
-            ],
-            ["Which celebrity does this claymation figure look like?", f"{examples_path}/example_images/kanye.jpg"],
-            # [
-            #     "Is there a celebrity look-alike in this image? What is happening to the person?",
-            #     f"{examples_path}/example_images/ryan-reynolds-borg.jpg",
-            # ],
-            # ["Can you describe this image in details please?", f"{examples_path}/example_images/dragons_playing.png"],
-            ["What can you tell me about the cap in this image?", f"{examples_path}/example_images/ironman_cap.png"],
-            [
-                "Can you write an advertisement for Coca-Cola based on this image?",
-                f"{examples_path}/example_images/polar_bear_coke.png",
-            ],
-            # [
-            #     "What is the rabbit doing in this image? Do you think this image is real?",
-            #     f"{examples_path}/example_images/rabbit_force.png",
-            # ],
-            # ["What is happening in this image and why is it unusual?", f"{examples_path}/example_images/ramen.png"],
-            # [
-            #     "What I should look most forward to when I visit this place?",
-            #     f"{examples_path}/example_images/tree_fortress.jpg",
-            # ],
-            # ["Who is the person in the image and what is he doing?", f"{examples_path}/example_images/tom-cruise-astronaut-pegasus.jpg"],
-            [
-                "What is happening in this image? Which famous personality does this person in center looks like?",
-                f"{examples_path}/example_images/gandhi_selfie.jpg",
-            ],
-            [
-                "What do you think the dog is doing and is it unusual?",
-                f"{examples_path}/example_images/surfing_dog.jpg",
-            ],
-        ],
-        inputs=[textbox, imagebox],
-        outputs=[textbox, imagebox, chatbot],
-        fn=process_example,
-        cache_examples=True,
-        examples_per_page=6,
-        label=(
-            "Click on any example below to get started.\nFor convenience, the model generations have been"
-            " pre-computed with `idefics-80b-instruct`."
-        ),
-    )
 
 demo.queue(concurrency_count=40, max_size=40)
 demo.launch()