Update app.py

app.py

@@ -134,6 +134,7 @@ def reset_textbox():
     return gr.update(value='', interactive=False), gr.update(interactive=False)
 
 def random_sample():
+    global SAMPLE_IDX
     SAMPLE_IDX = random.choices(range(25), k=TOTAL_K)
     image = np.array(Image.open(os.path.join(os.path.dirname(__file__), "data/Court.jpeg")))
     random_characters = [Image.open(os.path.join(os.path.dirname(__file__), f'data/characters_{idx:02}.png')).resize((64, 64)) for idx in SAMPLE_IDX]
@@ -165,7 +166,12 @@ In this app, you can explore the outputs of a gpt-3.5 LLM.
 """
 
 def sumbit(image):
-    pass
+    global CURRENT_POSITION
+    balloon = Image.open(os.path.join(os.path.dirname(__file__), 'data/balloon.png')).resize((64, 64))
+    new_image = Image.fromarray(image).convert('RGBA')
+    for k in range(TOTAL_K):
+        new_image.paste(balloon, [CURRENT_POSITION[k][0], CURRENT_POSITION[k][1] + 64], balloon)
+    return np.array(new_image)
 
 theme = gr.themes.Default(primary_hue="green")
 agents = []
@@ -235,6 +241,8 @@ with gr.Blocks(css = """#col_container { margin-left: auto; margin-right: auto;}
 
     b0.click(random_sample, inputs = [], outputs = [image])
     b2.click(reset_sample, inputs = [], outputs = [image])
+
+    b1.click(submin, inputs = [image], outputs = [image])
                   
     # inputs.submit(reset_textbox, [], [inputs, b1], queue=False)
     # inputs.submit(predict, [inputs, top_p, temperature, chat_counter, chatbot, state], [chatbot, state, chat_counter, server_status_code, inputs, b1],)  #openai_api_key