app.py

import streamlit as st
import requests
import streamlit.components.v1 as components
from gtts import gTTS
from gtts.lang import tts_langs
from io import BytesIO
import os

# 페이지 설정
st.set_page_config(page_title="ViDraft", layout="wide")

# 사이드바 타이틀 설정
st.sidebar.title("ViDraft")

# 메뉴 및 하위 메뉴 정의
menus = {
    "Home": [],
    "Free Stock": ["Template Video", "Search Video", "Search Image"],
    "Image": ["Face Swap", "Remove Background", "Compositing", "Image Matching", "URL Screenshot"],
    "Video": ["Video Music", "Remove Background", "Compositing"],
    "Sound": ["TTS(Voice)" ,"Image SFX", "Video SFX"],
    "Scripts": []
}

# 세션 상태 초기화
if 'current_menu' not in st.session_state:
    st.session_state['current_menu'] = 'Home'
if 'current_sub_menu' not in st.session_state:
    st.session_state['current_sub_menu'] = ''

# 메인 메뉴 선택
selected_menu = st.sidebar.selectbox("Menu", list(menus.keys()), key='main_menu')
st.session_state['current_menu'] = selected_menu

# 하위 메뉴 선택 (해당되는 경우)
if selected_menu in menus:
    selected_sub_menu = st.sidebar.selectbox("Sub Menu", [""] + menus[selected_menu], key=f'sub_menu_{selected_menu}')
    st.session_state['current_sub_menu'] = selected_sub_menu
else:
    st.session_state['current_sub_menu'] = ''


# 'Sound' 메뉴의 'TTS(Voice)' 선택 시
if selected_menu == "Sound" and selected_sub_menu == "TTS(Voice)":
    st.header("Text-to-Speech")
    # 텍스트 입력
    text = st.text_area("Enter text to synthesize", "Hello, welcome to ViDraft TTS service.")
    
    # 지원되는 언어 목록을 불러옵니다.
    languages_dict = tts_langs()
    # ISO 639-1 표준에 따라 두 글자 코드를 가진 언어만 필터링
    two_letter_languages = {code: lang for code, lang in languages_dict.items() if len(code) == 2}

    # 언어 선택을 위한 selectbox를 생성합니다.
    selected_language_code = st.selectbox(
        "Choose Language", 
        options=list(two_letter_languages.keys()), 
        format_func=lambda x: f"{two_letter_languages[x]} ({x})",
        index=list(two_letter_languages.keys()).index('en')  # 'en'을 기본 언어로 설정
    )
    
    # 'Synthesize' 버튼
    if st.button("Synthesize"):
        if text:
            try:
                # 선택된 언어로 gTTS 객체 생성
                tts = gTTS(text=text, lang=selected_language_code, slow=False)
                audio_file = BytesIO()
                tts.write_to_fp(audio_file)
                audio_file.seek(0)
                # 생성된 오디오 파일을 재생
                st.audio(audio_file, format="audio/mp3")
            except Exception as e:
                st.error(f"Error: {e}")
        else:
            st.warning("Please enter some text to synthesize.")

            
            
# Pexels API 키 설정
PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"

def search_images(keyword, per_page=80):
    """이미지 검색 함수"""
    url = f"https://api.pexels.com/v1/search?query={keyword}&per_page={per_page}"
    headers = {"Authorization": PEXELS_API_KEY}
    response = requests.get(url, headers=headers).json()
    images = []
    if 'photos' in response:
        for photo in response['photos']:
            images.append(photo['src']['original'])
    return images

def search_videos(keyword, per_page=80):
    """비디오 검색 함수"""
    url = f"https://api.pexels.com/videos/search?query={keyword}&per_page={per_page}"
    headers = {"Authorization": PEXELS_API_KEY}
    response = requests.get(url, headers=headers).json()
    videos = []
    if 'videos' in response:
        for video in response['videos']:
            videos.append(video['video_files'][0]['link'])
    return videos

# 'Search Image' 선택 시
if selected_menu == "Free Stock" and st.session_state['current_sub_menu'] == "Search Image":
    keyword = st.text_input("Enter a keyword to search for images")
    if keyword:
        images = search_images(keyword)
        cols = st.columns(3)  # 3열로 이미지 표시
        for idx, img in enumerate(images):
            with cols[idx % 3]:
                st.image(img)

