Update app.py

app.py

@@ -1,226 +1,57 @@
 import gradio as gr
-import json
-import os
 from datetime import datetime
-import base64
-from PIL import Image, ImageDraw, ImageFont
-import io
 import random
 
-# 간단한 메모리 저장소
+# 간단한 작업 목록
 jobs = []
-results = {}
-
-# 디렉토리 생성
-os.makedirs("temp", exist_ok=True)
-
-def create_test_image(prompt):
-    """테스트 이미지 생성"""
-    img = Image.new('RGB', (512, 512), color=(255, 255, 255))
-    draw = ImageDraw.Draw(img)
-    
-    # 배경 그라데이션
-    for i in range(512):
-        color = int(255 * (1 - i/512))
-        draw.rectangle([(0, i), (512, i+1)], fill=(color, color, 255))
-    
-    # 텍스트 추가
-    text_lines = [
-        f"Prompt: {prompt[:40]}...",
-        f"Generated at: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
-        f"Job ID: {random.randint(1000, 9999)}"
-    ]
-    
-    y_offset = 20
-    for line in text_lines:
-        draw.text((20, y_offset), line, fill='white')
-        y_offset += 30
-    
-    # 격자 패턴
-    for i in range(0, 512, 32):
-        draw.line([(i, 100), (i, 512)], fill=(200, 200, 200), width=1)
-        draw.line([(0, i+100), (512, i+100)], fill=(200, 200, 200), width=1)
-    
-    return img
 
 def submit_job(prompt):
     """작업 제출"""
     if not prompt:
-        return "❌ 프롬프트를 입력해주세요.", None
+        return "프롬프트를 입력해주세요."
     
