updated

app.py

@@ -3,18 +3,19 @@ import subprocess
 import gradio as gr
 import cv2
 import numpy as np
-from PIL import Image
 import tempfile
-import time
 import google.generativeai as genai
 
 # ---------------- GEMINI CONFIG ----------------
 API_KEY = os.environ.get("GEMINI_API_KEY")
-genai_client = genai.Client(api_key=API_KEY)
+if not API_KEY:
+    raise ValueError("❌ GEMINI_API_KEY not set in Hugging Face secrets!")
+
+genai.configure(api_key=API_KEY)
+model = genai.GenerativeModel("gemini-2.5-flash")
 
 # ---------------- EFFECT FUNCTIONS ----------------
 def create_fisheye(frame, strength=0.5):
-    # simplified fisheye effect for demo
     h, w = frame.shape[:2]
     y, x = np.indices((h, w), dtype=np.float32)
     cx, cy = w/2, h/2
@@ -114,12 +115,12 @@ def process_video(video_path, style="equirect", strength=0.7, eye_offset=10, vol
 # ---------------- GEMINI AUTO STYLE ----------------
 def suggest_style(video_description):
     try:
-        response = genai_client.models.generate_content(
-            model="gemini-2.5-flash",
-            contents=[{"text": f"Suggest best VR conversion style (fisheye, spherical, equirectangular) for this video: {video_description}"}]
+        response = model.generate_content(
+            f"Suggest best VR conversion style (fisheye, spherical, equirectangular) for this video: {video_description}"
         )
         return response.text.strip().lower()
-    except:
+    except Exception as e:
+        print("Gemini error:", e)
         return "equirect"
 
 # ---------------- GRADIO UI ----------------
@@ -131,7 +132,7 @@ with gr.Blocks(title="2D → VR Video Converter") as demo:
             with gr.Row():
                 with gr.Column():
                     video_in = gr.Video(label="Upload Video")
-                    description = gr.Textbox(label="Video Description (optional)", placeholder="Describe your video for auto style...")
+                    description = gr.Textbox(label="Video Description (optional)", placeholder="Describe your video...")
                     auto_style_btn = gr.Button("Auto Suggest Style")
                     style_choice = gr.Radio(["equirect", "fisheye", "spherical"], label="VR Style", value="equirect")
                     strength_slider = gr.Slider(0.3, 1.0, value=0.7, step=0.1, label="Effect Strength")
@@ -141,7 +142,7 @@ with gr.Blocks(title="2D → VR Video Converter") as demo:
                 with gr.Column():
                     output_video = gr.Video(label="VR Side-by-Side Output")
                     status_box = gr.Textbox(label="Status")
-        
+
     def run_auto_style(desc):
         if not desc.strip():
             return "equirect"
@@ -157,3 +158,4 @@ with gr.Blocks(title="2D → VR Video Converter") as demo:
                       outputs=[output_video, status_box])
 
 demo.launch()
+