Update app.py

app.py

@@ -12,6 +12,8 @@ import torch
 from diffusers import FluxPipeline
 from PIL import Image
 from transformers import pipeline
+import base64
+
 
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
@@ -268,37 +270,47 @@ with gr.Blocks(
                 value="3d.webp"
             )
 
-    # Gallery 섹션
+# Gallery 섹션
     with gr.Row(elem_classes="gallery-container"):
         gr.HTML("<h2 class='gallery-title'>Gallery</h2>")
         gallery_html = "<div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>"
         
-        for img_file, sample_prompt in SAMPLE_IMAGES.items():
-            if os.path.exists(img_file):
-                gallery_html += f"""
-                <div style='
-                    border: 1px solid #ddd;
-                    border-radius: 10px;
-                    padding: 10px;
-                    background: white;
-                    box-shadow: 0 4px 8px rgba(0,0,0,0.1);
-                '>
-                    <img src='file/{img_file}' 
-                        style='width: 100%; 
-                        border-radius: 8px;
-                        margin-bottom: 10px;'
-                    >
-                    <p style='
-                        margin: 5px 0;
-                        font-weight: bold;
-                        color: #333;
+        for img_file, prompt in SAMPLE_IMAGES.items():
+            img_path = os.path.abspath(img_file)  # 절대 경로로 변환
+            if os.path.exists(img_path):
+                try:
+                    # Base64로 이미지 인코딩
+                    with open(img_path, "rb") as img:
+                        img_data = base64.b64encode(img.read()).decode()
+                    
+                    gallery_html += f"""
+                    <div style='
+                        border: 1px solid #ddd;
+                        border-radius: 10px;
                         padding: 10px;
-                    '>Prompt: {sample_prompt}</p>
-                </div>
-                """
+                        background: white;
+                        box-shadow: 0 4px 8px rgba(0,0,0,0.1);
+                    '>
+                        <img src='data:image/webp;base64,{img_data}'
+                            style='width: 100%; 
+                            border-radius: 8px;
+                            margin-bottom: 10px;'
+                        >
+                        <p style='
+                            margin: 5px 0;
+                            font-weight: bold;
+                            color: #333;
+                            padding: 10px;
+                        '>Prompt: {prompt}</p>
+                    </div>
+                    """
+                except Exception as e:
+                    print(f"Error loading image {img_file}: {str(e)}")
         
         gallery_html += "</div>"
         gr.HTML(gallery_html)
+    
+
 
     # 이벤트 바인딩
     generate_btn.click(