Update app.py

app.py

@@ -1,103 +1,3 @@
-import os
-import sys
-from torchvision.transforms import functional
-sys.modules["torchvision.transforms.functional_tensor"] = functional
-
-from basicsr.archs.srvgg_arch import SRVGGNetCompact
-from gfpgan.utils import GFPGANer
-from realesrgan.utils import RealESRGANer
-
-import torch
-import cv2
-import gradio as gr
-
-# 필요한 모델 다운로드
-def download_upscaler_models():
-    if not os.path.exists('realesr-general-x4v3.pth'):
-        os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth -P .")
-    if not os.path.exists('GFPGANv1.4.pth'):
-        os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -P .")
-
-# 업스케일러 모델 초기화
-def init_upscaler_model():
-    model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, upscale=4, act_type='prelu')
-    model_path = 'realesr-general-x4v3.pth'
-    half = True if torch.cuda.is_available() else False
-    upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
-    
-    return upsampler
-
-# 이미지 업스케일링 함수
-def upscale_image(img):
-    try:
-        upsampler = init_upscaler_model()
-        
-        # 이미지 파일 읽기
-        img = cv2.imread(img, cv2.IMREAD_UNCHANGED)
-        if len(img.shape) == 3 and img.shape[2] == 4:
-            img_mode = 'RGBA'
-        elif len(img.shape) == 2:
-            img_mode = None
-            img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
-        else:
-            img_mode = None
-
-        # 작은 이미지 크기 조정
-        h, w = img.shape[0:2]
-        if h < 300:
-            img = cv2.resize(img, (w * 2, h * 2), interpolation=cv2.INTER_LANCZOS4)
-
-        # 고정된 GFPGANv1.4 모델 사용
-        face_enhancer = GFPGANer(
-            model_path='GFPGANv1.4.pth', 
-            upscale=2, 
-            arch='clean',
-            channel_multiplier=2,
-            bg_upsampler=upsampler
-        )
-
-        # 이미지 향상
-        try:
-            _, _, output = face_enhancer.enhance(img, has_aligned=False, only_center_face=False, paste_back=True)
-        except RuntimeError as error:
-            print('오류', error)
-            return None
-
-        # RGB로 변환하여 반환
-        output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
-        return output
-    except Exception as error:
-        print('글로벌 예외', error)
-        return None
-
-# 업스케일러 탭 UI 생성 함수
-def create_upscaler_tab():
-    # 필요한 모델 다운로드
-    download_upscaler_models()
-    
-    with gr.Tab("이미지 업스케일러"):
-        gr.Markdown(
-            """
-            # 이미지 업스케일러
-            
-            업로드한 이미지를 고해상도로 변환합니다. 특히 인물 사진의 얼굴 디테일을 향상시킵니다.
-            """
-        )
-        
-        with gr.Row():
-            with gr.Column():
-                input_image = gr.Image(type="filepath", label="입력 이미지")
-                upscale_btn = gr.Button("이미지 업스케일", variant="primary")
-            
-            with gr.Column():
-                output_image = gr.Image(type="numpy", label="결과 이미지")
-        
-        upscale_btn.click(
-            fn=upscale_image,
-            inputs=[input_image],
-            outputs=[output_image]
-        )
-
 import os
 import tempfile
 from PIL import Image
@@ -130,6 +30,13 @@ load_dotenv()
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
+# 필요한 모델 다운로드
+def download_upscaler_models():
+    if not os.path.exists('realesr-general-x4v3.pth'):
+        os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth -P .")
+    if not os.path.exists('GFPGANv1.4.pth'):
+        os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -P .")
+
 def save_binary_file(file_name, data):
     with open(file_name, "wb") as f:
         f.write(data)
@@ -411,13 +318,7 @@ def generate_multiple_images(image1, image2, image3, prompt, progress=gr.Progres
     
     return results[0], results[1], results[2], results[3], combined_status, combined_prompts
 
-# 이미지 업스케일러 기능
-def download_upscaler_models():
-    if not os.path.exists('realesr-general-x4v3.pth'):
-        os.system("wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.5.0/realesr-general-x4v3.pth -P .")
-    if not os.path.exists('GFPGANv1.4.pth'):
-        os.system("wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth -P .")
-
+# 이미지 업스케일 함수
 def upscale_image(img):
     try:
         # 모델 초기화
@@ -464,6 +365,9 @@ def upscale_image(img):
         print('글로벌 예외', error)
         return None
 
+# 시작 시 업스케일러 모델 다운로드
+download_upscaler_models()
+
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.HTML(
@@ -475,7 +379,7 @@ with gr.Blocks() as demo:
         """
     )
     
-    with gr.Tabs():
+    with gr.Tabs() as tabs:
         # 첫 번째 탭: 이미지 생성기
         with gr.Tab("이미지 생성기"):
             with gr.Row():
@@ -532,8 +436,7 @@ with gr.Blocks() as demo:
                     
                     gr.Examples(
                         examples=examples,
-                        inputs=[image1_input, image2_input, image3_input, prompt_input],
-                        elem_id="examples-grid"
+                        inputs=[image1_input, image2_input, image3_input, prompt_input]
                     )
                     
                 # 오른쪽 출력 영역
@@ -571,7 +474,7 @@ with gr.Blocks() as demo:
                 outputs=prompt_input
             )
             clothes_change_btn1.click(
-                fn=lambda: "(#1의 여성모겔)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
+                fn=lambda: "(#1의 여성모델)이 신체 비율과 포즈는 유치한 체 (#2의 선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미지를 생성하라.",
                 inputs=[],
                 outputs=prompt_input
             )
@@ -604,9 +507,6 @@ with gr.Blocks() as demo:
         
         # 두 번째 탭: 이미지 업스케일러
         with gr.Tab("이미지 업스케일러"):
-            # 필요한 모델 다운로드 호출
-            download_upscaler_models()
-            
             gr.Markdown(
                 """
                 # 이미지 업스케일러
@@ -630,4 +530,4 @@ with gr.Blocks() as demo:
             )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)선글라스)와 (#3의 청바지)를 직접 모델이 착용한것 처럼 자연스럽게 이미