Create app.py

app.py

@@ -0,0 +1,291 @@
+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
+)