app.py

import os
import streamlit as st
from moviepy.editor import VideoFileClip
from datetime import datetime

# MP4 -> MP3 변환 함수
def convert_mp4_to_mp3(video_file_path, output_dir):
    video = VideoFileClip(video_file_path)
    audio = video.audio
    # 현재 시간 기반으로 파일명 생성
    timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
    output_path = os.path.join(output_dir, f"video_{timestamp}.mp3")
    audio.write_audiofile(output_path)
    audio.close()
    video.close()
    return output_path

# Streamlit 앱 인터페이스
def main():
    st.title("MP4 to MP3 Converter")
    st.write("Upload an MP4 video file to convert it to MP3 format.")

    # 파일 업로드 위젯
    uploaded_file = st.file_uploader("Choose an MP4 file", type=["mp4"])

    if uploaded_file is not None:
        # 변환 및 저장 경로 설정
        modeltarget = "mp4_to_mp3_conversion"
        save_path = os.path.join("/home/user/app", modeltarget)
        os.makedirs(save_path, exist_ok=True)

        # 업로드된 파일을 임시 파일로 저장
        video_file_path = os.path.join(save_path, uploaded_file.name)
        with open(video_file_path, "wb") as f:
            f.write(uploaded_file.getbuffer())
        
        # MP4 -> MP3 변환
        try:
            mp3_file_path = convert_mp4_to_mp3(video_file_path, save_path)
            st.success("Conversion successful!")

            # MP3 파일 다운로드 링크 생성
            with open(mp3_file_path, "rb") as mp3_file:
                st.download_button(
                    label="Download MP3",
                    data=mp3_file,
                    file_name=os.path.basename(mp3_file_path),
                    mime="audio/mpeg"
                )
        except Exception as e:
            st.error(f"Error occurred during conversion: {str(e)}")

if __name__ == "__main__":
    main()


# import os
# import gradio as gr
# from moviepy.editor import VideoFileClip

# def convert_mp4_to_mp3(video_file_path, output_dir):
#     # MP3 변환 과정
#     video = VideoFileClip(video_file_path)
#     audio = video.audio
#     output_path = os.path.join(output_dir, os.path.splitext(os.path.basename(video_file_path))[0] + ".mp3")
#     audio.write_audiofile(output_path)
#     audio.close()
#     video.close()
#     return output_path

# def mp4_to_mp3_converter(video_file):
#     # 비디오 파일이 없을 때 처리
#     if video_file is None:
#         return "Error: No video file was uploaded."

#     modeltarget = "mp4_to_mp3_conversion"
#     save_path = os.path.join("/home/user/app", modeltarget)
#     os.makedirs(save_path, exist_ok=True)

#     # 비디오 파일 경로가 올바른지 확인 후 변환
#     try:
#         mp3_file_path = convert_mp4_to_mp3(video_file.name, save_path)
#         return mp3_file_path
#     except Exception as e:
#         return f"Error occurred during conversion: {str(e)}"

# # Gradio 인터페이스 설정
# iface = gr.Interface(
#     fn=mp4_to_mp3_converter,
#     inputs=gr.File(label="Input Video"),
#     outputs="text",  # 오류 메시지 또는 파일 경로 반환
#     title="MP4 to MP3 Converter",
#     description="Upload a video file to convert it to MP3 format."
# )

# if __name__ == "__main__":
#     iface.launch()