Update app.py

app.py

@@ -48,33 +48,33 @@ HF_REPO_ID = "cocoat/Re.cocoamixXL3"
 
 def upload_image_to_hub(image_pil, prompt_text):
 # ファイル名を生成（タイムスタンプとプロンプトの一部）
-timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
 # プロンプトから安全なファイル名の一部を生成
-safe_prompt = "".join(c for c in prompt_text if c.isalnum() or c in (' ', '.', '_')).replace(' ', '_')[:30]
-filename = f"image_{timestamp}_{safe_prompt}.png"
-filepath = f"temp_{filename}" # 一時的にローカルに保存するパス
+    safe_prompt = "".join(c for c in prompt_text if c.isalnum() or c in (' ', '.', '_')).replace(' ', '_')[:30]
+    filename = f"image_{timestamp}_{safe_prompt}.png"
+    filepath = f"temp_{filename}" # 一時的にローカルに保存するパス
 # PIL画像を一時ファイルとして保存
-image_pil.save(filepath)
+    image_pil.save(filepath)
 
 # Hubにアップロード
-try:
+    try:
 # リポジトリ内にディレクトリを作成する場合は path_in_repo を使う
-path_in_repo = f"generated_images/{filename}"
-upload_info = api.upload_file(
-path_or_fileobj=filepath,
-path_in_repo=path_in_repo,
-repo_id=HF_REPO_ID,
-repo_type="dataset", # または "space", "model" など、目的のリポジトリタイプ
+    path_in_repo = f"generated_images/{filename}"
+    upload_info = api.upload_file(
+    path_or_fileobj=filepath,
+    path_in_repo=path_in_repo,
+    repo_id=HF_REPO_ID,
+    repo_type="dataset", # または "space", "model" など、目的のリポジトリタイプ
 # 通常、画像保存には "dataset" タイプのリポジトリが適しています
-)
-print(f"Uploaded {filepath} to {upload_info.url}")
-return upload_info.url # アップロードされたファイルのURLを返す
-except Exception as e:
-print(f"Error uploading image to Hub: {e}")
-return None
-finally:
+    )
+    print(f"Uploaded {filepath} to {upload_info.url}")
+    return upload_info.url # アップロードされたファイルのURLを返す
+    except Exception as e:
+    print(f"Error uploading image to Hub: {e}")
+    return None
+    finally:
 # 一時ファイルを削除
-if os.path.exists(filepath):
+    if os.path.exists(filepath):
 os.remove(filepath)