-    job_id = f"job_{len(jobs)}_{datetime.now().strftime('%H%M%S')}"
-    job = {
+    job_id = f"JOB-{random.randint(1000, 9999)}"
+    jobs.append({
         "id": job_id,
         "prompt": prompt,
-        "status": "processing",
-        "created_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    }
-    jobs.append(job)
+        "time": datetime.now().strftime("%H:%M:%S")
+    })
     
-    # 시뮬레이션: 바로 이미지 생성
-    img = create_test_image(prompt)
-    
-    # 이미지를 임시 파일로 저장
-    img_path = f"temp/{job_id}.png"
-    img.save(img_path)
-    results[job_id] = img_path
-    job["status"] = "completed"
-    
-    return f"✅ 작업이 제출되었습니다!\n작업 ID: {job_id}", img
+    return f"✅ 작업 제출됨: {job_id}\n프롬프트: {prompt}"
 
 def get_status():
-    """시스템 상태"""
-    total_jobs = len(jobs)
-    completed = sum(1 for j in jobs if j["status"] == "completed")
-    
-    status_text = f"""## 📊 시스템 상태
-
-### 현재 상황
-- **전체 작업**: {total_jobs}개
-- **완료된 작업**: {completed}개
-- **대기중 작업**: {total_jobs - completed}개
-
-### 최근 작업 (최대 5개)
-"""
-    
-    recent_jobs = jobs[-5:][::-1] if jobs else []
-    for job in recent_jobs:
-        emoji = "✅" if job["status"] == "completed" else "⏳"
-        status_text += f"\n{emoji} **{job['id']}**: {job['prompt'][:30]}... ({job['created_at']})"
-    
+    """상태 확인"""
     if not jobs:
-        status_text += "\n\n*아직 작업이 없습니다.*"
-    
-    return status_text
-
-def get_gallery():
-    """갤러리 이미지 목록"""
-    gallery_images = []
+        return "아직 작업이 없습니다."
     
-    for job in jobs[-12:][::-1]:  # 최근 12개
-        if job["status"] == "completed" and job["id"] in results:
-            img_path = results[job["id"]]
-            if os.path.exists(img_path):
-                gallery_images.append((img_path, f"{job['prompt'][:50]}..."))
+    status = f"전체 작업: {len(jobs)}개\n\n최근 작업:\n"
+    for job in jobs[-5:]:
+        status += f"- {job['id']}: {job['prompt'][:30]}... ({job['time']})\n"
     
-    return gallery_images
-
-# 클라이언트 코드
-CLIENT_CODE = """# P2P GPU Client (Demo Version)
-import time
-import random
-from datetime import datetime
-
-print("🚀 P2P GPU 클라이언트 시뮬레이션")
-print("실제 구현에서는 Gradio API를 통해 서버와 통신합니다.")
-
-while True:
-    print(f"\\n[{datetime.now().strftime('%H:%M:%S')}] 작업 확인 중...")
-    
-    if random.random() > 0.7:  # 30% 확률로 작업 수신
-        print("📋 새 작업 발견!")
-        print("🎨 이미지 생성 중...")
-        time.sleep(2)
-        print("✅ 완료! 서버로 전송")
-    else:
-        print("대기 중...")
-    
-    time.sleep(5)
-"""
+    return status
 
 # Gradio 인터페이스
-with gr.Blocks(
+demo = gr.Interface(
+    fn=submit_job,
+    inputs=gr.Textbox(label="이미지 프롬프트", placeholder="원하는 이미지를 설명하세요..."),
+    outputs=gr.Textbox(label="결과"),
     title="P2P GPU 이미지 생성 허브",
-    theme=gr.themes.Soft(),
-    css="""
-    .container { max-width: 1200px; margin: 0 auto; }
-    #gallery { min-height: 400px; }
-    """
-) as demo:
-    
-    gr.Markdown("""
-    # 🤖 P2P GPU 이미지 생성 허브
-    
-    유휴 GPU를 활용한 분산 이미지 생성 시스템 (데모 버전)
-    """)
-    
-    with gr.Tabs():
-        # 작업 제출 탭
-        with gr.Tab("🎨 작업 제출"):
-            with gr.Row():
-                with gr.Column(scale=2):
-                    prompt_input = gr.Textbox(
-                        label="이미지 프롬프트",
-                        placeholder="생성하고 싶은 이미지를 설명하세요...",
-                        lines=3
-                    )
-                    submit_btn = gr.Button("🚀 작업 제출", variant="primary")
-                    
-                    result_msg = gr.Textbox(label="제출 결과", interactive=False)
-                    result_img = gr.Image(label="생성된 이미지", type="pil")
-                
-                with gr.Column(scale=1):
-                    gr.Markdown("""
-                    ### 💡 사용 방법
-                    1. 프롬프트 입력
-                    2. 작업 제출 클릭
-                    3. 결과 확인
-                    
-                    ### 📝 예시
-                    - 푸른 하늘과 구름
-                    - 일몰의 해변 풍경
-                    - 미래적인 도시 전경
-                    """)
-            
-            submit_btn.click(
-                fn=submit_job,
-                inputs=prompt_input,
-                outputs=[result_msg, result_img]
-            )
-        
-        # 시스템 상태 탭
-        with gr.Tab("📊 시스템 상태"):
-            status_display = gr.Markdown()
-            refresh_btn = gr.Button("🔄 새로고침")
-            
-            def refresh_status():
-                return get_status()
-            
-            refresh_btn.click(fn=refresh_status, outputs=status_display)
-            demo.load(fn=refresh_status, outputs=status_display)
-        
-        # 갤러리 탭
-        with gr.Tab("🖼️ 갤러리"):
-            gallery = gr.Gallery(
-                label="생성된 이미지들",
-                columns=3,
-                rows=2,
-                height="auto"
-            )
-            gallery_refresh_btn = gr.Button("🔄 갤러리 새로고침")
-            
-            gallery_refresh_btn.click(fn=get_gallery, outputs=gallery)
-            demo.load(fn=get_gallery, outputs=gallery)
-        
-        # 다운로드 탭
-        with gr.Tab("💻 클라이언트"):
-            gr.Markdown("""
-            ## 클라이언트 프로그램
-            
-            ### 데모 코드
-            아래는 P2P 클라이언트의 간단한 데모 코드입니다:
-            
-            ```python
-            """ + CLIENT_CODE + """
-            ```
-            
-            ### 실제 구현 시
-            - Gradio API를 통한 서버 통신
-            - GPU 자동 감지 및 활용
-            - 실제 이미지 생성 모델 연동
-            - 자동 작업 처리 및 결과 전송
-            """)
+    description="분산 GPU를 활용한 이미지 생성 시스템"
+)
+
+# 상태 확인용 두 번째 인터페이스
+status_demo = gr.Interface(
+    fn=get_status,
+    inputs=None,
+    outputs=gr.Textbox(label="시스템 상태"),
+    title="시스템 상태"
+)
+
+# 탭으로 결합
+app = gr.TabbedInterface(
+    [demo, status_demo],
+    ["작업 제출", "시스템 상태"]
+)
 
-# 앱 실행
 if __name__ == "__main__":
-    demo.launch()
+    app.launch(server_name="0.0.0.0", server_port=7860)