Update app.py

app.py

@@ -161,13 +161,23 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
     gallery_items = [(item[0], item[1]) for item in history]
 #    tables_html = "".join(f'<div style="margin-bottom:12px">{make_html_table(item[1])}</div>' for item in history)
     del img
+
     # Pythonのガベージコレクタを強制実行し、不要なメモリを回収します
-    import gc
-    gc.collect()
-    # GPUメモリを使用している場合、PyTorchのキャッシュをクリアします
-    if torch.cuda.is_available():
-        torch.cuda.empty_cache()
-    return img, gallery_items
+        import gc
+        gc.collect()
+        # GPUメモリを使用している場合、PyTorchのキャッシュをクリアします
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+        return img, gallery_items
+
+    except RuntimeError as e:
+        # CUDA Out of Memoryなどのエラーをキャッチ
+        error_message = f"画像生成中にエラーが発生しました: {e}\n\n"
+        if "CUDA out of memory" in str(e):
+            error_message += "GPUメモリが不足している可能性があります。画像のサイズを小さくするか、ステップ数を減らしてみてください。"
+        else:
+            error_message += "原因不明のエラーです。入力パラメータを確認してください。"
+        print(error_message) # サーバーログに出力
 
     
 # CSS 設定（ダークモード強制防止＋カフェ風テーマ）