Update app.py

app.py

@@ -3,30 +3,27 @@ from gradio_client import Client
 import os
 from dotenv import load_dotenv
 import warnings
-import torch
 
-# Suppress warnings
+# 경고 메시지 숨기기
 warnings.filterwarnings('ignore', category=UserWarning)
 
-# Load environment variables
+# 환경 변수 로드
 load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Force CPU mode
-os.environ['CUDA_VISIBLE_DEVICES'] = ''
-if torch.cuda.is_available():
-    device = torch.device('cuda')
-else:
-    device = torch.device('cpu')
-
 def generate_diagram(prompt, width=1024, height=1024):
-    """Generate a diagram using FLUX AI"""
+    """FLUX AI를 사용하여 다이어그램 생성"""
     try:
+        # API 호출을 통해 이미지 생성
+        result = gr.Image.update(value=None)  # 초기 상태는 빈 이미지
+        
+        # Hugging Face API 호출
         client = Client(
-            "black-forest-labs/FLUX.1-schnell",
+            "https://black-forest-labs-flux-1-schnell.hf.space",  # API 엔드포인트 직접 지정
             hf_token=HF_TOKEN,
-            cpu_only=True  # Force CPU mode
         )
+        
+        # 이미지 생성
         result = client.predict(
             prompt,
             1872187377,  # seed
@@ -38,7 +35,7 @@ def generate_diagram(prompt, width=1024, height=1024):
         )
         return result
     except Exception as e:
-        raise gr.Error(f"Error generating diagram: {str(e)}")
+        raise gr.Error(f"다이어그램 생성 중 오류 발생: {str(e)}")
 
 # Convert example format for Gradio
 GRADIO_EXAMPLES = [
@@ -284,52 +281,51 @@ GRADIO_EXAMPLES = [
     for example in EXAMPLES
 ]
 
-# Create Gradio interface
+
 demo = gr.Interface(
     fn=generate_diagram,
     inputs=[
         gr.Textbox(
-            label="Diagram Prompt",
-            placeholder="Enter your diagram structure...",
+            label="다이어그램 프롬프트",
+            placeholder="다이어그램 구조를 입력하세요...",
             lines=10
         ),
         gr.Slider(
-            label="Width",
+            label="너비",
             minimum=512,
             maximum=2048,
             step=128,
             value=1024
         ),
         gr.Slider(
-            label="Height",
+            label="높이",
             minimum=512,
             maximum=2048,
             step=128,
             value=1024
         )
     ],
-    outputs=gr.Image(label="Generated Diagram"),
-    title="🎨 FLUX Diagram Generator",
-    description="Generate beautiful hand-drawn style diagrams using FLUX AI",
+    outputs=gr.Image(label="생성된 다이어그램"),
+    title="🎨 FLUX 다이어그램 생성기",
+    description="FLUX AI를 사용하여 아름다운 손그림 스타일의 다이어그램을 생성합니다",
     article="""
-    ### Tips for Better Results
-    - Use clear hierarchical structures
-    - Include icon descriptions in brackets
-    - Keep text concise and meaningful
-    - Use consistent formatting
+    ### 더 나은 결과를 위한 팁
+    - 명확한 계층 구조 사용
+    - 대괄호 안에 아이콘 설명 포함
+    - 간결하고 의미 있는 텍스트 사용
+    - 일관된 형식 유지
     """,
     examples=GRADIO_EXAMPLES,
     cache_examples=True
 )
 
+# 앱 실행
 if __name__ == "__main__":
-    try:
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False,
-            show_error=True,
-            debug=True
-        )
-    except Exception as e:
-        print(f"Error launching app: {str(e)}")
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_error=True,
+        debug=True,
+        enable_queue=True  # 큐 활성화로 요청 관리
+    )