Update genesis/providers.py

genesis/providers.py

@@ -1,79 +1,164 @@
 # genesis/providers.py
 import os
 import requests
-from genesis.api_clients.pubmed_api import search_pubmed
-from genesis.api_clients.chembl_api import search_chembl
-from genesis.api_clients.bioportal_api import search_bioportal
-from genesis.api_clients.umls_api import search_umls
-from genesis.api_clients.ncbi_api import search_ncbi
+import base64
+from typing import Optional
 
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+# Load secrets from environment
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
-CLAUDE_API_KEY = os.getenv("CLAUDE_API_KEY")
-HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
-
-# ---------- LITERATURE ----------
-def run_pubmed_literature(query: str, max_results: int = 10):
-    """Fetch literature from PubMed."""
-    return search_pubmed(query, max_results=max_results)
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+HF_API_KEY = os.getenv("HF_API_KEY")
+ELEVEN_LABS_API_KEY = os.getenv("ELEVEN_LABS_API_KEY")
 
-# ---------- DEEPSEEK ----------
-def run_deepseek_summary(prompt: str):
-    """Summarize content using DeepSeek API."""
+# ------------------------------
+# 1 — DEEPSEEK SUMMARIZER
+# ------------------------------
+def run_deepseek_summary(prompt: str) -> Optional[str]:
+    if not DEEPSEEK_API_KEY:
+        return None
     try:
         url = "https://api.deepseek.com/v1/chat/completions"
         headers = {"Authorization": f"Bearer {DEEPSEEK_API_KEY}"}
-        data = {"model": "deepseek-chat", "messages": [{"role": "user", "content": prompt}]}
-        r = requests.post(url, headers=headers, json=data)
-        r.raise_for_status()
-        return r.json()["choices"][0]["message"]["content"]
+        payload = {
+            "model": "deepseek-chat",
+            "messages": [{"role": "user", "content": prompt}],
+            "temperature": 0.3
+        }
+        resp = requests.post(url, headers=headers, json=payload, timeout=60)
+        resp.raise_for_status()
+        return resp.json()["choices"][0]["message"]["content"]
     except Exception as e:
         print(f"[DeepSeek] Failed: {e}")
         return None
 
-# ---------- GEMINI ----------
-def run_gemini_polish(text: str):
-    """Polish text using Gemini."""
+
+# ------------------------------
+# 2 — GEMINI SUMMARIZER
+# ------------------------------
+def run_gemini_summary(prompt: str) -> Optional[str]:
+    if not GEMINI_API_KEY:
+        return None
     try:
         url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}"
         headers = {"Content-Type": "application/json"}
-        payload = {"contents": [{"parts": [{"text": text}]}]}
-        r = requests.post(url, headers=headers, json=payload)
-        r.raise_for_status()
-        return r.json()["candidates"][0]["content"]["parts"][0]["text"]
+        payload = {"contents": [{"parts": [{"text": prompt}]}]}
+        resp = requests.post(url, headers=headers, json=payload, timeout=60)
+        resp.raise_for_status()
+        return resp.json()["candidates"][0]["content"]["parts"][0]["text"]
     except Exception as e:
         print(f"[Gemini] Failed: {e}")
-        return text
+        return None
+
+
+# ------------------------------
+# 3 — OPENAI SUMMARIZER
+# ------------------------------
+def run_openai_summary(prompt: str) -> Optional[str]:
+    if not OPENAI_API_KEY:
+        return None
+    try:
+        url = "https://api.openai.com/v1/chat/completions"
+        headers = {"Authorization": f"Bearer {OPENAI_API_KEY}"}
+        payload = {
+            "model": "gpt-4o-mini",
+            "messages": [{"role": "user", "content": prompt}],
+            "temperature": 0.3
+        }
+        resp = requests.post(url, headers=headers, json=payload, timeout=60)
+        resp.raise_for_status()
+        return resp.json()["choices"][0]["message"]["content"]
+    except Exception as e:
+        print(f"[OpenAI] Failed: {e}")
+        return None
+
+
+# ------------------------------
+# 4 — GEMINI IMAGE GENERATOR
+# ------------------------------
+def run_gemini_image(prompt: str) -> Optional[str]:
+    if not GEMINI_API_KEY:
+        return None
+    try:
+        # Gemini image generation (beta)
+        url = f"https://generativelanguage.googleapis.com/v1beta/models/imagegeneration:generateImage?key={GEMINI_API_KEY}"
+        payload = {"prompt": prompt, "size": "1024x1024"}
+        headers = {"Content-Type": "application/json"}
+        resp = requests.post(url, headers=headers, json=payload, timeout=60)
+        resp.raise_for_status()
+        data = resp.json()
+        if "data" in data and len(data["data"]) > 0:
+            return data["data"][0]["url"]
+        return None
+    except Exception as e:
+        print(f"[Gemini Image] Failed: {e}")
+        return None
+
 
