app.py

import os
import streamlit as st
import replicate
from PIL import Image
import io
import base64
import tempfile

# 페이지 설정
st.set_page_config(
    page_title="AI Video Generator",
    page_icon="🎬",
    layout="wide"
)

# 스타일 적용
st.markdown("""
<style>
    .main {
        padding-top: 2rem;
    }
    .stButton>button {
        width: 100%;
        background-color: #4CAF50;
        color: white;
        font-weight: bold;
        padding: 0.5rem;
        border-radius: 0.5rem;
    }
    .stButton>button:hover {
        background-color: #45a049;
    }
</style>
""", unsafe_allow_html=True)

# 타이틀
st.title("🎬 AI Video Generator")
st.markdown("**Replicate API**를 사용하여 텍스트나 이미지로부터 비디오를 생성합니다.")

# API 토큰 설정
api_token = os.getenv("RAPI_TOKEN")

# 사이드바 설정
with st.sidebar:
    st.header("⚙️ 설정")
    
    # API 토큰 입력 (환경변수가 없는 경우)
    if not api_token:
        api_token_input = st.text_input(
            "Replicate API Token",
            type="password",
            help="환경변수 RAPI_TOKEN이 설정되지 않았습니다. API 토큰을 입력하세요."
        )
        if api_token_input:
            api_token = api_token_input
            os.environ["REPLICATE_API_TOKEN"] = api_token
    else:
        st.success("✅ API 토큰이 환경변수에서 로드되었습니다.")
        os.environ["REPLICATE_API_TOKEN"] = api_token
    
    st.divider()
    
    # 화면 비율 설정
    st.subheader("📐 화면 비율")
    aspect_ratios = {
        "16:9": "16:9 (YouTube, 일반 동영상)",
        "4:3": "4:3 (전통적인 TV 형식)",
        "1:1": "1:1 (Instagram 피드)",
        "3:4": "3:4 (Instagram 포트레이트)",
        "9:16": "9:16 (Instagram 릴스, TikTok)",
        "21:9": "21:9 (시네마틱 와이드)",
        "9:21": "9:21 (울트라 세로형)"
    }
    
    selected_ratio = st.selectbox(
        "비율 선택",
        options=list(aspect_ratios.keys()),
        format_func=lambda x: aspect_ratios[x],
        index=0
    )
    
    st.divider()
    
    # Seed 설정
    st.subheader("🎲 랜덤 시드")
    seed = st.number_input(
        "Seed 값",
        min_value=0,
        max_value=999999,
        value=42,
        help="동일한 시드값으로 동일한 결과를 재현할 수 있습니다."
    )
    
    st.divider()
    
    # 고정 설정 표시
    st.subheader("📋 고정 설정")
    st.info("""
    - **재생 시간**: 5초
    - **해상도**: 480p
    """)

# 메인 컨텐츠
col1, col2 = st.columns([1, 1])

with col1:
    st.header("🎯 생성 모드 선택")
    mode = st.radio(
        "모드를 선택하세요:",
        ["텍스트 to 비디오", "이미지 to 비디오"],
        help="텍스트 설명이나 이미지를 기반으로 비디오를 생성합니다."
    )
    
    # 이미지 업로드 (이미지 to 비디오 모드)
    uploaded_image = None
    image_base64 = None
    
    if mode == "이미지 to 비디오":
        st.subheader("📷 이미지 업로드")
        uploaded_file = st.file_uploader(
            "이미지를 선택하세요",
            type=['png', 'jpg', 'jpeg', 'webp'],
            help="업로드한 이미지를 기반으로 비디오가 생성됩니다."
        )
        
        if uploaded_file is not None:
            # 이미지 표시
            uploaded_image = Image.open(uploaded_file)
            st.image(uploaded_image, caption="업로드된 이미지", use_column_width=True)
            
            # 이미지를 base64로 변환
            buffered = io.BytesIO()
            uploaded_image.save(buffered, format="PNG")
            image_base64 = base64.b64encode(buffered.getvalue()).decode()

with col2:
    st.header("✍️ 프롬프트 입력")
    
    if mode == "텍스트 to 비디오":
        prompt_placeholder = "생성할 비디오를 설명해주세요.\n예: The sun rises slowly between tall buildings. [Ground-level follow shot] Bicycle tires roll over a dew-covered street at dawn."
    else:
        prompt_placeholder = "이미지를 어떻게 움직이게 할지 설명해주세요.\n예: Camera slowly zooms in while clouds move across the sky. The subject's hair gently moves in the wind."
    
    prompt = st.text_area(
        "프롬프트",
        height=150,
        placeholder=prompt_placeholder,
        help="자세하고 구체적인 설명일수록 더 좋은 결과를 얻을 수 있습니다."
    )

