Update app.py

app.py

@@ -1,73 +1,155 @@
+"""
+🎵 Audio Segment Extractor
+- 오디오 파일 업로드
+- 타임라인 시각화
+- 시작~종료 시간 구간 추출
+- 다운로드
+"""
+
 import gradio as gr
-import cv2
-import tempfile
 import os
-from PIL import Image
-
-def extract_last_frame(video_file):
-    """
-    비디오 파일에서 마지막 프레임을 추출하여 이미지로 반환
-    """
-    if video_file is None:
-        return None, "⚠️ Please upload a video file first!"
+import json
+import subprocess
+import tempfile
+import time
+
+# ============================================
+# 오디오 처리용 임시 디렉토리
+# ============================================
+UPLOAD_DIR = tempfile.mkdtemp()
+
+
+# ============================================
+# 🎵 오디오 처리 함수들
+# ============================================
+
+def get_audio_info_and_waveform(audio_file):
+    """오디오 정보 조회 및 파형 데이터 생성"""
+    if not audio_file:
+        return None, 0, 10, "🎵 파일을 업로드하세요", None
     
     try:
-        # OpenCV로 비디오 열기
-        cap = cv2.VideoCapture(video_file)
+        path = audio_file.name if hasattr(audio_file, 'name') else audio_file
+        
+        # 오디오 정보 조회
+        probe_cmd = ['ffprobe', '-v', 'error', 
+                     '-show_entries', 'format=duration,bit_rate,format_name:stream=sample_rate,channels,codec_name',
+                     '-of', 'json', path]
+        result = subprocess.run(probe_cmd, capture_output=True, text=True, timeout=30)
+        info = json.loads(result.stdout)
+        
+        format_info = info.get('format', {})
+        stream_info = info.get('streams', [{}])[0] if info.get('streams') else {}
         
-        if not cap.isOpened():
-            return None, "❌ Error: Cannot open video file!"
+        duration = float(format_info.get('duration', 0))
+        bit_rate = int(format_info.get('bit_rate', 0)) // 1000 if format_info.get('bit_rate') else 0
+        format_name = format_info.get('format_name', 'N/A')
+        sample_rate = stream_info.get('sample_rate', 'N/A')
+        channels = stream_info.get('channels', 'N/A')
+        codec = stream_info.get('codec_name', 'N/A')
         
-        # 비디오 정보 가져오기
-        total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-        fps = cap.get(cv2.CAP_PROP_FPS)
-        width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-        height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-        duration = total_frames / fps if fps > 0 else 0
+        # 파일 크기
+        file_size = os.path.getsize(path) / (1024 * 1024)  # MB
         
-        if total_frames <= 0:
-            cap.release()
-            return None, "❌ Error: Video has no frames!"
+        info_text = f"""🎵 오디오 정보
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+⏱️ 길이: {duration:.2f}초 ({duration/60:.1f}분)
+🎚️ 비트레이트: {bit_rate} kbps
+🔊 샘플레이트: {sample_rate} Hz
+📢 채널: {channels}
+🎛️ 코덱: {codec}
+📁 포맷: {format_name}
+💾 크기: {file_size:.2f} MB
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"""
         
-        # 마지막 프레임으로 이동
-        cap.set(cv2.CAP_PROP_POS_FRAMES, total_frames - 1)
+        # 오디오 미리듣기용 반환
+        return path, 0, duration, info_text, gr.update(maximum=duration, value=[0, min(duration, 10)])
         
-        # 프레임 읽기
-        ret, frame = cap.read()
-        cap.release()
+    except Exception as e:
+        return None, 0, 10, f"❌ 정보 조회 실패: {str(e)}", gr.update(maximum=100, value=[0, 10])
+
+
+def extract_audio_segment(audio_file, time_range):
+    """오디오 구간 추출"""
+    if not audio_file:
+        return None, "❌ 오디오 파일을 업로드하세요", None
+    
+    try:
+        path = audio_file.name if hasattr(audio_file, 'name') else audio_file
+        start_time = float(time_range[0]) if time_range else 0
+        end_time = float(time_range[1]) if time_range else 10
+        
+        if end_time <= start_time:
+            return None, "❌ 종료 시간이 시작 시간보다 커야 합니다", None
         
