Upload app.py

app.py

@@ -1,4 +1,6 @@
 import requests
+from PIL import Image
+import base64
 import gradio as gr
 
 def analyze_images(image1, image2, api_key, org_id):
@@ -8,8 +10,6 @@ def analyze_images(image1, image2, api_key, org_id):
         "Authorization": api_key,
     }
 
-    import base64
-
     with open(image1, "rb") as img1_file:
         image1_b64 = base64.b64encode(img1_file.read()).decode('utf-8')
 
@@ -28,7 +28,12 @@ def analyze_images(image1, image2, api_key, org_id):
         },
     }
 
-    response = requests.post(url, headers=headers, json=data).json()['entry']
+    response = requests.post(url, headers=headers, json=data).json()
+
+    if 'entry' not in response.keys():
+        return f"Request failed: {response}"
+
+    response = response['entry']
     print(response.keys())
 
     if response['status'] == 'success':
@@ -36,19 +41,27 @@ def analyze_images(image1, image2, api_key, org_id):
     else:
         return f"Request failed: {response['text']}"
 
+
+def assign_sample_images():
+    image1 = Image.open('./front.jpg')
+    image2 = Image.open('./right.jpg')
+    return image1, image2
+
 with gr.Blocks() as demo:
     gr.Markdown("## 画像解析アプリ")
     
     with gr.Row():
-        image1 = gr.Image(type="filepath", label="前からの写真")
-        image2 = gr.Image(type="filepath", label="右側からの写真")
+        image1 = gr.Image(type="filepath", label="前からの写真", format='jpeg')
+        image2 = gr.Image(type="filepath", label="右側からの写真", format='jpeg')
     
     api_key = gr.Textbox(label="APIキーを入力")
     org_id = gr.Textbox(label="組織IDを入力")
     
+    sample_button = gr.Button("サンプル画像")
     analyze_button = gr.Button("解析", variant='primary')
     result = gr.TextArea(label="結果表示")
     
+    sample_button.click(assign_sample_images, inputs=[], outputs=[image1, image2])
     analyze_button.click(analyze_images, inputs=[image1, image2, api_key, org_id], outputs=[result])
 
 if __name__ == '__main__':