# 생성 버튼
st.divider()

if st.button("🎬 비디오 생성", type="primary", use_container_width=True):
    # 입력 검증
    if not api_token:
        st.error("❌ API 토큰이 필요합니다. 사이드바에서 설정해주세요.")
    elif not prompt:
        st.error("❌ 프롬프트를 입력해주세요.")
    elif mode == "이미지 to 비디오" and uploaded_image is None:
        st.error("❌ 이미지를 업로드해주세요.")
    else:
        try:
            # 프로그레스 바
            progress_text = "비디오 생성 중... 잠시만 기다려주세요."
            progress_bar = st.progress(0, text=progress_text)
            
            # 입력 파라미터 설정
            input_params = {
                "prompt": prompt,
                "duration": 5,
                "resolution": "480p",
                "aspect_ratio": selected_ratio,
                "seed": seed
            }
            
            # 이미지 to 비디오 모드인 경우
            if mode == "이미지 to 비디오" and image_base64:
                input_params["image"] = f"data:image/png;base64,{image_base64}"
            
            # 진행률 업데이트
            progress_bar.progress(25, text="Replicate API 호출 중...")
            
            # Replicate 실행
            output = replicate.run(
                "bytedance/seedance-1-lite",
                input=input_params
            )
            
            # 진행률 업데이트
            progress_bar.progress(75, text="비디오 다운로드 중...")
            
            # 비디오 저장
            if hasattr(output, 'read'):
                video_data = output.read()
            else:
                # URL인 경우 다운로드
                import requests
                response = requests.get(output)
                video_data = response.content
            
            # 임시 파일로 저장
            with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmp_file:
                tmp_file.write(video_data)
                tmp_filename = tmp_file.name
            
            # 로컬 파일로도 저장
            output_filename = "output.mp4"
            with open(output_filename, "wb") as file:
                file.write(video_data)
            
            # 진행률 완료
            progress_bar.progress(100, text="완료!")
            
            # 성공 메시지
            st.success(f"✅ 비디오가 성공적으로 생성되었습니다! ({output_filename})")
            
            # 비디오 표시
            st.subheader("📹 생성된 비디오")
            st.video(tmp_filename)
            
            # 다운로드 버튼
            col1, col2, col3 = st.columns([1, 2, 1])
            with col2:
                st.download_button(
                    label="⬇️ 비디오 다운로드",
                    data=video_data,
                    file_name="generated_video.mp4",
                    mime="video/mp4",
                    use_container_width=True
                )
            
            # 생성 정보 표시
            with st.expander("📊 생성 정보"):
                st.json({
                    "mode": mode,
                    "aspect_ratio": selected_ratio,
                    "seed": seed,
                    "duration": "5초",
                    "resolution": "480p",
                    "prompt": prompt[:100] + "..." if len(prompt) > 100 else prompt
                })
                
        except Exception as e:
            st.error(f"❌ 오류가 발생했습니다: {str(e)}")
            st.info("💡 API 토큰이 올바른지, 모델이 사용 가능한지 확인해주세요.")

# 사용 방법
with st.expander("📖 사용 방법"):
    st.markdown("""
    ### 설치 및 실행
    
    1. **필요한 패키지 설치**:
    ```bash
    pip install streamlit replicate pillow requests
    ```
    
    2. **환경변수 설정** (선택사항):
    ```bash
    export RAPI_TOKEN="your-replicate-api-token"
    ```
    
    3. **애플리케이션 실행**:
    ```bash
    streamlit run video_generator.py
    ```
    
    ### 기능 설명
    
    - **텍스트 to 비디오**: 텍스트 설명만으로 비디오를 생성합니다.
    - **이미지 to 비디오**: 업로드한 이미지를 움직이는 비디오로 변환합니다.
    - **화면 비율**: 다양한 SNS 플랫폼에 최적화된 비율을 선택할 수 있습니다.
    - **Seed 값**: 동일한 시드값으로 동일한 결과를 재현할 수 있습니다.
    
    ### 팁
    
    - 구체적이고 상세한 프롬프트일수록 더 좋은 결과를 얻을 수 있습니다.
    - 카메라 움직임, 조명, 분위기 등을 설명에 포함시켜보세요.
    - 이미지 to 비디오 모드에서는 이미지의 어떤 부분을 어떻게 움직일지 설명하세요.
    """)

# 푸터
st.divider()
st.markdown(
    """
    <div style='text-align: center; color: gray;'>
        <p>Powered by Replicate AI and bytedance/seedance-1-lite model</p>
    </div>
    """,
    unsafe_allow_html=True
)