-        if not ret:
-            return None, "❌ Error: Cannot read the last frame!"
+        duration = end_time - start_time
         
-        # BGR to RGB 변환
-        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        # 출력 파일 확장자 결정
+        ext = os.path.splitext(path)[1].lower()
+        if ext not in ['.mp3', '.wav', '.ogg', '.m4a', '.aac', '.flac']:
+            ext = '.mp3'
         
-        # PIL Image로 변환
-        image = Image.fromarray(frame_rgb)
+        output_path = os.path.join(UPLOAD_DIR, f"audio_cut_{int(time.time())}{ext}")
         
-        # 정보 로그 생성
-        info_log = f"""✅ EXTRACTION COMPLETE!
-{'=' * 50}
-📹 Video Info:
-   • Total Frames: {total_frames:,}
-   • FPS: {fps:.2f}
-   • Duration: {duration:.2f} seconds
-   • Resolution: {width} x {height}
-{'=' * 50}
-🖼️ Extracted Frame:
-   • Frame Number: {total_frames} (Last Frame)
-   • Image Size: {width} x {height}
-{'=' * 50}
-💾 Ready to download!"""
+        # 먼저 스트림 복사 시도 (빠름, 무손실)
+        cmd = [
+            'ffmpeg', '-y', '-i', path,
+            '-ss', str(start_time), '-t', str(duration),
+            '-c', 'copy', output_path
+        ]
         
-        return image, info_log
+        result = subprocess.run(cmd, capture_output=True, text=True, timeout=120)
         
+        # copy가 실패하면 재인코딩 시도
+        if not os.path.exists(output_path) or os.path.getsize(output_path) == 0:
+            output_path = os.path.join(UPLOAD_DIR, f"audio_cut_{int(time.time())}.mp3")
+            cmd = [
+                'ffmpeg', '-y', '-i', path,
+                '-ss', str(start_time), '-t', str(duration),
+                '-acodec', 'libmp3lame', '-b:a', '192k', output_path
+            ]
+            subprocess.run(cmd, capture_output=True, timeout=120)
+        
+        if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
+            file_size = os.path.getsize(output_path) / 1024  # KB
+            
+            status_text = f"""✅ 오디오 추출 완료!
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
+⏱️ 시작: {start_time:.2f}초
+⏱️ 종료: {end_time:.2f}초
+⏱️ 길이: {duration:.2f}초
+📁 크기: {file_size:.1f} KB
+━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"""
+            
+            return output_path, status_text, output_path
+        else:
+            return None, "❌ 오디오 추출 실패", None
+            
     except Exception as e:
-        return None, f"❌ Error: {str(e)}"
+        return None, f"❌ 오류: {str(e)}", None
+
+
+def format_time(seconds):
+    """초를 MM:SS.ms 형식으로 변환"""
+    if seconds is None:
+        return "00:00.00"
+    mins = int(seconds // 60)
+    secs = seconds % 60
+    return f"{mins:02d}:{secs:05.2f}"
+
+
+def update_time_display(time_range):
+    """슬라이더 값에 따른 시간 표시 업데이트"""
+    if time_range:
+        start = time_range[0]
+        end = time_range[1]
+        duration = end - start
+        return f"🕐 {format_time(start)} → {format_time(end)} (길이: {duration:.2f}초)"
+    return "🕐 구간을 선택하세요"
 
 
 # ============================================
-# 🎨 Comic Classic Theme - Toon Playground
+# 🎨 Comic Classic Theme CSS
 # ============================================
 
 css = """
@@ -84,15 +166,26 @@ css = """
     font-family: 'Comic Neue', cursive, sans-serif !important;
 }
 
-/* ===== 허깅페이스 상단 요소 숨김 ===== */
+/* ===== 허깅페이스 상단 요소 숨김 (강화) ===== */
+.tabs, 
+.tab-nav,
 .huggingface-space-header,
 #space-header,
 .space-header,
 [class*="space-header"],
+.h-header,
+.gradio-container > header,
+.contain > header,
 .svelte-1ed2p3z,
 .space-header-badge,
 .header-badge,
 [data-testid="space-header"],
