add gradio code

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

README.md

@@ -1,6 +1,6 @@
 ---
 title: Geolocator
-emoji: 🌍
+emoji: 📍
 colorFrom: yellow
 colorTo: red
 sdk: gradio

app.py

@@ -0,0 +1,136 @@
+import json
+import mimetypes
+import os
+from typing import Tuple
+
+import gradio as gr
+import pandas as pd
+import plotly
+import plotly.express as px
+import requests
+from dotenv import load_dotenv
+
+load_dotenv()
+
+URL = os.getenv("ENDPOINT")
+
+
+def get_plotly_graph(
+    latitude: float, longitude: float, location: str
+) -> plotly.graph_objects.Figure:
+    lat_long_data = [[latitude, longitude, location]]
+    map_df = pd.DataFrame(lat_long_data, columns=["latitude", "longitude", "location"])
+
+    px.set_mapbox_access_token(os.getenv("MAPBOX_TOKEN"))
+    fig = px.scatter_mapbox(
+        map_df,
+        lat="latitude",
+        lon="longitude",
+        hover_name="location",
+        color_discrete_sequence=["fuchsia"],
+        zoom=5,
+        height=300,
+    )
+    fig.update_layout(mapbox_style="dark")
+    fig.update_layout(margin={"r": 0, "t": 0, "l": 0, "b": 0})
+    return fig
+
+
+def image_gradio(img_file: str) -> Tuple[str, plotly.graph_objects.Figure]:
+    data = json.loads(
+        requests.post(
+            f"{URL}predict-image",
+            files={
+                "image": (
+                    img_file,
+                    open(img_file, "rb"),
+                    mimetypes.guess_type(img_file)[0],
+                )
+            },
+        ).text
+    )
+
+    location = data["location"]
+    return data["location"], get_plotly_graph(
+        latitude=data["latitude"], longitude=data["longitude"], location=location
+    )
+
+
+def video_gradio(video_file: str) -> Tuple[str, plotly.graph_objects.Figure]:
+    data = json.loads(
+        requests.post(
+            f"{URL}predict-video",
+            files={
+                "video": (
+                    video_file,
+                    open(video_file, "rb"),
+                    "application/octet-stream",
+                )
+            },
+        ).text
+    )
+
+    location = data["location"]
+    return location, get_plotly_graph(
+        latitude=data["latitude"], longitude=data["longitude"], location=location
+    )
+
+
+def url_gradio(url: str) -> Tuple[str, plotly.graph_objects.Figure]:
+    data = json.loads(
+        requests.post(
+            f"{URL}predict-url",
+            headers={"content-type": "text/plain"},
+            data=url,
+        ).text
+    )
+
+    location = data["location"]
+    return location, get_plotly_graph(
+        latitude=data["latitude"], longitude=data["longitude"], location=location
+    )
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("# GeoLocator")
+    gr.Markdown(
+        "## An app that guesses the location of an image 🌌, a video 📹 or a YouTube link 🔗."
+    )
+    gr.Markdown(
+        "Find the code powering this application [here](https://github.com/samhita-alla/geolocator)."
+    )
+    with gr.Tab("Image"):
+        with gr.Row():
+            img_input = gr.Image(type="filepath", label="im")
+            with gr.Column():
+                img_text_output = gr.Textbox(label="Location")
+                img_plot = gr.Plot()
+        img_text_button = gr.Button("Go locate!")
+    with gr.Tab("Video"):
+        with gr.Row():
+            video_input = gr.Video(type="filepath", label="video")
+            with gr.Column():
+                video_text_output = gr.Textbox(label="Location")
+                video_plot = gr.Plot()
+        video_text_button = gr.Button("Go locate!")
+    with gr.Tab("YouTube Link"):
+        with gr.Row():
+            url_input = gr.Textbox(label="YouTube video link")
+            with gr.Column():
+                url_text_output = gr.Textbox(label="Location")
+                url_plot = gr.Plot()
+        url_text_button = gr.Button("Go locate!")
+
+    img_text_button.click(
+        image_gradio, inputs=img_input, outputs=[img_text_output, img_plot]
+    )
+    video_text_button.click(
+        video_gradio, inputs=video_input, outputs=[video_text_output, video_plot]
+    )
+    url_text_button.click(
+        url_gradio, inputs=url_input, outputs=[url_text_output, url_plot]
+    )
+
+    examples = gr.Examples(".", inputs=[img_input, url_input])
+
+demo.launch()

log.csv

