Update app-backup.py

app-backup.py

@@ -12,6 +12,7 @@ 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")
 
@@ -35,6 +36,33 @@ torch.backends.cuda.matmul.allow_tf32 = True
 if not path.exists(gallery_path):
     os.makedirs(gallery_path, exist_ok=True)
 
+# 샘플 이미지와 프롬프트 정의
+SAMPLE_IMAGES = {
+    "3d2.webp": "the most famous hero according to Yuri Milner",
+    "3d3.webp": "purple nest",
+    "3d4.webp": "Timothy's sabbath",
+    "3d5.webp": "A schoolboy friend of Julián Carax, fun-loving and loyal",
+    "3d6.webp": "Friend of Daniel and his father",
+    "3d7.webp": "WHERE ships of purple gently toss On seas of daffodil",
+    "3d8.webp": "Beat the drums of tragedy for me, And let the white violins whir thin and slow",
+    "3d9.webp": "And let the choir sing a stormy song To drown the rattle of my dying breath.",
+    "3d10.webp": "Beat the drums of tragedy and death",
+    "3d11.webp": "Beat the drums of tragedy for me.",
+    "3d12.webp": "Touching the infinite, else far and untrod, With oracles divine that speak of God.",
+    "3d13.webp": "Night, standing on her starry pulpit, free, Utters them in the dread, the silver roll Of spheres, woods, winds and waves, alternately",
+    "3d14.webp": "On sermons deep, fit time to feast the soul.",
+    "3d15.webp": "The bee is cradled in the bud; and far, Cold glittering lights, the azure curtain, throng— Planet on beaming planet, star on star.",
+    "3d16.webp": "The lark's sweet pipe has ceased its latest song",
+    "3d17.webp": "the snake was a roaming dog",
+    "3d18.webp": "Antonio Battistella portraying Father of Giulia",
+    "3d19.webp": "So straight to her father the brisk young lady went, And said, grant me one favour, do give your consent",
+    "3d20.webp": "Before that we are marry’d let me your father see, All fear is, now miscarry’d, my heart is full of glee",
+    "3d21.webp": "My heart you now have gained, you are all I prize, So make yourself contented, pray be satisfied.",
+    "3d22.webp": "O pray what is the favour that of me you crave? If it lies in my power you the same shall have",
+    "3d23.webp": "Could I but see your father, and my mind reveal, I have both gold and silver, and houses at my will",
+    "3d1.webp": "the most famous hero according to Zhou Qi"
+}
+
 class timer:
     def __init__(self, method_name="timed process"):
         self.method = method_name
@@ -49,11 +77,10 @@ class timer:
 if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
 
-# 인증된 모델 로드
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     torch_dtype=torch.bfloat16,
-    use_auth_token=HF_TOKEN
+    use_auth_token=HF_TOKEN  # 경고 메시지가 뜨지만 무시 가능
 )
 
 # Hyper-SD LoRA 로드
@@ -98,26 +125,15 @@ def save_image(image):
         print(f"Error in save_image: {str(e)}")
         return None
 
-# 예시 프롬프트 정의
-examples = [
-    ["A 3D Star Wars Darth Vader helmet, highly detailed metallic finish"],
-    ["A 3D Iron Man mask with glowing eyes and metallic red-gold finish"],
-    ["A detailed 3D Pokemon Pikachu figure with glossy surface"],
-    ["A 3D geometric abstract cube transforming into a sphere, metallic finish"],
-    ["A 3D steampunk mechanical heart with brass and copper details"],
-    ["A 3D crystal dragon with transparent iridescent scales"],
-    ["A 3D futuristic hovering drone with neon light accents"],
-    ["A 3D ancient Greek warrior helmet with ornate details"],
-    ["A 3D robotic butterfly with mechanical wings and metallic finish"],
-    ["A 3D floating magical crystal orb with internal energy swirls"]
-]
+def get_random_seed():
+    return torch.randint(0, 1000000, (1,)).item()
 
 @spaces.GPU
 def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt="", seed=None):
     global pipe
     
     if seed is None:
-        seed = torch.randint(0, 1000000, (1,)).item()
+        seed = get_random_seed()
     
     # 한글 감지 및 번역
     def contains_korean(text):
@@ -128,7 +144,7 @@ def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt=
         translated = translator(prompt)[0]['translation_text']
         prompt = translated
     
-    formatted_prompt = f"wbgmsst, 3D, {prompt} ,white background"
+    formatted_prompt = f"wbgmsst, 3D, {prompt}, white background"
     
     with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16), timer("inference"):
         try:
@@ -151,21 +167,10 @@ def process_and_save_image(height=1024, width=1024, steps=8, scales=3.5, prompt=
             print(f"Error in image generation: {str(e)}")
             return None
 
-def get_random_seed():
-    return torch.randint(0, 1000000, (1,)).item()
+def update_random_seed():
+    """버튼으로 눌렀을 때 새로운 시드를 업데이트"""
+    return gr.update(value=get_random_seed())
 
-    
-def process_example(prompt):
-    return process_and_save_image(
-        height=1024,
-        width=1024,
-        steps=8,
-        scales=3.5,
-        prompt=prompt,
-        seed=get_random_seed()
-    )
-
-    
 # Gradio 인터페이스
 with gr.Blocks(
     theme=gr.themes.Soft(),
@@ -190,13 +195,27 @@ with gr.Blocks(
             max-width: 1024px;
             margin: auto;
         }
+        .gallery-container {
+            margin-top: 40px;
+            padding: 20px;
+            background: #f5f5f5;
+            border-radius: 15px;
+            width: 100%;
+            margin: 0 auto;
+        }
+        .gallery-title {
+            text-align: center;
+            margin-bottom: 20px;
+            color: #333;
+            font-size: 1.5rem;
+        }
     """
 ) as demo:
     gr.HTML(
         """
         <div style="text-align: center; max-width: 800px; margin: 0 auto; padding: 20px;">
-            <h1 style="font-size: 2.5rem; color: #2196F3;">3D Style Image Generator</h1>
-            <p style="font-size: 1.2rem; color: #666;">Create amazing 3D-style images with AI</p>
+            <h1 style="font-size: 2.5rem; color: #2196F3;">3D Style Image Generator v2.0</h1>
+            <p style="font-size: 1.2rem; color: #666;">Create amazing 3D-style images with AI. https://discord.gg/openfreeai </p>
         </div>
         """
     )
@@ -263,33 +282,64 @@ with gr.Blocks(
                 value="3d.webp"
             )
 
-    # Examples 섹션
-    gr.Examples(
-        examples=examples,
-        inputs=prompt,
-        outputs=output,
-        fn=process_example,  # 수정된 함수 사용
-        cache_examples=False,
-        examples_per_page=5
-    )
-
-    def update_seed():
-        return get_random_seed()
-
+    # gallery-container 부분을 Group으로 감싸 화면 전체에 확장
+    with gr.Group(elem_classes="gallery-container"):
+        gr.HTML("<h2 class='gallery-title'>Gallery</h2>")
+        
+        gallery_html = """
+        <div style='
+            display: grid; 
+            grid-template-columns: repeat(4, 1fr); 
+            gap: 20px;
+            width: 100%;
+        '>
+        """
+        
+        for img_file, prompt_text in SAMPLE_IMAGES.items():
+            img_path = os.path.abspath(img_file)
+            if os.path.exists(img_path):
+                try:
+                    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;
+                        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_text}</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(
-        process_and_save_image,
+        fn=process_and_save_image,
         inputs=[height, width, steps, scales, prompt, seed],
         outputs=output
-    ).then(
-        update_seed,
-        outputs=[seed]
     )
 
     randomize_seed.click(
-        update_seed,
-        outputs=[seed]
+        fn=update_random_seed,
+        inputs=None,
+        outputs=seed
     )
 
 if __name__ == "__main__":
-    demo.launch(allowed_paths=[PERSISTENT_DIR])
+    demo.launch(allowed_paths=[PERSISTENT_DIR])