Update app-backup.py

app-backup.py

@@ -16,7 +16,7 @@ import modelscope_studio.components.legacy as legacy
 import modelscope_studio.components.antd as antd
 import html
 import urllib.parse
-from huggingface_hub import HfApi, create_repo
+from huggingface_hub import HfApi, create_repo, hf_hub_download
 import string
 import requests
 from selenium import webdriver
@@ -27,7 +27,64 @@ from selenium.common.exceptions import WebDriverException, TimeoutException
 from PIL import Image
 from io import BytesIO
 from datetime import datetime
+import spaces
+from safetensors.torch import load_file
+from diffusers import FluxPipeline
+import torch
+from os import path  # 이 줄을 추가
 
+# 캐시 경로 설정
+cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
+os.environ["TRANSFORMERS_CACHE"] = cache_path
+os.environ["HF_HUB_CACHE"] = cache_path
+os.environ["HF_HOME"] = cache_path
+
+
+# Hugging Face 토큰 설정
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    print("Warning: HF_TOKEN not found in environment variables")
+
+# FLUX 모델 초기화
+if not path.exists(cache_path):
+    os.makedirs(cache_path, exist_ok=True)
+
+try:
+    pipe = FluxPipeline.from_pretrained(
+        "black-forest-labs/FLUX.1-dev",
+        torch_dtype=torch.bfloat16,
+        use_auth_token=HF_TOKEN  # Hugging Face 토큰 추가
+    )
+    pipe.load_lora_weights(
+        hf_hub_download(
+            "ByteDance/Hyper-SD",
+            "Hyper-FLUX.1-dev-8steps-lora.safetensors",
+            token=HF_TOKEN  # Hugging Face 토큰 추가
+        )
+    )
+    pipe.fuse_lora(lora_scale=0.125)
+    pipe.to(device="cuda", dtype=torch.bfloat16)
+    print("Successfully initialized FLUX model with authentication")
+except Exception as e:
+    print(f"Error initializing FLUX model: {str(e)}")
+    pipe = None
+    
+
+
+# 이미지 생성 함수 추가
+@spaces.GPU
+def generate_image(prompt, height=512, width=512, steps=8, scales=3.5, seed=3413):
+    with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
+        return pipe(
+            prompt=[prompt],
+            generator=torch.Generator().manual_seed(int(seed)),
+            num_inference_steps=int(steps),
+            guidance_scale=float(scales),
+            height=int(height),
+            width=int(width),
+            max_sequence_length=256
+        ).images[0]
+        
 # SystemPrompt 부분을 직접 정의
 SystemPrompt = """You are 'MOUSE-I', an advanced AI visualization expert. Your mission is to transform every response into a visually stunning and highly informative presentation.
 
@@ -211,7 +268,7 @@ async def try_claude_api(system_message, claude_messages, timeout=15):
         start_time = time.time()
         with claude_client.messages.stream(
             model="claude-3-5-sonnet-20241022",
-            max_tokens=7800,
+            max_tokens=7860,
             system=system_message,
             messages=claude_messages
         ) as stream:
@@ -256,35 +313,36 @@ class Demo:
     def __init__(self):
         pass
 
-    async def generation_code(self, query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
+    async def generation_code(self, query: Optional[str], _setting: Dict[str, str]):
         if not query or query.strip() == '':
             query = get_random_placeholder()
         
-        if _history is None:
-            _history = []
+        # 이미지 생성이 필요한지 확인
+        needs_image = '이미지' in query or '그림' in query or 'image' in query.lower()
+        image_prompt = None
         
-        messages = history_to_messages(_history, _setting['system'])
-        system_message = messages[0]['content']
+        # 이미지 프롬프트 추출
+        if needs_image:
+            for keyword in ['이미지:', '그림:', 'image:']:
+                if keyword in query.lower():
+                    image_prompt = query.split(keyword)[1].strip()
+                    break
+            if not image_prompt:
+                image_prompt = query  # 명시적 프롬프트가 없으면 전체 쿼리 사용
         
-        claude_messages = [
-            {"role": msg["role"] if msg["role"] != "system" else "user", "content": msg["content"]} 
-            for msg in messages[1:] + [{'role': Role.USER, 'content': query}]
-            if msg["content"].strip() != ''
-        ]
+        messages = [{'role': Role.SYSTEM, 'content': _setting['system']}]
+        messages.append({'role': Role.USER, 'content': query})
         
-        openai_messages = [{"role": "system", "content": system_message}]
-        for msg in messages[1:]:
-            openai_messages.append({
-                "role": msg["role"],
-                "content": msg["content"]
-            })
-        openai_messages.append({"role": "user", "content": query})
+        system_message = messages[0]['content']
+        claude_messages = [{"role": "user", "content": query}]
+        openai_messages = [
+            {"role": "system", "content": system_message},
+            {"role": "user", "content": query}
+        ]
         
         try:
-            # 로딩 시작
             yield [
-                "",  # 빈 문자열로 변경 (코드 출력하지 않음)
-                _history,
+                "",
                 None,
                 gr.update(active_key="loading"),
                 gr.update(open=True)
@@ -295,8 +353,7 @@ class Demo:
             try:
                 async for content in try_claude_api(system_message, claude_messages):
                     yield [
-                        "",  # 빈 문자열로 변경 (코드 출력하지 않음)
-                        _history,
+                        "",
                         None,
                         gr.update(active_key="loading"),
                         gr.update(open=True)
@@ -309,8 +366,7 @@ class Demo:
                 
                 async for content in try_openai_api(openai_messages):
                     yield [
-                        "",  # 빈 문자열로 변경 (코드 출력하지 않음)
-                        _history,
+                        "",
                         None,
                         gr.update(active_key="loading"),
                         gr.update(open=True)
@@ -319,22 +375,72 @@ class Demo:
                     collected_content = content
 
             if collected_content:
-                _history = messages_to_history([
-                    {'role': Role.SYSTEM, 'content': system_message}
-                ] + claude_messages + [{
-                    'role': Role.ASSISTANT,
-                    'content': collected_content
-                }])
-        
-                # 최종 결과만 표시
+                # 이미지 생성이 필요한 경우
+                if needs_image and image_prompt:
+                    try:
+                        print(f"Generating image for prompt: {image_prompt}")
+                        # FLUX 모델을 사용하여 이미지 생성
+                        if pipe is not None:
+                            image = generate_image(
+                                prompt=image_prompt,
+                                height=512,
+                                width=512,
+                                steps=8,
+                                scales=3.5,
+                                seed=random.randint(1, 10000)
+                            )
+                            
+                            # 이미지를 Base64로 인코딩
+                            buffered = BytesIO()
+                            image.save(buffered, format="PNG")
+                            img_str = base64.b64encode(buffered.getvalue()).decode()
+                            
+                            # HTML에 이미지 추가
+                            image_html = f'''
+                            <div class="generated-image" style="margin: 20px 0; text-align: center;">
+                                <h3 style="color: #333; margin-bottom: 10px;">Generated Image:</h3>
+                                <img src="data:image/png;base64,{img_str}" 
+                                     style="max-width: 100%; 
+                                            border-radius: 10px; 
+                                            box-shadow: 0 4px 8px rgba(0,0,0,0.1);">
+                                <p style="color: #666; margin-top: 10px; font-style: italic;">
+                                    Prompt: {html.escape(image_prompt)}
+                                </p>
+                            </div>
+                            '''
+                            
+                            # HTML 응답에 이미지 삽입
+                            if '```html' in collected_content:
+                                # HTML 코드 블록 내부에 이미지 추가
+                                collected_content = collected_content.replace('```html\n', f'```html\n{image_html}')
+                            else:
+                                # HTML 코드 블록으로 감싸서 이미지 추가
+                                collected_content = f'```html\n{image_html}\n```\n{collected_content}'
+                            
+                            print("Image generation successful")
+                        else:
+                            raise Exception("FLUX model not initialized")
+                            
+                    except Exception as e:
+                        print(f"Image generation error: {str(e)}")
+                        error_message = f'''
+                        <div style="color: #ff4d4f; padding: 10px; margin: 10px 0; 
+                                    border-left: 4px solid #ff4d4f; background: #fff2f0;">
+                            <p>Failed to generate image: {str(e)}</p>
+                        </div>
+                        '''
+                        if '```html' in collected_content:
+                            collected_content = collected_content.replace('```html\n', f'```html\n{error_message}')
+                        else:
+                            collected_content = f'```html\n{error_message}\n```\n{collected_content}'
+
+                # 최종 결과 표시
                 yield [
                     collected_content,
-                    _history,
                     send_to_sandbox(remove_code_block(collected_content)),
                     gr.update(active_key="render"),
                     gr.update(open=False)
                 ]
-
             else:
                 raise ValueError("No content was generated from either API")
                 
@@ -457,7 +563,7 @@ def get_random_placeholder():
 
 def update_placeholder():
     return gr.update(placeholder=get_random_placeholder())
-    
+
 def create_main_interface():
     """메인 인터페이스 생성 함수"""
     
@@ -480,29 +586,150 @@ def create_main_interface():
     with open('app.css', 'r', encoding='utf-8') as f:
         custom_css = f.read()
 
-    demo = gr.Blocks(css=custom_css, theme=theme)
+    demo = gr.Blocks(css=custom_css + """
+
+        .empty-content {
+            padding: 40px !important;
+            background: #f8f9fa !important;
+            border-radius: 10px !important;
+            margin: 20px !important;
+        }    
+        
+        .container {
+            background: #f0f0f0;
+            min-height: 100vh;
+            padding: 20px;
+            display: flex;
+            justify-content: center;
+            align-items: center;
+            font-family: -apple-system, BlinkMacSystemFont, sans-serif;
+        }
+        
+        .app-window {
+            background: white;
+            border-radius: 10px;
+            box-shadow: 0 20px 40px rgba(0,0,0,0.1);
+            width: 100%;
+            max-width: 1400px;
+            overflow: hidden;
+        }
+        
+        .window-header {
+            background: #f0f0f0;
+            padding: 12px 16px;
+            display: flex;
+            align-items: center;
+            border-bottom: 1px solid #e0e0e0;
+        }
+        
+        .window-controls {
+            display: flex;
+            gap: 8px;
+        }
+        
+        .control {
+            width: 12px;
+            height: 12px;
+            border-radius: 50%;
+            cursor: pointer;
+        }
+        
+        .control.close { background: #ff5f57; }
+        .control.minimize { background: #febc2e; }
+        .control.maximize { background: #28c840; }
+        
+        .window-title {
+            flex: 1;
+            text-align: center;
+            color: #333;
+            font-size: 14px;
+            font-weight: 500;
+        }
+        
+        .main-content {
+            display: flex;
+            height: calc(100vh - 100px);
+        }
+        
+        .left-panel {
+            width: 40%;
+            border-right: 1px solid #e0e0e0;
+            padding: 20px;
+            display: flex;
+            flex-direction: column;
+        }
+        
+        .right-panel {
+            width: 60%;
+            background: #fff;
+            position: relative;
+        }
+        
+        .input-area {
+            background: #f8f9fa;
+            border-radius: 10px;
+            padding: 20px;
+            margin-top: 20px;
+        }
+        
+        .button-group {
+            display: flex;
+            gap: 10px;
+            margin-top: 20px;
+        }
+        
+        .custom-button {
+            background: #007aff;
+            color: white;
+            border: none;
+            padding: 10px 20px;
+            border-radius: 6px;
+            cursor: pointer;
+            transition: all 0.2s;
+        }
+        
+        .custom-button:hover {
+            background: #0056b3;
+        }
+    """, theme=theme)
+
+
 
     with demo:
         with gr.Tabs(elem_classes="main-tabs") as tabs:
             with gr.Tab("Visual AI Assistant", elem_id="mouse-tab", elem_classes="mouse-tab"):
-                history = gr.State([])
+                # SystemPrompt 설정을 위한 State 추가
                 setting = gr.State({
                     "system": SystemPrompt,
                 })
-
+                
                 with ms.Application() as app:
                     with antd.ConfigProvider():
-                        # Drawer 컴포넌트들    
+
                         with antd.Drawer(open=False, title="AI is Creating...", placement="left", width="750px") as code_drawer:
                             gr.HTML("""
                                 <div class="thinking-container">
                                     <style>
+                                        .custom-textarea {
+                                            background: #f8f9fa !important;
+                                            border: 1px solid #e0e0e0 !important;
+                                            border-radius: 10px !important;
+                                            padding: 15px !important;
+                                            min-height: 150px !important;
+                                            font-family: -apple-system, BlinkMacSystemFont, sans-serif !important;
+                                        }
+        
+                                        .custom-textarea:focus {
+                                            border-color: #007aff !important;
+                                            box-shadow: 0 0 0 2px rgba(0,122,255,0.2) !important;
+                                        }
+        
                                         .thinking-container {
                                             text-align: center;
                                             padding: 20px;
                                             background: #f8f9fa;
                                             border-radius: 15px;
-                                            font-family: 'Arial', sans-serif;
+                                            font-family: -apple-system, BlinkMacSystemFont, sans-serif;
                                         }
                                         
                                         .progress-bar {
@@ -517,7 +744,7 @@ def create_main_interface():
                                         .progress-bar-inner {
                                             width: 30%;
                                             height: 100%;
-                                            background: linear-gradient(90deg, #1890ff, #52c41a);
+                                            background: linear-gradient(90deg, #007aff, #28c840);
                                             animation: progress 2s ease-in-out infinite;
                                         }
                                         
@@ -536,13 +763,8 @@ def create_main_interface():
                                             transition: all 0.3s ease;
                                         }
                                         
-                                        .tip-box:hover {
-                                            transform: translateY(-5px);
-                                            box-shadow: 0 6px 16px rgba(0,0,0,0.15);
-                                        }
-                                        
                                         .status-text {
-                                            color: #1890ff;
+                                            color: #007aff;
                                             font-size: 18px;
                                             margin: 15px 0;
                                             animation: fade 1.5s ease infinite;
@@ -557,7 +779,7 @@ def create_main_interface():
                                         
                                         .icon-item {
                                             padding: 10px;
-                                            background: rgba(24,144,255,0.1);
+                                            background: rgba(0,122,255,0.1);
                                             border-radius: 8px;
                                             animation: pulse 2s ease infinite;
                                         }
@@ -585,22 +807,18 @@ def create_main_interface():
                                     </style>
                                     
                                     <div class="thinking-icon">🎨</div>
-                                    
                                     <div class="status-text">Creating Your Visualization...</div>
-                                    
                                     <div class="progress-bar">
                                         <div class="progress-bar-inner"></div>
                                     </div>
-                                    
                                     <div class="icon-grid">
                                         <div class="icon-item">📊</div>
                                         <div class="icon-item">🎯</div>
                                         <div class="icon-item">💡</div>
                                         <div class="icon-item">✨</div>
                                     </div>
-                                    
                                     <div class="tip-box">
-                                        <h3 style="color: #1890ff; margin-bottom: 10px;">Did You Know?</h3>
+                                        <h3 style="color: #007aff; margin-bottom: 10px;">Did You Know?</h3>
                                         <div id="tip-content" style="font-size: 16px; line-height: 1.6;"></div>
                                     </div>
                                     
@@ -637,106 +855,120 @@ def create_main_interface():
                             code_output = legacy.Markdown(visible=False)
 
 
-                
-                        with antd.Drawer(open=False, title="history", placement="left", width="900px") as history_drawer:
-                            history_output = legacy.Chatbot(show_label=False, flushing=False, height=960, elem_classes="history_chatbot")
-
-                
-                        with antd.Drawer(open=False, title="history", placement="left", width="900px") as history_drawer:
-                            history_output = legacy.Chatbot(show_label=False, flushing=False, height=960, elem_classes="history_chatbot")
-
-                        # 메인 컨텐츠를 위한 Row
+# 메인 컨텐츠를 위한 Row
                         with antd.Row(gutter=[32, 12]) as layout:    
                             # 좌측 패널
                             with antd.Col(span=24, md=8):
                                 with antd.Flex(vertical=True, gap="middle", wrap=True):
-                                    # 헤더 부분
-                                    header = gr.HTML(f"""
-                                        <div class="left_header">
-                                            <img src="data:image/gif;base64,{get_image_base64('mouse.gif')}" width="360px" />
-                                            <h1 style="font-size: 18px;">MOUSE-Chat: Visual AI Assistant</h1>
-                                            <h1 style="font-size: 10px;">Transform your questions into stunning visual presentations. Every response is crafted with beautiful graphics, charts, and interactive elements.</h1>
+                                    # macOS 스타일 윈도우 헤더
+                                    header = gr.HTML("""
+                                        <div class="window-frame">
+                                            <div class="window-header">
+                                                <div class="window-controls">
+                                                    <div class="control close"></div>
+                                                    <div class="control minimize"></div>
+                                                    <div class="control maximize"></div>
+                                                </div>
+                                                <div class="window-title">
+                                                    <div class="window-address">
+                                                        <div class="secure-icon">🔒</div>
+                                                        <div class="url-bar">https://VIDraft-mouse-chat.hf.space</div>
+                                                    </div>
+                                                </div>
+                                            </div>
+                                            <div class="app-content">
+                                                <img src="data:image/gif;base64,{}" width="360px" />
+                                                <h1 class="app-title">MOUSE-Chat Visual AI</h1>
+                                                <p class="app-description">Creates visualized web pages from text input, and when you include keywords like "image:", "그림:", or "image:" in your input, it automatically generates AI images based on the description and incorporates them into the web page.
+Use the "Generate" button for basic creation, "Enhance" button for prompt improvement, "Share" button to deploy results to the web, and input like "image: a dog playing in the park" to create results containing both text and generated images.</p>
+                                            </div>
                                         </div>
-                                    """)
-                                    
-
+                                    """.format(get_image_base64('mouse.gif')))
 
                                     # 입력 영역
                                     input = antd.InputTextarea(
                                         size="large", 
-                                        allow_clear=True, 
+                                        allow_clear=True,
+                                        placeholder=get_random_placeholder(),
+                                        elem_classes="custom-textarea"  # style 대신 class 사용
 
-                                        placeholder=get_random_placeholder() 
                                     )
  
                                     # 버튼 그룹
-                                    with antd.Flex(gap="small", justify="space-between"):
-                                        btn = antd.Button("Generate", type="primary", size="large")
-                                        boost_btn = antd.Button("Enhance", type="default", size="large")
-                                        deploy_btn = antd.Button("Share", type="default", size="large")
-                                        clear_btn = antd.Button("Clear", type="default", size="large")
-                                        historyBtn = antd.Button("📜 History", type="default")
-                                        
-
-
-                                    deploy_result = gr.HTML(label="배포 결과")
+                                    with antd.Flex(gap="small", justify="flex-start"):
+                                        btn = antd.Button(
+                                            "Generate", 
+                                            type="primary", 
+                                            size="large",
+                                            elem_classes="generate-btn"
+                                        )
+                                        boost_btn = antd.Button(
+                                            "Enhance", 
+                                            type="default", 
+                                            size="large",
+                                            elem_classes="enhance-btn"
+                                        )
+                                        deploy_btn = antd.Button(
+                                            "Share", 
+                                            type="default", 
+                                            size="large",
+                                            elem_classes="share-btn"
+                                        )
+
+
+                                    deploy_result = gr.HTML(
+                                        label="Share Result",
+                                        elem_classes="deploy-result"
+                                    )
 
                             # 우측 패널
                             with antd.Col(span=24, md=16):
                                 with ms.Div(elem_classes="right_panel"):
-                                    # 상단 버튼들
-                                    with antd.Flex(gap="small", elem_classes="setting-buttons"):
-                                        historyBtn = antd.Button("📜 히스토리", type="default")
-
-                                    gr.HTML('<div class="render_header"><span class="header_btn"></span><span class="header_btn"></span><span class="header_btn"></span></div>')
+                                    # macOS 스타일 윈도우 헤더
+                                    gr.HTML("""
+                                        <div class="window-frame">
+                                            <div class="window-header">
+                                                <div class="window-controls">
+                                                    <div class="control close"></div>
+                                                    <div class="control minimize"></div>
+                                                    <div class="control maximize"></div>
+                                                </div>
+                                                <div class="window-title">Preview</div>
+                                            </div>
+                                        </div>
+                                    """)
 
                                     # 탭 컨텐츠
                                     with antd.Tabs(active_key="empty", render_tab_bar="() => null") as state_tab:
                                         with antd.Tabs.Item(key="empty"):
+                                            empty = antd.Empty(
+                                                description="Enter your question to begin",
 
-                                            empty = antd.Empty(description="Enter your question to begin", elem_classes="right_content")
-                            
+                                                elem_classes="right_content empty-content"  # style 대신 class 사용
+
+                                            )
+                                        
                                         with antd.Tabs.Item(key="loading"):
-                                            loading = antd.Spin(True, tip="Creating visual presentation...", size="large", elem_classes="right_content")
+                                            loading = antd.Spin(
+                                                True,
+                                                tip="Creating visual presentation...",
+                                                size="large",
+                                                elem_classes="right_content"
+                                            )
+                                        
                                         with antd.Tabs.Item(key="render"):
                                             sandbox = gr.HTML(elem_classes="html_content")
-
-
-
-                        
-                        # 이벤트 핸들러 연결
-                        historyBtn.click(
-                            history_render,
-                            inputs=[history],
-                            outputs=[history_drawer, history_output]
-                        )
-
-                        history_drawer.close(
-                            lambda: gr.update(open=False),
-                            inputs=[],
-                            outputs=[history_drawer]
-                        )
+                            
 
                         btn.click(
                             demo_instance.generation_code,
-                            inputs=[input, setting, history],
-                            outputs=[code_output, history, sandbox, state_tab, code_drawer]
-                        ).then(  # then을 사용하여 연속 동작 추가
+                            inputs=[input, setting],  # setting이 이제 정의됨
+                            outputs=[code_output, sandbox, state_tab, code_drawer]
+                        ).then(
                             fn=update_placeholder,
                             inputs=[],
                             outputs=[input]
                         )
-                        
-
-
-                        clear_btn.click(
-                            fn=lambda: (
-                            demo_instance.clear_history(),  # history 초기화
-                            update_placeholder()  # placeholder 업데이트
-                        ),
-                        inputs=[],
-                        outputs=[history, input]  # input도 출력에 추가
-                        )
 
 
                         boost_btn.click(
@@ -746,11 +978,97 @@ def create_main_interface():
                         )
 
                         deploy_btn.click(
-                            fn=lambda code: deploy_to_vercel(remove_code_block(code)) if code else "코드가 없습니다.",
+                            fn=lambda code: deploy_to_vercel(remove_code_block(code)) if code else "No code to share.",
                             inputs=[code_output],
                             outputs=[deploy_result]
                         )
 
+    gr.HTML("""
+        <style>
+            .generate-btn {
+                background: #007aff !important;
+                border-radius: 8px !important;
+                box-shadow: 0 2px 4px rgba(0,0,0,0.1) !important;
+            }
+            
+            .enhance-btn {
+                border-radius: 8px !important;
+                border: 1px solid #007aff !important;
+                color: #007aff !important;
+            }
+            
+            .share-btn {
+                border-radius: 8px !important;
+                border: 1px solid #28c840 !important;
+                color: #28c840 !important;
+            }
+            
+            /* hover 효과 */
+            .generate-btn:hover {
+                background: #0056b3 !important;
+            }
+            
+            .enhance-btn:hover {
+                background: rgba(0,122,255,0.1) !important;
+            }
+        
+            .share-btn:hover {
+                background: rgba(40,200,64,0.1) !important;
+            }
+
+            .app-content {
+                padding: 20px;
+                text-align: center;
+            }
+            
+            .app-title {
+                font-size: 24px;
+                color: #333;
+                margin: 20px 0 10px;
+                font-weight: 600;
+            }
+            
+            .app-description {
+                color: #666;
+                font-size: 14px;
+                margin-bottom: 30px;
+            }
+            
+            .deploy-result {
+                margin-top: 20px;
+                padding: 15px;
+                background: #f8f9fa;
+                border-radius: 8px;
+                font-family: -apple-system, BlinkMacSystemFont, sans-serif;
+            }
+            
+            .deploy-result a {
+                color: #007aff;
+                text-decoration: none;
+                font-weight: 500;
+            }
+            
+            .deploy-result a:hover {
+                text-decoration: underline;
+            }
+            
+            /* 반응형 디자인을 위한 미디어 쿼리 */
+            @media (max-width: 768px) {
+                .window-frame {
+                    border-radius: 0;
+                }
+                
+                .left-panel, .right-panel {
+                    width: 100%;
+                }
+                
+                .main-content {
+                    flex-direction: column;
+                }
+            }
+        </style>
+    """)
+
     return demo
 
 # 메인 실행 부분
@@ -758,7 +1076,16 @@ if __name__ == "__main__":
     try:
         demo_instance = Demo()  # Demo 인스턴스 생성
         demo = create_main_interface()  # 인터페이스 생성
-        demo.queue(default_concurrency_limit=20).launch(server_name="0.0.0.0", server_port=7860)  # 서버 설정 추가
+        demo.queue(
+            default_concurrency_limit=20,  # concurrency_count 대신 default_concurrency_limit 사용
+            status_update_rate=10,
+            api_open=False
+        ).launch(
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False,
+            debug=False
+        )
     except Exception as e:
         print(f"Initialization error: {e}")
         raise