Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from openai import OpenAI
+import google.generativeai as genai
 import json
 import time
 import io
@@ -169,36 +169,19 @@ def call_worker(index, chunk_payload):
 # --- AI PODCAST SCRIPT LOGIC ---
 def generate_podcast_in_background(task_id, system_prompt, safety_settings):
     try:
-        keys_str = os.environ.get("HF_TOKENS")
+        keys_str = os.environ.get("ALL_GEMINI_API_KEYS")
         keys_list = [k.strip() for k in keys_str.split(',') if k.strip()] if keys_str else []
-        
-        if not keys_list:
-            single_key = os.environ.get("HF_TOKEN")
-            if single_key:
-                keys_list = [single_key.strip()]
-            else:
-                raise ValueError("No HF_TOKENS or HF_TOKEN found in environment variables")
+        if not keys_list: raise ValueError("No AI Keys")
 
         MAX_ATTEMPTS = 50
         for attempt in range(MAX_ATTEMPTS):
             key = random.choice(keys_list)
             try:
-                client = OpenAI(
-                    base_url="https://router.huggingface.co/v1",
-                    api_key=key,
-                )
-                
-                completion = client.chat.completions.create(
-                    model="google/gemma-4-31B-it:novita",
-                    messages=[
-                        {
-                            "role": "user",
-                            "content": system_prompt
-                        }
-                    ],
-                )
+                genai.configure(api_key=key)
+                model = genai.GenerativeModel('gemini-2.5-flash')
+                res = model.generate_content(system_prompt, safety_settings=safety_settings)
                 
-                raw_text = completion.choices[0].message.content
+                raw_text = res.text
                 json_string = None
                 match = re.search(r"```json\s*(\{.*?\})\s*```", raw_text, re.DOTALL)
                 if match: json_string = match.group(1)
@@ -207,7 +190,7 @@ def generate_podcast_in_background(task_id, system_prompt, safety_settings):
                     e_idx = raw_text.rfind('}')
                     if s_idx != -1 and e_idx != -1: json_string = raw_text[s_idx:e_idx+1]
                 
-                if not json_string: raise ValueError("No JSON found in response")
+                if not json_string: raise ValueError("No JSON found")
                 
                 data = json.loads(json_string)
                 if "script" in data:
@@ -436,4 +419,4 @@ atexit.register(persist_data_to_hub)
 
 if __name__ == '__main__':
     port = int(os.environ.get('PORT', 7860))
-    app.run(host='0.0.0.0', port=port)
+    app.run(host='0.0.0.0', port=port)