Update app.py

app.py

@@ -31,6 +31,7 @@ def respond(message, history):
     #prompt = list(itertools.chain.from_iterable(history))
     #prompt.append(message)
 
+    # プロンプトの作成
     prompt = pd.DataFrame(
         {"prompt": [message], "num_inference_steps": 25}
     )
@@ -41,31 +42,40 @@ def respond(message, history):
 
     embed_image_markdown = ""
 
-    response = requests.request(method="POST", headers=headers, url=local_endpoint, data=data_json)    
-    response_data = response.json()
-    #print(response_data["predictions"])
-    
-    im_array = np.array(response_data["predictions"], dtype=np.uint8)
-    #print(im_array)
-    im = Image.fromarray(im_array, 'RGB')
+    try:
+        # モデルサービングエンドポイントに問い合わせ
+        response = requests.request(method="POST", headers=headers, url=local_endpoint, data=data_json)    
+        response_data = response.json()
+        #print(response_data["predictions"])
+
+        # numpy arrayに変換
+        im_array = np.array(response_data["predictions"], dtype=np.uint8)
+        #print(im_array)
+        # 画像に変換
+        im = Image.fromarray(im_array, 'RGB')
 
     
-    # debug
-    #image_url = "https://upload.wikimedia.org/wikipedia/commons/thumb/e/ec/Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg/687px-Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg"
-    #print("image_url:", image_url)
-    #im = Image.open(io.BytesIO(requests.get(image_url).content))
-    #numpydata = np.asarray(im)
-
-    rawBytes = io.BytesIO()
-    im.save(rawBytes, "PNG")
-    rawBytes.seek(0)  # ファイルの先頭に移動
-    image_encoded = base64.b64encode(rawBytes.read()).decode('ascii')
-    #print(image_encoded)
-        
-    embed_image_markdown = f"![](data:image/png;base64,{image_encoded})"
-    #print(embed_image_markdown)
-
-    # print(response.json())
+        # debug
+        #image_url = "https://upload.wikimedia.org/wikipedia/commons/thumb/e/ec/Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg/687px-Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg"
+        #print("image_url:", image_url)
+        #im = Image.open(io.BytesIO(requests.get(image_url).content))
+        #numpydata = np.asarray(im)
+
+        rawBytes = io.BytesIO()
+        im.save(rawBytes, "PNG")
+        rawBytes.seek(0)  # ファイルの先頭に移動
+        # base64にエンコード
+        image_encoded = base64.b64encode(rawBytes.read()).decode('ascii')
+        #print(image_encoded)
+
+        # マークダウンに埋め込み
+        embed_image_markdown = f"![](data:image/png;base64,{image_encoded})"
+        #print(embed_image_markdown)
+    
+    except Exception as error:
+        response_data = f"ERROR status_code: {type(error).__name__}" 
+        #+ str(response.status_code) + " response:" + response.text    
+
     return embed_image_markdown