Update app.py

app.py

@@ -11,7 +11,7 @@ app = Flask(__name__)
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
-timeout = 100
+timeout = 300  # タイムアウトを300秒に設定
 
 # Function to query the API and return the generated image
 def query(prompt, negative_prompt="", steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
@@ -41,17 +41,24 @@ def query(prompt, negative_prompt="", steps=35, cfg_scale=7, sampler="DPM++ 2M K
         }
     }
 
-    # Send the request to the API
-    try:
-        response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
-        if response.status_code != 200:
-            return None, f"Error: Failed to get image. Status code: {response.status_code}, Details: {response.text}"
-        
-        image_bytes = response.content
-        image = Image.open(io.BytesIO(image_bytes))
-        return image, None
-    except Exception as e:
-        return None, f"Error when trying to open the image: {e}"
+    for attempt in range(3):  # 最大3回の再試行
+        try:
+            response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
+            if response.status_code != 200:
+                return None, f"Error: Failed to get image. Status code: {response.status_code}, Details: {response.text}"
+
+            image_bytes = response.content
+            image = Image.open(io.BytesIO(image_bytes))
+            return image, None
+        except requests.exceptions.Timeout:
+            if attempt < 2:  # 最後の試行でない場合は再試行
+                print("Timeout occurred, retrying...")
+                continue
+            return None, "Error: The request timed out. Please try again."
+        except requests.exceptions.RequestException as e:
+            return None, f"Request Exception: {str(e)}"
+        except Exception as e:
+            return None, f"Error when trying to open the image: {e}"
 
 # HTML template for the index page
 index_html = """
@@ -141,7 +148,7 @@ def generate_image():
     image_filename = f"generated_image_{random.randint(1, 10000)}.png"
     
     # Save the image to a file
-    image.save(image_filename)
+    image.save(os.path.join('static', image_filename))
 
     # Return the index page with the generated image URL
     image_url = f"/static/{image_filename}"