Update app.py

app.py

@@ -24,10 +24,17 @@ def respond(message):
             num_inference_steps=28,
             api_name="/infer_t2i"
         )
-        logging.info("Image generated: %s", result['url'])
+        logging.info("API response received: %s", result)
         
-        # 결과 이미지 반환
-        return result['url']
+        # 결과 확인 및 처리
+        if isinstance(result, dict) and 'url' in result:
+            return result['url']
+        elif isinstance(result, tuple):
+            logging.error("Unexpected tuple response: %s", result)
+            # 튜플 응답 처리, 예를 들어 result[0] 이 url을 포함하고 있다고 가정
+            return result[0]
+        else:
+            raise ValueError("Unexpected API response format")
     except Exception as e:
         logging.error("Error during API request: %s", str(e))
         return "Failed to generate image due to an error."