+.gr-prose > div:first-child,
+.contain > div:first-child > div:first-child,
+.app > div:first-child > header,
+iframe[src*="huggingface"],
+.huggingface-logo,
+.hf-logo,
 .svelte-kqij2n,
 .svelte-1ax1toq,
 .embed-container > div:first-child {
@@ -237,8 +330,7 @@ button.primary:active,
 
 /* ===== 🎨 Secondary 버튼 - 코믹 레드 ===== */
 .gr-button-secondary,
-button.secondary,
-.extract-btn {
+button.secondary {
     background: #EF4444 !important;
     border: 3px solid #1F2937 !important;
     border-radius: 8px !important;
@@ -253,43 +345,57 @@ button.secondary,
 }
 
 .gr-button-secondary:hover,
-button.secondary:hover,
-.extract-btn:hover {
+button.secondary:hover {
     background: #DC2626 !important;
     transform: translate(-2px, -2px) !important;
     box-shadow: 6px 6px 0px #1F2937 !important;
 }
 
 .gr-button-secondary:active,
-button.secondary:active,
-.extract-btn:active {
+button.secondary:active {
     transform: translate(2px, 2px) !important;
     box-shadow: 2px 2px 0px #1F2937 !important;
 }
 
-/* ===== 🎨 로그 출력 영역 ===== */
-.info-log textarea {
-    background: #1F2937 !important;
-    color: #10B981 !important;
-    font-family: 'Courier New', monospace !important;
-    font-size: 0.9rem !important;
-    font-weight: 400 !important;
-    border: 3px solid #10B981 !important;
+/* ===== 🎨 Extract 버튼 - 코믹 그린 ===== */
+.extract-btn {
+    background: #10B981 !important;
+    border: 3px solid #1F2937 !important;
     border-radius: 8px !important;
-    box-shadow: 4px 4px 0px #10B981 !important;
+    color: #FFFFFF !important;
+    font-family: 'Bangers', cursive !important;
+    font-weight: 400 !important;
+    font-size: 1.3rem !important;
+    letter-spacing: 2px !important;
+    box-shadow: 5px 5px 0px #1F2937 !important;
+    transition: all 0.1s ease !important;
+    text-shadow: 1px 1px 0px #1F2937 !important;
 }
 
-/* ===== �� 비디오 업로드 영역 ===== */
-.video-upload {
-    border: 4px dashed #3B82F6 !important;
-    border-radius: 12px !important;
-    background: #EFF6FF !important;
-    transition: all 0.2s ease !important;
+.extract-btn:hover {
+    background: #059669 !important;
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 7px 7px 0px #1F2937 !important;
+}
+
+.extract-btn:active {
+    transform: translate(3px, 3px) !important;
+    box-shadow: 2px 2px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 슬라이더 ===== */
+input[type="range"] {
+    accent-color: #EF4444 !important;
+    height: 8px !important;
+}
+
+.gr-slider {
+    background: #FACC15 !important;
 }
 
-.video-upload:hover {
-    border-color: #EF4444 !important;
-    background: #FEF2F2 !important;
+/* ===== 🎨 Range Slider 특별 스타일 ===== */
+.range-slider input[type="range"] {
+    accent-color: #10B981 !important;
 }
 
 /* ===== 🎨 아코디언 - 말풍선 스타일 ===== */
@@ -307,16 +413,6 @@ button.secondary:active,
     font-size: 1.1rem !important;
 }
 
-/* ===== 🎨 이미지 출력 영역 ===== */
-.gr-image,
-.image-container {
-    border: 4px solid #1F2937 !important;
-    border-radius: 8px !important;
-    box-shadow: 8px 8px 0px #1F2937 !important;
-    overflow: hidden !important;
-    background: #FFFFFF !important;
-}
-
 /* ===== 🎨 라벨 스타일 ===== */
 label,
 .gr-input-label,
@@ -339,12 +435,46 @@ span.gr-label {
     font-size: 0.9rem !important;
 }
 
-/* ===== 🎨 프로그레스 바 ===== */
-.progress-bar,
-.gr-progress-bar {
-    background: #3B82F6 !important;
-    border: 2px solid #1F2937 !important;
-    border-radius: 4px !important;
+/* ===== 🎨 상태 텍스트 영역 ===== */
+.status-text textarea {
+    background: #E0F2FE !important;
+    border: 3px solid #0EA5E9 !important;
+    color: #0C4A6E !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-weight: 700 !important;
+    font-size: 0.95rem !important;
+}
+
+/* ===== 🎨 정보 박스 ===== */
+.info-box textarea {
+    background: #FEF3C7 !important;
+    border: 3px solid #F59E0B !important;
+    color: #92400E !important;
+    font-family: 'Courier New', monospace !important;
+    font-weight: 600 !important;
+    font-size: 0.9rem !important;
+}
+
+/* ===== 🎨 시간 표시 ===== */
+.time-display {
+    font-family: 'Bangers', cursive !important;
+    font-size: 1.5rem !important;
+    color: #1F2937 !important;
+    text-align: center !important;
+    padding: 15px !important;
+    background: linear-gradient(135deg, #FACC15 0%, #FDE68A 100%) !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+    letter-spacing: 1px !important;
+}
+
+/* ===== 🎨 오디오 플레이어 ===== */
+audio {
+    width: 100% !important;
+    border-radius: 8px !important;
+    border: 3px solid #1F2937 !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
 }
 
 /* ===== 🎨 스크롤바 - 코믹 스타일 ===== */
@@ -379,10 +509,12 @@ a {
     color: #3B82F6 !important;
     text-decoration: none !important;
     font-weight: 700 !important;
+    border-bottom: 2px solid #3B82F6 !important;
 }
 
 a:hover {
     color: #EF4444 !important;
+    border-bottom-color: #EF4444 !important;
 }
 
 /* ===== 🎨 Row/Column 간격 ===== */
@@ -394,6 +526,18 @@ a:hover {
     gap: 1rem !important;
 }
 
+/* ===== 🎨 파일 업로드 영역 ===== */
+.gr-file-upload {
+    border: 3px dashed #1F2937 !important;
+    border-radius: 8px !important;
+    background: #FEF9C3 !important;
+}
+
+.gr-file-upload:hover {
+    border-color: #3B82F6 !important;
+    background: #EFF6FF !important;
+}
+
 /* ===== 반응형 조정 ===== */
 @media (max-width: 768px) {
     .header-text h1 {
@@ -413,6 +557,10 @@ a:hover {
     .block {
         box-shadow: 4px 4px 0px #1F2937 !important;
     }
+    
+    .time-display {
+        font-size: 1.2rem !important;
+    }
 }
 
 /* ===== 🎨 다크모드 비활성화 (코믹은 밝아야 함) ===== */
@@ -423,7 +571,11 @@ a:hover {
 }
 """
 
-# Build the Gradio interface
+
+# ============================================
+# Gradio UI
+# ============================================
+
 with gr.Blocks(fill_height=True, css=css) as demo:
     
     # HOME Badge
@@ -438,73 +590,126 @@ with gr.Blocks(fill_height=True, css=css) as demo:
     # Header Title
     gr.Markdown(
         """
-        # 🎬 VIDEO LAST FRAME EXTRACTOR 🖼️
+        # 🎵 AUDIO SEGMENT EXTRACTOR 🎵
         """,
         elem_classes="header-text"
     )
     
     gr.Markdown(
         """
-        <p class="subtitle">📹 Upload a video and extract the LAST FRAME instantly! 💾</p>
+        <p class="subtitle">✂️ 오디오 파일에서 원하는 구간만 잘라내기! 🎶</p>
         """,
     )
     
     with gr.Row(equal_height=False):
-        # Left column - Input
-        with gr.Column(scale=1, min_width=320):
-            video_input = gr.Video(
-                label="📹 Upload Your Video",
-                sources=["upload"],
-                elem_classes="video-upload"
+        # ============================================
+        # Left Column - Input
+        # ============================================
+        with gr.Column(scale=1, min_width=350):
+            # 파일 업로드
+            audio_input = gr.File(
+                label="🎵 오디오 파일 업로드",
+                file_types=["audio"],
+            )
+            
+            # 오디오 정보
+            audio_info = gr.Textbox(
+                label="📊 오디오 정보",
+                placeholder="오디오를 업로드하면 정보가 표시됩니다...",
+                lines=10,
+                interactive=False,
+                elem_classes="info-box"
+            )
+            
+            # 원본 오디오 미리듣기
+            gr.Markdown("### 🎧 원본 오디오 미리듣기")
+            original_audio = gr.Audio(
+                label="",
+                type="filepath",
+                interactive=False,
+            )
+        
+        # ============================================
+        # Right Column - Controls & Output
+        # ============================================
+        with gr.Column(scale=1, min_width=350):
+            # 구간 선택
+            gr.Markdown("### ✂️ 추출 구간 선택")
+            
+            # 타임라인 슬라이더 (Range Slider)
+            time_range = gr.Slider(
+                minimum=0,
+                maximum=100,
+                value=[0, 10],
+                step=0.1,
+                label="⏱️ 구간 선택 (초)",
+                info="슬라이더를 드래그하여 시작/종료 시간을 선택하세요",
+                elem_classes="range-slider"
+            )
+            
+            # 시간 표시
+            time_display = gr.Textbox(
+                label="",
+                value="🕐 구간을 선택하세요",
+                interactive=False,
+                elem_classes="time-display"
             )
             
+            # 추출 버튼
             extract_btn = gr.Button(
-                "🎬 EXTRACT LAST FRAME! 🖼️",
+                "✂️ 오디오 구간 추출!",
                 variant="primary",
                 size="lg",
                 elem_classes="extract-btn"
             )
             
-            with gr.Accordion("📜 Extraction Info", open=True):
-                info_log = gr.Textbox(
-                    label="",
-                    placeholder="Upload a video and click extract to see info...",
-                    lines=12,
-                    max_lines=20,
-                    interactive=False,
-                    elem_classes="info-log"
-                )
-        
-        # Right column - Output
-        with gr.Column(scale=1, min_width=320):
-            output_image = gr.Image(
-                label="🖼️ Last Frame",
-                type="pil",
-                show_label=True,
-                height=500,
+            # 추출 상태
+            extract_status = gr.Textbox(
+                label="📋 추출 상태",
+                placeholder="추출 결과가 여기에 표시됩니다...",
+                lines=8,
+                interactive=False,
+                elem_classes="status-text"
             )
             
-            gr.Markdown(
-                """
-                <p style="text-align: center; margin-top: 10px; font-weight: 700; color: #1F2937;">
-                    💡 Right-click on the image to save, or use the download button!
-                </p>
-                """
+            # 추출된 오디오 미리듣기
+            gr.Markdown("### 🎧 추출된 오디오 미리듣기")
+            extracted_audio = gr.Audio(
+                label="",
+                type="filepath",
+                interactive=False,
             )
-
-    # Connect the extract button
-    extract_btn.click(
-        fn=extract_last_frame,
-        inputs=[video_input],
-        outputs=[output_image, info_log],
+            
+            # 다운로드
+            download_file = gr.File(
+                label="📥 다운로드",
+            )
+    
+    # ============================================
+    # Event Handlers
+    # ============================================
+    
+    # 파일 업로드 시 정보 조회
+    audio_input.change(
+        fn=get_audio_info_and_waveform,
+        inputs=[audio_input],
+        outputs=[original_audio, gr.State(), gr.State(), audio_info, time_range]
     )
     
-    # Auto-extract when video is uploaded
-    video_input.change(
-        fn=extract_last_frame,
-        inputs=[video_input],
-        outputs=[output_image, info_log],
+    # 슬라이더 변경 시 시간 표시 업데이트
+    time_range.change(
+        fn=update_time_display,
+        inputs=[time_range],
+        outputs=[time_display]
     )
+    
+    # 추출 버튼 클릭
+    extract_btn.click(
+        fn=extract_audio_segment,
+        inputs=[audio_input, time_range],
+        outputs=[extracted_audio, extract_status, download_file]
+    )
+
 
 if __name__ == "__main__":
     demo.launch()