Fix: Add-multi env

Dockerfile

@@ -1,14 +1,13 @@
 # Dockerfile
 
-# 1. 베이스 이미지 설정
 FROM python:3.10-slim
 
-# 2. 메타데이터
+# 메타데이터
 LABEL maintainer="[email protected]"
-LABEL description="SimSimi-based Conversational AI Agent"
+LABEL description="SimSimi AI Agent - Multi Environment Support"
 LABEL version="1.0.0"
 
-# 3. 시스템 의존성 설치
+# 시스템 의존성 설치
 RUN apt-get update && apt-get install -y \
     gcc \
     g++ \
@@ -18,36 +17,45 @@ RUN apt-get update && apt-get install -y \
     && git lfs install \
     && rm -rf /var/lib/apt/lists/*
 
-# 4. 작업 디렉토리 설정
 WORKDIR /app
 
-# 5. 환경 변수 설정
+# 환경 변수 설정
 ENV HF_HOME=/app/cache
 ENV HF_DATASETS_CACHE=/app/cache
 ENV TRANSFORMERS_CACHE=/app/cache
+ENV PYTHONPATH=/app
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
 
-# 6. Python 의존성 설치
+# Python 의존성 설치
 COPY requirements.txt .
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 7. 데이터 다운로드를 빌드 단계에서 미리 실행
+# 애플리케이션 코드 복사
+COPY . .
+
+# 필요한 디렉토리 생성
+RUN mkdir -p /app/data /app/cache /app/logs /app/static
+RUN chmod -R 777 /app/data /app/cache /app/logs
+
+# 조건부 데이터 다운로드
 RUN huggingface-cli download \
     youdie006/simsimi-ai-agent-data \
     --repo-type dataset \
     --local-dir /app/data \
-    --local-dir-use-symlinks False
-RUN chmod -R 777 /app/data /app/cache
+    --local-dir-use-symlinks False || echo "⚠️ 데이터 다운로드 실패 - 런타임에 생성"
 
-# 8. 애플리케이션 코드 복사
-COPY . .
+# 스마트 시작 스크립트 복사
+COPY start.sh /app/start.sh
+RUN chmod +x /app/start.sh
 
-# 9. 포트 노출
-EXPOSE 8000
+# 포트 노출 (7860 고정)
+EXPOSE 7860
 
-# 10. [최종 수정] 헬스체크 시작 대기 시간을 모델 로딩 시간보다 넉넉하게 변경
-HEALTHCHECK --interval=60s --timeout=30s --start-period=300s --retries=3 \
-    CMD curl -f http://localhost:8000/api/v1/health || exit 1
+# 헬스체크
+HEALTHCHECK --interval=30s --timeout=30s --start-period=300s --retries=3 \
+    CMD curl -f http://localhost:7860/api/v1/health || exit 1
 
-# 11. 운영용 서버 실행 (타임아웃 및 프리로드 적용)
-CMD ["gunicorn", "-w", "4", "-k", "uvicorn.workers.UvicornWorker", "--timeout", "300", "--preload", "-b", "0.0.0.0:8000", "main:app"]
+# 🎯 스마트 시작 - 환경 자동 감지!
+CMD ["/app/start.sh"]

README.md

@@ -6,7 +6,7 @@ emoji: 💙
 colorFrom: purple
 colorTo: blue
 sdk: docker
-app_port: 8000
+app_port: 7860
 pinned: false
 ---
 

docker-compose.yml

@@ -1,3 +1,5 @@
+# docker-compose.yml - 로컬 개발 최적화
+
 services:
   simsimi-ai-agent:
     build:
@@ -5,26 +7,28 @@ services:
       dockerfile: Dockerfile
     container_name: simsimi_ai_agent
     ports:
-      - "8000:8000"
+      - "7860:7860"  # 허깅페이스와 동일한 포트
     volumes:
-      # 소스코드 실시간 반영 (개발용)
+      # 🔄 소스코드 실시간 반영 (개발용)
       - ./src:/app/src
       - ./scripts:/app/scripts
       - ./main.py:/app/main.py
-      # 데이터 영구 저장
+      - ./static:/app/static
+      
+      # 💾 데이터 영구 저장
       - ./data:/app/data
       - ./logs:/app/logs
-      # [추가] 캐시 데이터 영구 저장
-      # 이렇게 하면 컨테이너를 껐다 켜도 매번 모델을 새로 다운로드하지 않습니다.
       - ./cache:/app/cache
-      # 환경변수 (로컬에서만)
+      
+      # 🔑 환경변수 (로컬에서만)
       - ./.env:/app/.env:ro
     environment:
       - PYTHONPATH=/app
       - PYTHONDONTWRITEBYTECODE=1
       - PYTHONUNBUFFERED=1
-      # [추가] Hugging Face 캐시 디렉토리 환경 변수
       - HF_HOME=/app/cache
+      - LOCAL_DEV=true           # 🏠 로컬 개발 환경 표시
+      - DEVELOPMENT_MODE=true    # 🔧 개발 모드 활성화
     env_file:
       - .env
     restart: unless-stopped
@@ -33,6 +37,37 @@ services:
     networks:
       - simsimi_network
 
+  # 🚀 프로덕션 테스트용 (옵션)
+  simsimi-production-test:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    container_name: simsimi_production_test
+    ports:
+      - "7861:7860"  # 다른 포트로 동시 실행
+    volumes:
+      - ./data:/app/data
+      - ./logs:/app/logs
+      - ./cache:/app/cache
+      - ./.env:/app/.env:ro
+    environment:
+      - PYTHONPATH=/app
+      - PYTHONDONTWRITEBYTECODE=1
+      - PYTHONUNBUFFERED=1
+      - HF_HOME=/app/cache
+      - PRODUCTION=true          # 🏭 프로덕션 모드 테스트
+    env_file:
+      - .env
+    restart: unless-stopped
+    profiles:
+      - production-test  # docker-compose --profile production-test up
+    networks:
+      - simsimi_network
+
 networks:
   simsimi_network:
-    driver: bridge
+    driver: bridge
+
+# 사용법:
+# 개발: docker-compose up
+# 프로덕션 테스트: docker-compose --profile production-test up

main.py

@@ -1,61 +1,281 @@
-# main.py
+# main.py - 환경 자동 감지 및 최적화
 
-"""
-청소년 공감형 AI 챗봇 메인 서버
-"""
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse, HTMLResponse
-from fastapi.staticfiles import StaticFiles
 import os
 import sys
 from datetime import datetime
 from dotenv import load_dotenv
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse, HTMLResponse
+from fastapi.staticfiles import StaticFiles
 
 # 환경 변수 로드
 load_dotenv()
 
-# FastAPI 앱 생성
+
+# 🌍 환경 감지
+class EnvironmentDetector:
+    @staticmethod
+    def detect_environment():
+        """실행 환경 자동 감지"""
+        if os.getenv("SPACE_ID") or os.getenv("SPACE_AUTHOR_NAME"):
+            return "huggingface"
+        elif os.getenv("LOCAL_DEV") == "true" or os.getenv("DEVELOPMENT_MODE") == "true":
+            return "local_dev"
+        elif os.getenv("PRODUCTION") == "true":
+            return "production"
+        else:
+            return "default"
+
+    @staticmethod
+    def get_environment_config(env_type: str) -> dict:
+        """환경별 설정 반환"""
+        configs = {
+            "huggingface": {
+                "debug": False,
+                "reload": False,
+                "log_level": "info",
+                "cors_origins": ["*"],
+                "description": "🤗 허깅페이스 Spaces에서 실행 중",
+                "features": {
+                    "static_files": True,
+                    "api_docs": True,
+                    "debug_routes": False
+                }
+            },
+            "local_dev": {
+                "debug": True,
+                "reload": True,
+                "log_level": "debug",
+                "cors_origins": ["*"],
+                "description": "🏠 로컬 개발 환경에서 실행 중",
+                "features": {
+                    "static_files": True,
+                    "api_docs": True,
+                    "debug_routes": True
+                }
+            },
+            "production": {
+                "debug": False,
+                "reload": False,
+                "log_level": "warning",
+                "cors_origins": ["https://yourdomain.com"],
+                "description": "🏭 프로덕션 환경에서 실행 중",
+                "features": {
+                    "static_files": True,
+                    "api_docs": False,
+                    "debug_routes": False
+                }
+            },
+            "default": {
+                "debug": False,
+                "reload": False,
+                "log_level": "info",
+                "cors_origins": ["*"],
+                "description": "🔧 기본 환경에서 실행 중",
+                "features": {
+                    "static_files": True,
+                    "api_docs": True,
+                    "debug_routes": False
+                }
+            }
+        }
+        return configs.get(env_type, configs["default"])
+
+
+# 환경 감지 및 설정
+ENVIRONMENT = EnvironmentDetector.detect_environment()
+CONFIG = EnvironmentDetector.get_environment_config(ENVIRONMENT)
+
+print(f"🌍 감지된 환경: {ENVIRONMENT}")
+print(f"📋 설정: {CONFIG['description']}")
+
+# FastAPI 앱 생성 (환경별 설정 적용)
 app = FastAPI(
     title="💙 마음이 - 청소년 상담 챗봇",
-    description="13-19세 청소년을 위한 AI 공감 상담사",
-    version=os.getenv("VERSION", "2.0.0")
+    description=f"13-19세 청소년을 위한 AI 공감 상담사 ({CONFIG['description']})",
+    version=os.getenv("VERSION", "2.0.0"),
+    docs_url="/docs" if CONFIG["features"]["api_docs"] else None,
+    redoc_url="/redoc" if CONFIG["features"]["api_docs"] else None,
+    debug=CONFIG["debug"]
 )
 
-# CORS 설정
+# CORS 설정 (환경별)
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],
+    allow_origins=CONFIG["cors_origins"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
 )
 
-# 정적 파일 서빙
-try:
-    app.mount("/static", StaticFiles(directory="static"), name="static")
-    print("✅ 정적 파일 서빙 설정 완료")
-except RuntimeError:
-    print("정적 파일 디렉토리를 찾을 수 없습니다. API 서버는 계속 실행됩니다.")
+# 정적 파일 서빙 (환경별)
+if CONFIG["features"]["static_files"]:
+    try:
+        static_dir = "static"
+        if not os.path.exists(static_dir):
+            os.makedirs(static_dir)
 
-# 라우터 등록
+        index_path = os.path.join(static_dir, "index.html")
+        if not os.path.exists(index_path):
+            create_default_html(index_path)
+
+        app.mount("/static", StaticFiles(directory=static_dir), name="static")
+        print("✅ 정적 파일 서빙 설정 완료")
+    except Exception as e:
+        print(f"⚠️ 정적 파일 설정 실패: {e}")
+
+# 라우터 등록 (오류 처리 포함)
 try:
     from src.api import chat, openai, vector
+
     app.include_router(chat.router, prefix="/api/v1/chat", tags=["💙 Teen Chat"])
     app.include_router(openai.router, prefix="/api/v1/openai", tags=["🤖 OpenAI GPT-4"])
     app.include_router(vector.router, prefix="/api/v1/vector", tags=["🗄️ Vector Store"])
     print("✅ API 라우터 등록 완료")
 except ImportError as e:
     print(f"⚠️ API 라우터 import 실패: {e}")
+    add_demo_routes()
 
+
+# 기본 라우트들
 @app.get("/", response_class=HTMLResponse)
 async def root():
+    """메인 페이지"""
     html_file_path = "static/index.html"
     if os.path.exists(html_file_path):
         with open(html_file_path, "r", encoding="utf-8") as f:
             return HTMLResponse(content=f.read())
-    return HTMLResponse(content="<h1>Welcome to 마음이 AI</h1><p>UI not found, but API is running. Access docs at /docs.</p>")
+    return HTMLResponse(content=get_default_html())
+
 
 @app.get("/api/v1/health")
 async def health_check():
-    return {"status": "healthy"}
+    """헬스 체크 (환경 정보 포함)"""
+    return {
+        "status": "healthy",
+        "environment": ENVIRONMENT,
+        "config": CONFIG['description'],
+        "features": CONFIG['features'],
+        "timestamp": datetime.now().isoformat(),
+        "python_version": sys.version.split()[0],
+        "debug_mode": CONFIG["debug"]
+    }
+
+
+@app.get("/api/v1/environment")
+async def get_environment_info():
+    """환경 정보 조회"""
+    return {
+        "environment": ENVIRONMENT,
+        "config": CONFIG,
+        "env_vars": {
+            "SPACE_ID": bool(os.getenv("SPACE_ID")),
+            "LOCAL_DEV": os.getenv("LOCAL_DEV"),
+            "DEVELOPMENT_MODE": os.getenv("DEVELOPMENT_MODE"),
+            "PRODUCTION": os.getenv("PRODUCTION"),
+            "OPENAI_API_KEY_SET": bool(os.getenv("OPENAI_API_KEY"))
+        }
+    }
+
+
+# 디버그 라우트 (개발 환경에서만)
+if CONFIG["features"]["debug_routes"]:
+    @app.get("/api/v1/debug/reload")
+    async def debug_reload():
+        """개발용: 서버 재시작 없이 모듈 리로드"""
+        return {"message": "개발 모드에서만 사용 가능", "environment": ENVIRONMENT}
+
+
+    @app.get("/api/v1/debug/logs")
+    async def debug_logs():
+        """개발용: 최근 로그 조회"""
+        try:
+            with open("/app/logs/app.log", "r") as f:
+                logs = f.readlines()[-50:]  # 최근 50줄
+            return {"logs": logs}
+        except FileNotFoundError:
+            return {"logs": ["로그 파일이 없습니다."]}
+
+
+def create_default_html(file_path: str):
+    """기본 HTML 파일 생성"""
+    html_content = get_default_html()
+    with open(file_path, "w", encoding="utf-8") as f:
+        f.write(html_content)
+
+
+def get_default_html() -> str:
+    """환경별 기본 HTML 반환"""
+    env_emoji = {
+        "huggingface": "🤗",
+        "local_dev": "🏠",
+        "production": "🏭",
+        "default": "🔧"
+    }
+
+    return f'''
+<!DOCTYPE html>
+<html lang="ko">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>마음이 AI | {CONFIG['description']}</title>
+    <style>
+        body {{ font-family: 'Noto Sans KR', sans-serif; background: #f4f7f6; display: flex; justify-content: center; align-items: center; min-height: 100vh; margin: 0; }}
+        .container {{ text-align: center; background: white; padding: 40px; border-radius: 20px; box-shadow: 0 8px 32px rgba(0,0,0,0.1); }}
+        .title {{ color: #8A2BE2; font-size: 2rem; margin-bottom: 20px; }}
+        .env {{ color: #666; font-size: 1rem; margin-bottom: 15px; }}
+        .message {{ color: #666; font-size: 1.1rem; margin-bottom: 30px; }}
+        .status {{ padding: 10px 20px; background: #e8f5e8; color: #4a5e4a; border-radius: 8px; display: inline-block; }}
+        {"" if not CONFIG['debug'] else ".debug { background: #fff3cd; color: #856404; padding: 10px; border-radius: 5px; margin-top: 20px; }"}
+    </style>
+</head>
+<body>
+    <div class="container">
+        <h1 class="title">💙 마음이 AI</h1>
+        <div class="env">{env_emoji.get(ENVIRONMENT, "🔧")} {CONFIG['description']}</div>
+        <p class="message">청소년 공감 상담 챗봇이 곧 시작됩니다!</p>
+        <div class="status">시스템 초기화 중...</div>
+        {"" if not CONFIG['debug'] else '<div class="debug">🔧 개발 모드 - 디버그 정보 활성화</div>'}
+        <script>
+            setTimeout(() => {{
+                window.location.reload();
+            }}, 5000);
+        </script>
+    </div>
+</body>
+</html>
+    '''
+
+
+def add_demo_routes():
+    """데모용 라우터 추가"""
+
+    @app.post("/api/v1/chat/teen-chat")
+    async def demo_chat(request: dict):
+        return {
+            "response": f"안녕! 마음이가 곧 준비될 예정이야. ({ENVIRONMENT} 환경) 💙",
+            "status": "demo_mode",
+            "environment": ENVIRONMENT
+        }
+
+
+# 환경별 실행 (스크립트로 직접 실행할 때)
+if __name__ == "__main__":
+    import uvicorn
+
+    port = int(os.getenv("API_PORT", 7860))
+
+    print(f"🚀 {CONFIG['description']} 서버 시작")
+    print(f"📍 포트: {port}")
+    print(f"🔧 디버그 모드: {CONFIG['debug']}")
+    print(f"🔄 리로드: {CONFIG['reload']}")
+
+    uvicorn.run(
+        "main:app" if not CONFIG['reload'] else app,
+        host="0.0.0.0",
+        port=port,
+        reload=CONFIG['reload'],
+        log_level=CONFIG['log_level']
+    )

start.sh

@@ -0,0 +1,84 @@
+#!/bin/bash
+# start.sh - 환경 자동 감지 스마트 시작 스크립트
+
+echo "🚀 마음이 AI 시작 중..."
+
+# 환경 감지
+if [ -n "$SPACE_ID" ] || [ -n "$SPACE_AUTHOR_NAME" ]; then
+    echo "🤗 허깅페이스 Spaces 환경 감지"
+    ENVIRONMENT="huggingface"
+    PORT=7860
+    WORKERS=1
+    RELOAD=""
+elif [ "$DEVELOPMENT_MODE" = "true" ] || [ -f "/.dockerenv" ] && [ -n "$LOCAL_DEV" ]; then
+    echo "🏠 로컬 개발 환경 감지"
+    ENVIRONMENT="local_dev"
+    PORT=7860
+    WORKERS=1
+    RELOAD="--reload"
+elif [ -n "$PRODUCTION" ]; then
+    echo "🏭 프로덕션 환경 감지"
+    ENVIRONMENT="production"
+    PORT=7860
+    WORKERS=2
+    RELOAD=""
+else
+    echo "🔧 기본 환경으로 시작"
+    ENVIRONMENT="default"
+    PORT=7860
+    WORKERS=1
+    RELOAD=""
+fi
+
+echo "📊 환경 정보:"
+echo "  - 환경: $ENVIRONMENT"
+echo "  - 포트: $PORT"
+echo "  - 워커 수: $WORKERS"
+echo "  - 리로드: ${RELOAD:-"비활성화"}"
+
+# Python 환경 확인
+echo "🐍 Python 환경 확인:"
+python --version
+pip show fastapi uvicorn | grep Version
+
+# 필수 디렉토리 확인
+echo "📁 디렉토리 확인:"
+mkdir -p /app/data /app/cache /app/logs /app/static
+ls -la /app/
+
+# 환경별 시작 방식
+case $ENVIRONMENT in
+    "huggingface")
+        echo "🤗 허깅페이스 모드로 시작..."
+        exec python -m uvicorn main:app \
+            --host 0.0.0.0 \
+            --port $PORT \
+            --timeout-keep-alive 300 \
+            --log-level info
+        ;;
+    "local_dev")
+        echo "🏠 로컬 개발 모드로 시작..."
+        exec python -m uvicorn main:app \
+            --host 0.0.0.0 \
+            --port $PORT \
+            --reload \
+            --log-level debug
+        ;;
+    "production")
+        echo "🏭 프로덕션 모드로 시작..."
+        exec gunicorn main:app \
+            -w $WORKERS \
+            -k uvicorn.workers.UvicornWorker \
+            --bind 0.0.0.0:$PORT \
+            --timeout 300 \
+            --preload \
+            --log-level info
+        ;;
+    *)
+        echo "🔧 기본 모드로 시작..."
+        exec python -m uvicorn main:app \
+            --host 0.0.0.0 \
+            --port $PORT \
+            --log-level info
+        ;;
+esac