Give perspective choice

app.py

@@ -2,17 +2,24 @@ import os
 import gradio as gr
 import googlemaps
 from skimage import io
-
+from urllib import parse
 from inferences import ClimateGAN
 
 API_KEY = os.environ.get("API_KEY")
 gmaps = googlemaps.Client(key=API_KEY)
 model = ClimateGAN(model_path="config/model/masker")
 
-def predict(place):
+def predict(place, perspective):
     geocode_result = gmaps.geocode(place)
-    loc = geocode_result[0]['geometry']['location']
-    static_map_url = f"https://maps.googleapis.com/maps/api/streetview?size=640x640&location={loc['lat']},{loc['lng']}&source=outdoor&key={API_KEY}"
+
+    static_map_url = ""
+
+    if perspective == "A":
+        address = geocode_result[0]['formatted_address']
+        static_map_url = f"https://maps.googleapis.com/maps/api/streetview?size=640x640&location={parse.quote(address)}&source=outdoor&key={API_KEY}"
+    else:
+        loc = geocode_result[0]['geometry']['location']
+        static_map_url = f"https://maps.googleapis.com/maps/api/streetview?size=640x640&location={loc['lat']},{loc['lng']}&source=outdoor&key={API_KEY}"
 
     img_np = io.imread(static_map_url)
     flood, wildfire, smog = model.inference(img_np)
@@ -22,7 +29,8 @@ def predict(place):
 gr.Interface(
     predict,
     inputs=[
-        gr.inputs.Textbox(label="Address or place name")
+        gr.inputs.Textbox(label="Address or place name"),
+        gr.inputs.Radio(["A", "B"], label="Perspective")
     ],
     outputs=[
         gr.outputs.Image(type="numpy", label="Original image"),
@@ -31,12 +39,12 @@ gr.Interface(
         gr.outputs.Image(type="numpy", label="Smog"),
     ],
     title="ClimateGAN",
-    description="Enter an address or place name, and ClimateGAN will generate images showing how the location could be impacted by flooding, wildfires, or smog.",
+    description="Enter an address or place name, and ClimateGAN will generate images showing how the location could be impacted by flooding, wildfires, or smog. Changing the \"perspective\" will give you a different image of the same location.",
     article="<p style='text-align: center'>This project is a clone of <a href='https://thisclimatedoesnotexist.com/'>ThisClimateDoesNotExist</a> | <a href='https://github.com/cc-ai/climategan'>ClimateGAN GitHub Repo</a></p>",
     examples=[
-        "Vancouver Art Gallery",
-        "Chicago Bean",
-        "Duomo Siracusa"
+        ["Vancouver Art Gallery", "A"],
+        ["Chicago Bean", "B"],
+        ["Duomo Siracusa", "A"],
     ],
     css=".footer{display:none !important}",
 ).launch(cache_examples=True)