@@ -0,0 +1,2 @@
+im,text
+"data/test/greece.jpg","https://www.youtube.com/watch?v=wxeQkJTZrsw"

requirements.in

@@ -0,0 +1,5 @@
+gradio
+pandas
+plotly
+requests
+python-dotenv

requirements.txt

@@ -0,0 +1,176 @@
+#
+# This file is autogenerated by pip-compile with python 3.9
+# To update, run:
+#
+#    pip-compile --output-file=gradio_requirements.txt gradio_requirements.in
+#
+aiohttp==3.8.3
+    # via gradio
+aiosignal==1.2.0
+    # via aiohttp
+anyio==3.6.1
+    # via
+    #   httpcore
+    #   starlette
+async-timeout==4.0.2
+    # via aiohttp
+attrs==22.1.0
+    # via aiohttp
+bcrypt==4.0.1
+    # via paramiko
+certifi==2022.9.24
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+cffi==1.15.1
+    # via
+    #   cryptography
+    #   pynacl
+charset-normalizer==2.1.1
+    # via
+    #   aiohttp
+    #   requests
+click==8.1.3
+    # via uvicorn
+contourpy==1.0.5
+    # via matplotlib
+cryptography==38.0.1
+    # via paramiko
+cycler==0.11.0
+    # via matplotlib
+fastapi==0.85.0
+    # via gradio
+ffmpy==0.3.0
+    # via gradio
+fonttools==4.37.4
+    # via matplotlib
+frozenlist==1.3.1
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2022.8.2
+    # via gradio
+gradio==3.4.1
+    # via -r gradio_requirements.in
+h11==0.12.0
+    # via
+    #   gradio
+    #   httpcore
+    #   uvicorn
+httpcore==0.15.0
+    # via httpx
+httpx==0.23.0
+    # via gradio
+idna==3.4
+    # via
+    #   anyio
+    #   requests
+    #   rfc3986
+    #   yarl
+jinja2==3.1.2
+    # via gradio
+kiwisolver==1.4.4
+    # via matplotlib
+linkify-it-py==1.0.3
+    # via markdown-it-py
+markdown-it-py[linkify,plugins]==2.1.0
+    # via
+    #   gradio
+    #   mdit-py-plugins
+markupsafe==2.1.1
+    # via jinja2
+matplotlib==3.6.1
+    # via gradio
+mdit-py-plugins==0.3.1
+    # via markdown-it-py
+mdurl==0.1.2
+    # via markdown-it-py
+multidict==6.0.2
+    # via
+    #   aiohttp
+    #   yarl
+numpy==1.23.3
+    # via
+    #   contourpy
+    #   gradio
+    #   matplotlib
+    #   pandas
+orjson==3.8.0
+    # via gradio
+packaging==21.3
+    # via matplotlib
+pandas==1.5.0
+    # via
+    #   -r gradio_requirements.in
+    #   gradio
+paramiko==2.11.0
+    # via gradio
+pillow==9.2.0
+    # via
+    #   gradio
+    #   matplotlib
+plotly==5.10.0
+    # via -r gradio_requirements.in
+pycparser==2.21
+    # via cffi
+pycryptodome==3.15.0
+    # via gradio
+pydantic==1.10.2
+    # via
+    #   fastapi
+    #   gradio
+pydub==0.25.1
+    # via gradio
+pynacl==1.5.0
+    # via paramiko
+pyparsing==3.0.9
+    # via
+    #   matplotlib
+    #   packaging
+python-dateutil==2.8.2
+    # via
+    #   matplotlib
+    #   pandas
+python-dotenv==0.21.0
+    # via -r gradio_requirements.in
+python-multipart==0.0.5
+    # via gradio
+pytz==2022.4
+    # via pandas
+pyyaml==6.0
+    # via gradio
+requests==2.28.1
+    # via
+    #   -r gradio_requirements.in
+    #   gradio
+rfc3986[idna2008]==1.5.0
+    # via httpx
+six==1.16.0
+    # via
+    #   paramiko
+    #   python-dateutil
+    #   python-multipart
+sniffio==1.3.0
+    # via
+    #   anyio
+    #   httpcore
+    #   httpx
+starlette==0.20.4
+    # via fastapi
+tenacity==8.1.0
+    # via plotly
+typing-extensions==4.4.0
+    # via
+    #   pydantic
+    #   starlette
+uc-micro-py==1.0.1
+    # via linkify-it-py
+urllib3==1.26.12
+    # via requests
+uvicorn==0.18.3
+    # via gradio
+websockets==10.3
+    # via gradio
+yarl==1.8.1
+    # via aiohttp