-# ---------- GEMINI IMAGE (Fallback to OpenAI, then HuggingFace) ----------
-def run_image_generation(prompt: str):
-    """Generate image using Gemini > OpenAI > Hugging Face fallback."""
-    # Try Gemini first
+# ------------------------------
+# 5 — OPENAI IMAGE GENERATOR
+# ------------------------------
+def run_openai_image(prompt: str) -> Optional[str]:
+    if not OPENAI_API_KEY:
+        return None
     try:
-        url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateImage?key={GEMINI_API_KEY}"
-        r = requests.post(url, json={"prompt": prompt})
-        r.raise_for_status()
-        return r.json().get("image_url")
-    except:
-        pass
+        url = "https://api.openai.com/v1/images/generations"
+        headers = {"Authorization": f"Bearer {OPENAI_API_KEY}"}
+        payload = {"model": "gpt-image-1", "prompt": prompt, "size": "1024x1024"}
+        resp = requests.post(url, headers=headers, json=payload, timeout=60)
+        resp.raise_for_status()
+        return resp.json()["data"][0]["url"]
+    except Exception as e:
+        print(f"[OpenAI Image] Failed: {e}")
+        return None
+
 
-    # Try OpenAI
+# ------------------------------
+# 6 — HF STABLE DIFFUSION IMAGE
+# ------------------------------
+def run_hf_image(prompt: str) -> Optional[str]:
+    if not HF_API_KEY:
+        return None
     try:
-        import openai
-        openai.api_key = OPENAI_API_KEY
-        result = openai.Image.create(model="gpt-image-1", prompt=prompt, size="1024x1024")
-        return result["data"][0]["url"]
-    except:
-        pass
+        url = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
+        headers = {"Authorization": f"Bearer {HF_API_KEY}"}
+        payload = {"inputs": prompt}
+        resp = requests.post(url, headers=headers, json=payload, timeout=120)
+        resp.raise_for_status()
+        if resp.headers.get("content-type") == "image/png":
+            filename = f"generated_{hash(prompt)}.png"
+            with open(filename, "wb") as f:
+                f.write(resp.content)
+            return filename
+        return None
+    except Exception as e:
+        print(f"[HF Image] Failed: {e}")
+        return None
 
-    # Try Hugging Face
+
+# ------------------------------
+# 7 — ELEVENLABS TEXT-TO-SPEECH
+# ------------------------------
+def narrate_text_elevenlabs(text: str) -> Optional[str]:
+    if not ELEVEN_LABS_API_KEY:
+        return None
     try:
-        hf_url = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
-        headers = {"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"}
-        r = requests.post(hf_url, headers=headers, json={"inputs": prompt})
-        r.raise_for_status()
-        return r.json()
+        url = "https://api.elevenlabs.io/v1/text-to-speech"
+        headers = {
+            "xi-api-key": ELEVEN_LABS_API_KEY,
+            "Content-Type": "application/json"
+        }
+        payload = {
+            "text": text,
+            "voice": "Bella",  # You can make this configurable
+            "model_id": "eleven_monolingual_v1"
+        }
+        resp = requests.post(url, headers=headers, json=payload, timeout=60)
+        resp.raise_for_status()
+        filename = f"narration_{hash(text)}.mp3"
+        with open(filename, "wb") as f:
+            f.write(resp.content)
+        return filename
     except Exception as e:
-        print(f"[Image Generation] Failed: {e}")
+        print(f"[ElevenLabs] Failed: {e}")
         return None