Create fma_player.py

fma_player.py

@@ -0,0 +1,260 @@
+import gradio as gr
+import cv2
+import numpy as np
+import os
+import requests
+from PIL import Image
+import io
+import time
+import tempfile
+from IPython.display import display, HTML
+
+# Costanti per il rilevamento
+MATCH_THRESHOLD = 0.8  # Soglia per il template matching (0-1)
+
+# Funzione per caricare un'immagine da file o URL
+def load_image(path_or_url):
+    try:
+        # Se è un URL
+        if path_or_url.startswith(('http://', 'https://')):
+            response = requests.get(path_or_url)
+            img = Image.open(io.BytesIO(response.content))
+            return np.array(img)
+        # Se è un percorso locale
+        else:
+            return cv2.imread(path_or_url)
+    except Exception as e:
+        print(f"Errore nel caricamento dell'immagine: {e}")
+        return None
+
+# Funzione per il template matching
+def compute_similarity(frame, ref):
+    if frame is None or ref is None:
+        return 0
+    
+    # Converto da RGB a BGR se necessario
+    if len(frame.shape) == 3 and frame.shape[2] == 3:
+        frame_bgr = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
+    else:
+        frame_bgr = frame
+        
+    # Ridimensiona la reference per adattarla al frame
+    ref_resized = cv2.resize(ref, (frame_bgr.shape[1], frame_bgr.shape[0]))
+    
+    # Converto in scala di grigi
+    frame_gray = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2GRAY)
+    ref_gray = cv2.cvtColor(ref_resized, cv2.COLOR_BGR2GRAY)
+    
+    # Template matching
+    res = cv2.matchTemplate(frame_gray, ref_gray, cv2.TM_CCOEFF_NORMED)
+    return res.max()
+
+class FMAPlayer:
+    def __init__(self):
+        # URL degli episodi (sostituisci con i tuoi URL)
+        self.base_url = "https://example.com/fullmetal/episode_{:02d}.mp4"
+        self.episode_number = 1
+        self.current_url = self.base_url.format(self.episode_number)
+        
+        # Percorsi delle immagini di riferimento (usa URL per Hugging Face)
+        self.reference_images = {
+            "intro_start": "https://raw.githubusercontent.com/your-repo/fma-player/main/intro_start.png",
+            "intro_end": "https://raw.githubusercontent.com/your-repo/fma-player/main/intro_end.png",
+            "outro_start": "https://raw.githubusercontent.com/your-repo/fma-player/main/outro_start.png"
+        }
+        
+        # Carica le immagini di riferimento
+        self.ref_images = {}
+        for key, url in self.reference_images.items():
+            self.ref_images[key] = None  # Inizialmente None, caricalo solo quando necessario
+    
+    def load_references(self):
+        # Carica le immagini di riferimento solo quando necessario
+        for key, url in self.reference_images.items():
+            if self.ref_images[key] is None:
+                self.ref_images[key] = load_image(url)
+    
+    def get_episode_url(self, episode_number):
+        return self.base_url.format(episode_number)
+    
+    def detect_intro_outro(self, frame):
+        self.load_references()
+        
+        # Controlla la similarità con l'immagine di inizio intro
+        sim_intro = compute_similarity(frame, self.ref_images["intro_start"])
+        if sim_intro >= MATCH_THRESHOLD:
+            return "intro", sim_intro
+        
+        # Controlla la similarità con l'immagine di inizio outro
+        sim_outro = compute_similarity(frame, self.ref_images["outro_start"])
+        if sim_outro >= MATCH_THRESHOLD:
+            return "outro", sim_outro
+        
+        return None, 0
+
+# Interfaccia Gradio
+def create_interface():
+    player = FMAPlayer()
+    
+    # CSS personalizzato per migliorare l'aspetto dell'interfaccia
+    css = """
+    body {
+        background-color: #1e1e2e;
+        color: #cdd6f4;
+        font-family: 'Segoe UI', Arial, sans-serif;
+    }
+    
+    .gradio-container {
+        max-width: 1200px !important;
+    }
+    
+    h1, h2, h3 {
+        color: #cba6f7;
+    }
+    
+    .fma-title {
+        font-size: 24px;
+        font-weight: bold;
+        text-align: center;
+        margin-bottom: 20px;
+        color: #cba6f7;
+    }
+    
+    button, select, .gradio-button {
+        background-color: #313244 !important;
+        color: #cdd6f4 !important;
+        border: none !important;
+        padding: 10px !important;
+        border-radius: 6px !important;
+        font-weight: bold !important;
+    }
+    
+    button:hover, select:hover, .gradio-button:hover {
+        background-color: #45475a !important;
+    }
+    
+    .video-container {
+        border-radius: 10px;
+        overflow: hidden;
+        background-color: #11111b;
+    }
+    
+    .special-button {
+        background-color: #7f849c !important;
+        color: #1e1e2e !important;
+    }
+    
+    .special-button:hover {
+        background-color: #9399b2 !important;
+    }
+    
+    .footer {
+        margin-top: 20px;
+        text-align: center;
+        font-size: 12px;
+        color: #9399b2;
+    }
+    """
+    
+    # Funzione per gestire il cambio di episodio
+    def change_episode(episode_num):
+        player.episode_number = episode_num
+        player.current_url = player.get_episode_url(episode_num)
+        return player.current_url, f"Fullmetal Alchemist Brotherhood - Episodio {episode_num:02d}"
+    
+    # Funzione per saltare l'intro
+    def skip_intro():
+        return "Skip intro attivato"
+    
+    # Funzione per saltare l'outro
+    def skip_outro():
+        next_ep = player.episode_number + 1
+        if next_ep <= 64:
+            url, title = change_episode(next_ep)
+            return f"Caricato episodio successivo: {next_ep}"
+        else:
+            return "Fine della serie raggiunta!"
+    
+    # Creazione dell'interfaccia Gradio
+    with gr.Blocks(css=css) as interface:
+        gr.HTML("<div class='fma-title'>Fullmetal Alchemist Brotherhood Player</div>")
+        
+        with gr.Row():
+            with gr.Column(scale=3):
+                # Titolo episodio
+                title = gr.Textbox(value=f"Fullmetal Alchemist Brotherhood - Episodio {player.episode_number:02d}", 
+                                   label="", 
+                                   interactive=False)
+                
+                # Container video
+                with gr.Box(elem_classes=["video-container"]):
+                    video = gr.Video(
+                        player.current_url,
+                        label="",
+                        width=800,
+                        height=450,
+                        interactive=True,
+                        autoplay=True
+                    )
+                
+                # Controlli
+                with gr.Row():
+                    rewind_btn = gr.Button("⏪ 10s", elem_classes=["control-button"])
+                    play_pause_btn = gr.Button("⏯️ Play/Pausa", elem_classes=["special-button", "control-button"])
+                    forward_btn = gr.Button("10s ⏩", elem_classes=["control-button"])
+                    skip_intro_btn = gr.Button("Salta Intro", elem_classes=["control-button"])
+                    skip_outro_btn = gr.Button("Salta Outro", elem_classes=["control-button"])
+                
+                # Volume
+                with gr.Row():
+                    volume_slider = gr.Slider(0, 100, 75, label="Volume")
+            
+            with gr.Column(scale=1):
+                # Selettore episodi
+                episode_selector = gr.Dropdown(
+                    [f"Episodio {i:02d}" for i in range(1, 64)],
+                    value=f"Episodio {player.episode_number:02d}",
+                    label="Seleziona Episodio"
+                )
+                
+                # Informazioni sull'episodio
+                episode_info = gr.Textbox(
+                    "Questo è l'episodio 1 di Fullmetal Alchemist Brotherhood.",
+                    label="Informazioni Episodio",
+                    interactive=False,
+                    lines=8
+                )
+                
+                # Status
+                status = gr.Textbox(
+                    "Player pronto",
+                    label="Status",
+                    interactive=False
+                )
+        
+        # Eventi
+        skip_intro_btn.click(skip_intro, inputs=[], outputs=[status])
+        skip_outro_btn.click(skip_outro, inputs=[], outputs=[status])
+        
+        # Conversione dell'indice episodio
+        def parse_episode(ep_str):
+            return int(ep_str.split()[1])
+        
+        # Evento cambio episodio
+        episode_selector.change(
+            lambda ep_str: change_episode(parse_episode(ep_str)),
+            inputs=[episode_selector],
+            outputs=[video, title]
+        )
+        
+        gr.HTML("<div class='footer'>Powered by Hugging Face Spaces | Fullmetal Alchemist Brotherhood Player</div>")
+    
+    return interface
+
+# Creazione e avvio dell'app
+def main():
+    interface = create_interface()
+    interface.launch()
+
+if __name__ == "__main__":
+    main()