Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-import google.generativeai as genai
+from openai import OpenAI
 import json
 import time
 import io
@@ -169,19 +169,30 @@ 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("ALL_GEMINI_API_KEYS")
+        keys_str = os.environ.get("HF_TOKENS")
         keys_list = [k.strip() for k in keys_str.split(',') if k.strip()] if keys_str else []
-        if not keys_list: raise ValueError("No AI Keys")
+        if not keys_list: raise ValueError("No HF Tokens")
 
         MAX_ATTEMPTS = 50
         for attempt in range(MAX_ATTEMPTS):
             key = random.choice(keys_list)
             try:
-                genai.configure(api_key=key)
-                model = genai.GenerativeModel('gemini-2.5-flash')
-                res = model.generate_content(system_prompt, safety_settings=safety_settings)
+                client = OpenAI(
+                    base_url="https://router.huggingface.co/v1",
+                    api_key=key,
+                )
                 
-                raw_text = res.text
+                completion = client.chat.completions.create(
+                    model="google/gemma-4-31B-it:novita",
+                    messages=[
+                        {
+                            "role": "user",
+                            "content": system_prompt
+                        }
+                    ]
+                )
+                
+                raw_text = completion.choices[0].message.content
                 json_string = None
                 match = re.search(r"```json\s*(\{.*?\})\s*```", raw_text, re.DOTALL)
                 if match: json_string = match.group(1)
@@ -419,4 +430,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)