Update app.py

app.py

@@ -109,12 +109,8 @@ euler_scheduler = EulerAncestralDiscreteScheduler.from_config(
 dpm_scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.scheduler = euler_scheduler
 
-def upload_image_to_hub(image_pil, prompt_text):
+def upload_image_to_hub(image_pil, prompt_text, filename)):
     # ファイル名を生成（タイムスタンプとプロンプトの一部）
-    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}.png"
     filepath = f"temp_{filename}"
     image_pil.save(filepath)
 
@@ -159,10 +155,9 @@ def upload_image_to_hub(image_pil, prompt_text):
     finally:
         pass
 
-def upload_image_to_private_hub(image_pil, prompt_text):
-    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
-    filename = f"image_{timestamp}.png"
-    filepath = f"temp_private_{filename}" # 一時ファイル名が重複しないように private を追加
+def upload_image_to_private_hub(image_pil, prompt_text, filename):
+
+    filepath = f"temp_private_{filename}"
     image_pil.save(filepath)
 
     try:
@@ -262,9 +257,9 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
         caption_text_for_history = caption_text.replace("\n", "|-|").strip()
         # 画像をHubにアップロードし、そのURLとリポジトリ内パスを取得
         # 公開用リポジトリにアップロード
-        uploaded_image_url, path_in_public_repo_for_history = upload_image_to_hub(img, caption_text)
+        uploaded_image_url, path_in_public_repo_for_history = upload_image_to_hub(img, caption_text, filename)
         # 非公開リポジトリにアップロード
-        path_in_private_repo_for_history = upload_image_to_private_hub(img, caption_text)
+        path_in_private_repo_for_history = upload_image_to_private_hub(img, caption_text, filename)
     
 
         # 履歴を更新