Update app.py

app.py

@@ -1,105 +1,92 @@
 import gradio as gr
-from moviepy.editor import ImageClip, concatenate_videoclips, TextClip, CompositeVideoClip
 from PIL import Image
 import requests
 from io import BytesIO
+from moviepy.editor import ImageClip, concatenate_videoclips
+import random
+import os
 
-# Datenbank: Horror-Bilder (Pixabay, frei verwendbar)
+# Sichere, getestete Horror-Bilder (GitHub gehostet)
 IMAGE_DB = {
     "haus": [
-        "https://cdn.pixabay.com/photo/2016/11/29/05/15/haunted-house-1867832_1280.jpg",
-        "https://cdn.pixabay.com/photo/2016/11/18/22/15/old-house-1836494_1280.jpg"
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/haunted_house_1.jpg",
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/haunted_house_2.jpg"
     ],
     "geist": [
-        "https://cdn.pixabay.com/photo/2017/09/02/13/17/ghost-2709306_1280.jpg",
-        "https://cdn.pixabay.com/photo/2018/10/18/22/48/halloween-3751882_1280.jpg"
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/ghost_1.jpg",
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/ghost_2.jpg"
     ],
     "spuk": [
-        "https://cdn.pixabay.com/photo/2016/11/21/12/04/spooky-1846021_1280.jpg",
-        "https://cdn.pixabay.com/photo/2017/11/20/00/35/abandoned-2963828_1280.jpg"
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/spooky_forest.jpg",
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/abandoned_building.jpg"
     ],
     "katze": [
-        "https://cdn.pixabay.com/photo/2016/11/29/03/53/cat-1867274_1280.jpg"
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/black_cat.jpg"
     ],
     "ritual": [
-        "https://cdn.pixabay.com/photo/2018/10/31/21/59/halloween-3783701_1280.jpg"
+        "https://raw.githubusercontent.com/chrispawluk/horror-img/main/ritual_circle.jpg"
     ],
 }
 
-# Bildauswahl nach Titel
-def select_images(title):
-    title_lower = title.lower()
-    selected_urls = []
-    for keyword, urls in IMAGE_DB.items():
-        if keyword in title_lower:
-            selected_urls.extend(urls)
-    if not selected_urls:
-        for urls in IMAGE_DB.values():
-            selected_urls.extend(urls)
-    return selected_urls[:6]
+def find_images_for_title(title):
+    # Suche passende Bilder basierend auf Keywords im Titel
+    keywords = IMAGE_DB.keys()
+    selected_images = []
+    for keyword in keywords:
+        if keyword in title.lower():
+            selected_images.extend(IMAGE_DB[keyword])
+    # Falls keine Keywords gefunden, zufällige Bilder
+    if not selected_images:
+        # alle Bilder aus allen Kategorien sammeln
+        all_images = sum(IMAGE_DB.values(), [])
+        selected_images = random.sample(all_images, min(5, len(all_images)))
+    else:
+        selected_images = selected_images[:5]  # max 5 Bilder
+    return selected_images
 
-# Sicheres Bild-Download mit Prüfung
 def download_image(url):
     try:
-        response = requests.get(url, timeout=10)
-        if response.status_code == 200 and 'image' in response.headers.get("Content-Type", ""):
-            img = Image.open(BytesIO(response.content)).convert("RGB")
-            return img
-        else:
-            print(f"WARNUNG: Ungültiges Bild: {url}")
-            return None
+        response = requests.get(url)
+        img = Image.open(BytesIO(response.content)).convert("RGB")
+        return img
     except Exception as e:
-        print(f"FEHLER beim Bild-Download: {url} — {str(e)}")
+        print(f"WARNUNG: Ungültiges Bild: {url}")
         return None
 
-# Video erstellen aus Titel & Bildern
 def create_video(title):
-    image_urls = select_images(title)
-    clips = []
-    duration_per_image = 10  # Sekunden pro Bild
-
-    for i, url in enumerate(image_urls):
+    image_urls = find_images_for_title(title)
+    images = []
+    for url in image_urls:
         img = download_image(url)
-        if img is None:
-            continue
-
-        img_clip = ImageClip(img).set_duration(duration_per_image)
-
-        # Text-Overlay
-        txt = TextClip(
-            f"{title}\nSzene {i+1}",
-            fontsize=40,
-            color='white',
-            font='Amiri-Bold',
-            method='caption',
-            size=img.size
-        )
-        txt = txt.set_duration(duration_per_image).set_pos('bottom').margin(bottom=30, opacity=0)
-
-        video_clip = CompositeVideoClip([img_clip, txt])
-        clips.append(video_clip)
-
-    if not clips:
+        if img:
+            images.append(img)
+    if not images:
         raise RuntimeError("Keine gültigen Bilder gefunden. Bitte gib einen anderen Titel ein.")
 
-    final_clip = concatenate_videoclips(clips, method="compose")
-    out_path = "output.mp4"
-    final_clip.write_videofile(out_path, fps=24)
-    return out_path
+    clips = []
+    duration_per_image = 12  # 5 Bilder x 12s = 60s Video
+    for img in images:
+        clip = ImageClip(img).set_duration(duration_per_image)
+        clips.append(clip)
+
+    video = concatenate_videoclips(clips, method="compose")
+    output_path = "output.mp4"
+    video.write_videofile(output_path, fps=24, codec="libx264", audio=False)
+    return output_path
 
-# Gradio-Funktion
 def run(title):
-    video_path = create_video(title)
-    return video_path
+    try:
+        video_path = create_video(title)
+        return video_path
+    except Exception as e:
+        return str(e)
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Creepy Horror Movie Shorts Generator")
+    title_input = gr.Textbox(label="Gib den Titel oder die Idee des Videos ein", placeholder="z.B. Spuk im alten Haus")
+    submit_btn = gr.Button("Submit")
+    video_output = gr.Video(label="Dein Horror-Video")
 
-# Gradio-UI
-iface = gr.Interface(
-    fn=run,
-    inputs=gr.Textbox(label="🎬 Gib den Titel deines Horror-Videos ein"),
-    outputs=gr.Video(label="👻 Generiertes Horror-Video"),
-    title="Creepy Horror Shorts Generator (kostenfrei)",
-    description="Gib einen gruseligen Titel ein – und erhalte automatisch ein Horror-Video aus freien Bildern.",
-)
+    submit_btn.click(run, inputs=title_input, outputs=video_output)
 
-if __name__ == "__main__":
-    iface.launch(share=True)
+demo.launch(share=True)