# 'Search Video' 선택 시
if selected_menu == "Free Stock" and st.session_state['current_sub_menu'] == "Search Video":
    keyword = st.text_input("Enter a keyword to search for videos")
    if keyword:
        videos = search_videos(keyword)
        cols = st.columns(3)  # 3열로 비디오 표시
        for idx, video in enumerate(videos):
            with cols[idx % 3]:
                st.video(video)

                
# 'Image' 메뉴에서 'Remove Background' 선택 시
if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Remove Background":
    st.header("Remove Background")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidnuki.hf.space", width=None, height=768, scrolling=True)

# 'Image' 메뉴에서 'Compositing' 선택 시
if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Compositing":
    st.header("Compositing")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidistudio.hf.space", width=None, height=1800, scrolling=True)
     
# 'Image' 메뉴에서 'Face Swap' 선택 시
if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Face Swap":
    st.header("Face Swap")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidifs.hf.space", width=None, height=1800, scrolling=True)

# 'video' 메뉴에서 'Compositing' 선택 시
if selected_menu == "Video" and st.session_state['current_sub_menu'] == "Compositing":
    st.header("Compositing")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidvstudio.hf.space", width=None, height=1800, scrolling=True)
         
# 'video' 메뉴에서 'Compositing' 선택 시
if selected_menu == "Video" and st.session_state['current_sub_menu'] == "Remove Background":
    st.header("Remove Background")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidvback.hf.space", width=None, height=1800, scrolling=True)

    
# 'sound' 메뉴에서 'SFX' 선택 시
if selected_menu == "Sound" and st.session_state['current_sub_menu'] == "Image SFX":
    st.header("Image SFX")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidisfx.hf.space", width=None, height=1800, scrolling=True)

# 'sound' 메뉴에서 'SFX' 선택 시
if selected_menu == "Sound" and st.session_state['current_sub_menu'] == "Video SFX":
    st.header("Video SFX")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidvsfx.hf.space", width=None, height=1800, scrolling=True)
    

# 'Image' 메뉴에서 'Image Matching' 선택 시
if selected_menu == "Image" and st.session_state['current_sub_menu'] == "Image Matching":
    st.header("Image Matching")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidimatch.hf.space", width=None, height=1800, scrolling=True)
     
# 'Image' 메뉴에서 'URL Screenshot' 선택 시
if selected_menu == "Image" and st.session_state['current_sub_menu'] == "URL Screenshot":
    st.header("URL Screenshot")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidiscreen.hf.space", width=None, height=1800, scrolling=True)

# 'sound' 메뉴에서 'Video Music' 선택 시
if selected_menu == "Sound" and st.session_state['current_sub_menu'] == "Video Music":
    st.header("Video Music")
    # iframe을 사용하여 외부 URL 임베드
    components.iframe("https://arxivgpt-vidsm.hf.space", width=None, height=1800, scrolling=True)
    


