Update genesis/providers.py

genesis/providers.py

@@ -22,15 +22,14 @@ def run_deepseek_summary(prompt: str) -> str:
         }
         r = requests.post(url, headers=headers, json=payload, timeout=60)
         r.raise_for_status()
-        data = r.json()
-        return data["choices"][0]["message"]["content"]
+        return r.json()["choices"][0]["message"]["content"]
     except Exception as e:
         print(f"[DeepSeek] Failed: {e}")
-        return prompt  # fallback
+        return prompt
 
 # -------- Gemini Polish --------
 def run_gemini_polish(text: str) -> str:
-    """Polish the summary using Gemini for clarity and presentation."""
+    """Polish the summary using Gemini for clarity."""
     if not GEMINI_API_KEY:
         return text
     try:
@@ -38,15 +37,14 @@ def run_gemini_polish(text: str) -> str:
         payload = {"contents": [{"parts": [{"text": f"Polish and clarify this research report:\n\n{text}"}]}]}
         r = requests.post(url, json=payload, timeout=30)
         r.raise_for_status()
-        data = r.json()
-        return data["candidates"][0]["content"]["parts"][0]["text"]
+        return r.json()["candidates"][0]["content"]["parts"][0]["text"]
     except Exception as e:
         print(f"[Gemini] Failed: {e}")
         return text
 
-# -------- OpenAI + HuggingFace Image Generation --------
-def run_gemini_image(query: str) -> str:
-    """Generate a research diagram — OpenAI first, Hugging Face fallback."""
+# -------- Image Generation (OpenAI → HF Fallback) --------
+def run_openai_image(query: str) -> str:
+    """Generate a research diagram using OpenAI image API, fallback to HF Stable Diffusion."""
     # Try OpenAI first
     if OPENAI_API_KEY:
         try:
@@ -55,12 +53,11 @@ def run_gemini_image(query: str) -> str:
             payload = {"model": "gpt-image-1", "prompt": f"Scientific diagram about: {query}", "size": "1024x1024"}
             r = requests.post(url, headers=headers, json=payload, timeout=60)
             r.raise_for_status()
-            data = r.json()
-            return data["data"][0]["url"]
+            return r.json()["data"][0]["url"]
         except Exception as e:
             print(f"[OpenAI Image] Failed: {e}")
 
-    # Hugging Face Fallback
+    # Hugging Face fallback
     if HF_TOKEN:
         try:
             print("[Image] Falling back to Hugging Face Stable Diffusion...")
@@ -69,7 +66,6 @@ def run_gemini_image(query: str) -> str:
             payload = {"inputs": f"highly detailed scientific diagram about: {query}"}
             r = requests.post(url, headers=headers, json=payload, timeout=60)
             r.raise_for_status()
-            # Save to local file
             img_path = f"/tmp/{query.replace(' ', '_')}.png"
             with open(img_path, "wb") as f:
                 f.write(r.content)