Update app.py

app.py

@@ -19,48 +19,6 @@ GALLERY_JSON = "/home/user/app/gallery.json"
 # 갤러리 디렉토리 생성
 os.makedirs(GALLERY_DIR, exist_ok=True)
 
-
-def save_to_gallery(image_path, prompt):
-    print(f"File permissions: {oct(os.stat(new_image_path).st_mode)[-3:]}")  # 로그 추가
-
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    new_image_path = os.path.join(GALLERY_DIR, f"{timestamp}.png")
-    
-    print(f"Saving image to: {new_image_path}")  # 로그 추가
-    shutil.copy2(image_path, new_image_path)
-
-    
-    # 갤러리 정보 저장
-    gallery_info = {
-        "image": new_image_path,
-        "prompt": prompt,
-        "timestamp": timestamp
-    }
-    
-    if os.path.exists(GALLERY_JSON):
-        with open(GALLERY_JSON, "r") as f:
-            gallery = json.load(f)
-    else:
-        gallery = []
-    
-    gallery.append(gallery_info)
-    
-    with open(GALLERY_JSON, "w") as f:
-        json.dump(gallery, f, indent=2)
-    
-    return new_image_path
-
-def load_gallery():
-    if os.path.exists(GALLERY_JSON):
-    print(f"Gallery JSON permissions: {oct(os.stat(GALLERY_JSON).st_mode)[-3:]}")  # 로그 추가
-   
-    print(f"Loading gallery from: {GALLERY_JSON}")  # 로그 추가
-    if os.path.exists(GALLERY_JSON):
-        with open(GALLERY_JSON, "r") as f:
-            gallery = json.load(f)
-        return [(item["image"], item["prompt"]) for item in reversed(gallery)]
-    return []
-
 def respond(message, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     logging.info(f"Received message: {message}, seed: {seed}, randomize_seed: {randomize_seed}, "
                  f"width: {width}, height: {height}, guidance_scale: {guidance_scale}, "
@@ -187,14 +145,66 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
         outputs=gallery
     )
 
+def load_gallery():
+    print(f"Loading gallery from: {GALLERY_JSON}")  # 로그 추가
+    if os.path.exists(GALLERY_JSON):
+        print(f"Gallery JSON permissions: {oct(os.stat(GALLERY_JSON).st_mode)[-3:]}")  # 로그 추가
+        try:
+            with open(GALLERY_JSON, "r") as f:
+                gallery = json.load(f)
+            return [(item["image"], item["prompt"]) for item in reversed(gallery)]
+        except Exception as e:
+            print(f"Error loading gallery: {str(e)}")
+    return []
+
 def check_gallery_status():
     print(f"Gallery directory exists: {os.path.exists(GALLERY_DIR)}")
     print(f"Gallery JSON exists: {os.path.exists(GALLERY_JSON)}")
     if os.path.exists(GALLERY_JSON):
-        with open(GALLERY_JSON, "r") as f:
-            gallery = json.load(f)
-        print(f"Number of items in gallery: {len(gallery)}")
+        try:
+            with open(GALLERY_JSON, "r") as f:
+                gallery = json.load(f)
+            print(f"Number of items in gallery: {len(gallery)}")
+        except Exception as e:
+            print(f"Error reading gallery JSON: {str(e)}")
+
+def save_to_gallery(image_path, prompt):
+    try:
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        new_image_path = os.path.join(GALLERY_DIR, f"{timestamp}.png")
+        
+        print(f"Saving image to: {new_image_path}")  # 로그 추가
+        shutil.copy2(image_path, new_image_path)
+        
+        print(f"File permissions: {oct(os.stat(new_image_path).st_mode)[-3:]}")  # 로그 추가
 
+        gallery_info = {
+            "image": new_image_path,
+            "prompt": prompt,
+            "timestamp": timestamp
+        }
+        
+        if os.path.exists(GALLERY_JSON):
+            with open(GALLERY_JSON, "r") as f:
+                gallery = json.load(f)
+        else:
+            gallery = []
+        
+        gallery.append(gallery_info)
+        
+        with open(GALLERY_JSON, "w") as f:
+            json.dump(gallery, f, indent=2)
+        
+        return new_image_path
+    except Exception as e:
+        print(f"Error saving to gallery: {str(e)}")
+        raise
+
+# 메인 부분에 다음 코드 추가
 if __name__ == "__main__":
+    print("Environment variables:")
+    for key, value in os.environ.items():
+        print(f"{key}: {value}")
+    
     check_gallery_status()
     demo.launch()