# 'Home' 페이지 표시
if selected_menu == "Home":
    st.image("banner2.jpg", use_column_width=True)    
    st.header("Welcome to ViDraft")
    st.header("'Create Contents, ViDraft Value-UP'")

    # 환영 메시지 및 서비스 설명
    welcome_text = """
    Welcome to our dynamic platform, where creativity meets technology across a vast spectrum of multimedia services. Our service offers an extensive array of options for every multimedia need:

    - **Video Template (OpenAI SORA)**: "Discover a world of creativity with OpenAI SORA's Video Templates. From captivating intros to engaging outros, our AI-powered templates transform your visions into cinematic realities."
    - **Free Stock: Video Search**: "Dive into an expansive ocean of premium stock videos. Whether it's breathtaking landscapes or dynamic urban life, find the perfect clip to elevate your project, all at your fingertips."
    - **Free Stock: Image Search**: "Explore a vast gallery of stunning images, curated to perfection. From the serene to the spectacular, uncover the ideal visual complement for your next creative endeavor."
    - **Multilingual Text to Speech(TTS)**: "Bring your content to life in any language with our Multilingual Text to Speech technology. Experience seamless, natural-sounding voices that captivate your audience, no matter where they are."
    - **Remove Image Background**: "Eradicate distractions and focus on what matters. Our AI swiftly removes any background, offering you a clean slate to unleash your creativity or perfectly blend your subject into new environments."
    - **Remove Video Background**: "Transform your videos with the ability to remove and replace backgrounds effortlessly. Engage your audience with stunning visuals, no green screen required."
    - **Video Compositing**: "Merge reality with imagination. Our Video Compositing technology seamlessly integrates various elements into your footage, creating a cohesive and visually stunning masterpiece."
    - **Image Compositing**: "Elevate your images to art. Combine multiple elements with unparalleled precision, crafting compositions that tell a story, evoke emotions, or simply astound."
    - **Face Swap Image**: "Step into a new identity or bring historical figures to life. Our Face Swap technology allows you to seamlessly transpose faces in images, unlocking endless creative possibilities."
    - **Image SFX**: "Enhance your images with special effects that captivate. From subtle enhancements to dramatic transformations, our Image SFX tools let your creativity run wild."
    - **Video SFX**: "Elevate your videos with extraordinary special effects. Whether you're creating an action-packed adventure or a serene landscape, our Video SFX library brings your vision to life."
    - **Image Matching**: "Unveil the synergy between visuals with our Image Matching technology. Let AI seamlessly analyze and match disparate images, bridging the gap between creativity and coherence."
    - **Remote URL Screenshot**: "Capture the essence of any website instantly with Remote URL Screenshot. Simply input a URL and let our system fetch and present a crisp snapshot of the site's main page, effortlessly."

    Join us to transform your creative ideas into reality with cutting-edge technology designed for creators, by creators.

    - **Contact: [email protected] **  Powered by ArXivGPT, Huggingface, Gradio, Streamlit
    """    
    st.write(welcome_text)

    # 이미지들을 한 줄에 3장씩 배치
    image_files = [
        "ViDraft-Video-Templet.png",
        "ViDraft-Video-search.png",
        "ViDraft-Image-search.png",
        "ViDraft-TTS.png",
        "ViDraft-image-remove background.png",
        "ViDraft-video-remove-back.png",
        "ViDraft-video-comp.png",
        "ViDraft-image-comp.png",
        "ViDraft-image-face swap.png",
        "ViDraft-image-sfx.png",
        "ViDraft-video-sfx.png",
        "ViDraft-image-screen.png",
        "ViDraft-image-match.png"
    ]
    
    image_texts = [
        "Video Template(OpenAI SORA)",
        "Free Stock: Video Search",
        "Free Stock: Image Search",
        "Multilingual Text to Speech(TTS)",
        "Remove image background",
        "Remove Video background",
        "Video Compositing",
        "Image Compositing",
        "Face Swap Image",
        "Image SFX",
        "Video SFX",
        "Remote URL Screenshot",
        "Image Matching"
    ]

    for i in range(0, len(image_files), 3):
        cols = st.columns(3)
        for idx, col in enumerate(cols):
            if i + idx < len(image_files):
                col.image(image_files[i + idx], use_column_width=True)
                col.markdown(f"<div style='text-align: center; font-weight: bold; font-size: 20px;'>{image_texts[i + idx]}</div>", unsafe_allow_html=True)





elif selected_menu == "Free Stock":
    # 'Free Stock' 메뉴 선택 시 처리 로직
    if st.session_state['current_sub_menu'] == "Template Video":
        st.header("Template Videos")
        st.write("Explore a variety of video templates crafted with the innovative technology of OpenAI SORA. Dive into an immersive experience with samples generated by the cutting-edge AI, showcasing the potential to transform your creative ideas into captivating visual stories.")
        # 비디오 파일 경로 설정 및 비디오 표시 로직
        video_files = ["ex1.mp4", "ex2.mp4", "ex3.mp4", "ex4.mp4", "ex5.mp4", "ex6.mp4", "ex7.mp4", "ex8.mp4", "ex9.mp4", "ex10.mp4", "ex11.mp4", "ex12.mp4", "ex13.mp4", "ex14.mp4", "ex15.mp4", "ex16.mp4", "ex17.mp4", "ex18.mp4"]
        # 갤러리 형태로 비디오 표시
        cols = st.columns(2)  # 2개의 열로 넓게 배치
        for index, video_file in enumerate(video_files):
            with cols[index % 2]:
                st.video(video_file)
        
# 'Scripts' 메뉴 선택 시 처리 로직
elif selected_menu == "Scripts":
    st.header("Scripts")
    st.write("Script writing tools and resources.")