Update app.py

app.py

@@ -24,10 +24,12 @@ def run_vlm(image, prompt):
     return processor.decode(out[0], skip_special_tokens=True)
 # return f"Model output for: '{prompt}' and selected image."
 
-def random_image():
-    path = random.choice(images)
-    return Image.open(path), path
-
+def get_random_image():
+    jpgs = [f for f in os.listdir(IMAGE_FOLDER) if f.endswith(".jpg")]
+    if not jpgs:
+        return STATIC_IMAGE_PATH, STATIC_IMAGE_PATH
+    selected = os.path.join(IMAGE_FOLDER, random.choice(jpgs))
+    return selected, selected 
 
 def process(image_path, user_prompt):
     image = Image.open(image_path)
@@ -39,9 +41,14 @@ with gr.Blocks() as demo:
 
     image_path = gr.Textbox(value=STATIC_IMAGE_PATH, visible=False)
     user_prompt = gr.Textbox(label="User Prompt")
-    run_button = gr.Button("Run Model")
+
+    with gr.Row():
+        random_button = gr.Button("Random Photo")
+        run_button = gr.Button("Run Model")
+
     output = gr.Textbox(label="Model Output")
 
+    random_button.click(fn=get_random_image, outputs=[image_display, image_path])
     run_button.click(fn=process, inputs=[image_path, user_prompt], outputs=output)
 
 # with gr.Blocks() as demo:
@@ -58,4 +65,4 @@ with gr.Blocks() as demo:
 #     run_button.click(fn=process, inputs=[image_path, user_prompt], outputs=output)
 
 
-demo.launch(share=True)
+demo.launch()