add images & videos

Signed-off-by: Samhita Alla <[email protected]>

app.py

@@ -9,6 +9,7 @@ import plotly
 import plotly.express as px
 import requests
 from dotenv import load_dotenv
+
 from gantry_callback.gantry_util import GantryImageToTextLogger
 from gantry_callback.s3_util import make_unique_bucket_name
 
@@ -19,6 +20,8 @@ GANTRY_APP_NAME = os.getenv("GANTRY_APP_NAME")
 GANTRY_KEY = os.getenv("GANTRY_API_KEY")
 MAPBOX_TOKEN = os.getenv("MAPBOX_TOKEN")
 
+examples = json.load(open("examples.json"))
+
 
 def get_plotly_graph(
     latitude: float, longitude: float, location: str
@@ -125,6 +128,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             # Flag button
             img_flag_button = gr.Button("Flag this output")
+        gr.Examples(examples["images"], inputs=[img_input])
     with gr.Tab("Video"):
         with gr.Row():
             video_input = gr.Video(type="filepath", label="Video")
@@ -133,6 +137,7 @@ with gr.Blocks() as demo:
                 video_coordinates = gr.Textbox(label="Coordinates")
                 video_plot = gr.Plot()
         video_text_button = gr.Button("Go locate!")
+        gr.Examples(examples["videos"], inputs=[video_input])
     with gr.Tab("YouTube Link"):
         with gr.Row():
             url_input = gr.Textbox(label="Link")
@@ -141,6 +146,7 @@ with gr.Blocks() as demo:
                 url_coordinates = gr.Textbox(label="Coordinates")
                 url_plot = gr.Plot()
         url_text_button = gr.Button("Go locate!")
+        gr.Examples(examples["video_urls"], inputs=[url_input])
 
     # Gantry flagging for image #
     callback = GantryImageToTextLogger(application=GANTRY_APP_NAME, api_key=GANTRY_KEY)
@@ -174,8 +180,6 @@ with gr.Blocks() as demo:
         outputs=[url_text_output, url_coordinates, url_plot],
     )
 
-    examples = gr.Examples(".", inputs=[img_input, video_input, url_input])
-
     gr.Markdown(
         "Check out the [GitHub repository](https://github.com/samhita-alla/geolocator) that this demo is based off of."
     )

examples.json

@@ -0,0 +1,22 @@
+{
+  "images": [
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/images/argentina.jpg",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/images/beijing.jpg",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/images/greece.jpg",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/images/russia.jpg",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/images/arizona.jpg",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/images/switzerland.jpg",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/images/dubai.jpg"
+  ],
+  "videos": [
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/newyork.mp4",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/faultinourstars-amsterdam.mp4",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/lostintranslation-tokyo.mp4",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/main/services/gradio/data/test/vicky-spain.mp4",
+    "https://raw.githubusercontent.com/samhita-alla/geolocator/63ee0ae2c1d1263f2da88221e800c58ce8c02351/services/gradio/data/test/3idiots-india.mp4"
+  ],
+  "video_urls": [
+    "https://www.youtube.com/watch?v=wxeQkJTZrsw",
+    "https://www.youtube.com/watch?v=Mvdr-epa1hc"
+  ]
+}

log.csv

@@ -1,2 +0,0 @@
-Image,Video,Link
-"data/test/greece.jpg","https://cdn.discordapp.com/attachments/545481172399030272/1029739008986714132/video.mp4","https://www.youtube.com/watch?v=wxeQkJTZrsw"