Update app.py

app.py

@@ -1,23 +1,147 @@
-import streamlit as st
+import gradio as gr
 import requests
 import os
 from moviepy.editor import VideoFileClip, concatenate_videoclips
 import tempfile
-import shutil
 
-# ... (همون تابع make_video که قبلاً دادم، فقط ورودی/خروجی رو با st عوض کن)
+# ---------------- تنظیمات ----------------
+PIXABAY_API_KEY = os.getenv("PIXABAY_API_KEY")  # در Settings → Secrets در Hugging Face بگذار
 
-st.title("Pixabay → ویدیو ۳ ثانیه‌ای")
+if not PIXABAY_API_KEY:
+    raise ValueError("لطفاً PIXABAY_API_KEY را در Secrets فضای خود قرار دهید!")
 
-keywords = st.text_input("کلمات کلیدی (با فاصله یا کاما)", "باران جنگل شب")
+VIDEO_SEARCH_URL = "https://pixabay.com/api/videos/"
+CLIP_DURATION = 3.0  # ثانیه
 
-if st.button("ساخت ویدیو"):
-    if keywords:
-        video_path, status = make_video(keywords)
-        if video_path:
-            st.video(video_path)
-            st.success(status)
-        else:
-            st.error(status)
-    else:
-        st.warning("کلمه کلیدی وارد کن")
+def download_video(url, temp_dir):
+    """دانلود ویدیو و برگرداندن مسیر فایل"""
+    response = requests.get(url, stream=True)
+    if response.status_code != 200:
+        return None
+    
+    temp_path = os.path.join(temp_dir, "clip.mp4")
+    with open(temp_path, "wb") as f:
+        for chunk in response.iter_content(chunk_size=8192):
+            f.write(chunk)
+    return temp_path
+
+def search_pixabay_videos(query, per_page=10):
+    params = {
+        "key": PIXABAY_API_KEY,
+        "q": query,
+        "video_type": "all",     # all / film / animation
+        "per_page": per_page,
+        "min_duration": 4,       # حداقل ۴ ثانیه که بتوان ۳ ثانیه برداشت
+        "safesearch": "true"
+    }
+    
+    try:
+        r = requests.get(VIDEO_SEARCH_URL, params=params, timeout=15)
+        r.raise_for_status()
+        data = r.json()
+        
+        videos = []
+        for hit in data.get("hits", []):
+            # بهترین کیفیت ممکن (large یا medium)
+            vid = hit.get("videos", {})
+            if "large" in vid and vid["large"]["url"]:
+                videos.append(vid["large"]["url"])
+            elif "medium" in vid and vid["medium"]["url"]:
+                videos.append(vid["medium"]["url"])
+        
+        return videos[:8]  # حداکثر ۸ تا ویدیو بگیریم که پردازش سنگین نشود
+    except Exception as e:
+        print(e)
+        return []
+
+def create_montage(keywords):
+    if not keywords.strip():
+        return None, "لطفاً کلمات کلیدی وارد کنید"
+    
+    query = keywords.strip().replace(",", " ").replace("  ", " ")
+    
+    with tempfile.TemporaryDirectory() as tmpdirname:
+        video_urls = search_pixabay_videos(query)
+        
+        if not video_urls:
+            return None, f"هیچ ویدیویی برای '{query}' پیدا نشد."
+        
+        clips = []
+        count = 0
+        
+        for url in video_urls:
+            try:
+                path = download_video(url, tmpdirname)
+                if not path:
+                    continue
+                
+                clip = VideoFileClip(path)
+                
+                # اگر ویدیو کوتاه‌تر از ۳ ثانیه بود → skip
+                if clip.duration < CLIP_DURATION:
+                    clip.close()
+                    continue
+                
+                # برش ۳ ثانیه (مثلاً از وسط یا اول)
+                start = max(0, (clip.duration - CLIP_DURATION) / 2)  # وسط ویدیو
+                subclip = clip.subclip(start, start + CLIP_DURATION)
+                
+                clips.append(subclip)
+                count += 1
+                
+                clip.close()
+                if count >= 5:  # مثلاً حداکثر ۵ قطعه → حدود ۱۵ ثانیه
+                    break
+                    
+            except Exception as e:
+                print(f"خطا در پردازش ویدیو: {e}")
+                continue
+        
+        if not clips:
+            return None, "متأسفانه نتوانستیم ویدیوهای مناسب دانلود/برش دهیم."
+        
+        try:
+            final_clip = concatenate_videoclips(clips, method="compose")
+            
+            # خروجی موقت
+            output_path = os.path.join(tmpdirname, "montage.mp4")
+            final_clip.write_videofile(output_path,
+                                     codec="libx264",
+                                     audio_codec="aac",
+                                     fps=24,
+                                     preset="medium",
+                                     threads=4,
+                                     logger=None)
+            
+            return output_path, f"و��دیو با موفقیت ساخته شد ({len(clips)} قطعه × ۳ ثانیه)"
+        
+        except Exception as e:
+            return None, f"خطا در ساخت ویدیو: {str(e)}"
+        
+        finally:
+            # تمیز کردن حافظه
+            for c in clips:
+                c.close()
+            if 'final_clip' in locals():
+                final_clip.close()
+
+# ---------------- رابط کاربری Gradio ----------------
+with gr.Blocks(title="Pixabay Video Montage") as demo:
+    gr.Markdown("# ساخت مونتاژ ویدیو از Pixabay\nکلمات کلیدی را وارد کنید (مثال: nature, sunset, drone)")
+    
+    textbox = gr.Textbox(label="کلمات کلیدی (با کاما یا فاصله جدا کنید)", placeholder="ocean waves, beach, sunset")
+    
+    btn = gr.Button("ساخت ویدیو (۳ ثانیه از هر کلیپ)")
+    
+    output_video = gr.Video(label="ویدیوی نهایی")
+    status = gr.Textbox(label="وضعیت")
+    
+    btn.click(
+        fn=create_montage,
+        inputs=textbox,
+        outputs=[output_video, status],
+        api_name="make_montage"
+    )
+
+if __name__ == "__main__":
+    demo.launch()