Update genesis/providers.py

genesis/providers.py

@@ -1,65 +1,113 @@
-# genesis/providers.py
-"""
-API Providers for GENESIS-AI
-Handles calls to DeepSeek, Gemini, OpenAI, and PubMed.
-"""
-
 import os
 import requests
+from typing import Optional
 
-# ENV
+# Load API keys from environment
 DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-NCBI_API_KEY = os.getenv("NCBI_API_KEY")
-NCBI_EMAIL = os.getenv("NCBI_EMAIL")
+HF_API_KEY = os.getenv("HF_API_KEY")
 
-def run_deepseek_summary(prompt):
-    """Summarize using DeepSeek API."""
+# === DeepSeek Summarization ===
+def run_deepseek_summary(prompt: str) -> str:
+    if not DEEPSEEK_API_KEY:
+        return "[DeepSeek] API key missing."
     try:
-        r = requests.post(
+        resp = requests.post(
             "https://api.deepseek.com/v1/chat/completions",
-            headers={"Authorization": f"Bearer {DEEPSEEK_API_KEY}"},
-            json={"model": "deepseek-chat", "messages": [{"role": "user", "content": prompt}]}
+            headers={
+                "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
+                "Content-Type": "application/json"
+            },
+            json={
+                "model": "deepseek-chat",
+                "messages": [{"role": "user", "content": prompt}],
+                "temperature": 0.3
+            }
         )
-        r.raise_for_status()
-        return r.json()["choices"][0]["message"]["content"]
+        resp.raise_for_status()
+        return resp.json()["choices"][0]["message"]["content"].strip()
     except Exception as e:
-        return f"[DeepSeek Error] {e}"
+        return f"[DeepSeek] Failed: {str(e)}"
+
 
-def run_gemini_polish(text):
-    """Polish text with Gemini."""
+# === Gemini Polishing ===
+def run_gemini_polish(text: str) -> str:
+    if not GEMINI_API_KEY:
+        return text
     try:
-        r = requests.post(
+        resp = requests.post(
             f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}",
-            json={"contents": [{"parts": [{"text": text}]}]}
+            headers={"Content-Type": "application/json"},
+            json={
+                "contents": [{"parts": [{"text": f"Polish and refine the following scientific text:\n\n{text}"}]}]
+            }
+        )
+        resp.raise_for_status()
+        candidates = resp.json().get("candidates", [])
+        if candidates and "content" in candidates[0]:
+            return candidates[0]["content"]["parts"][0]["text"].strip()
+        return text
+    except Exception as e:
+        return f"[Gemini] Failed: {str(e)}"
+
+
+# === Gemini Image Generation ===
+def run_gemini_image(prompt: str) -> Optional[str]:
+    if not GEMINI_API_KEY:
+        return None
+    try:
+        resp = requests.post(
+            f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateImage?key={GEMINI_API_KEY}",
+            headers={"Content-Type": "application/json"},
+            json={"prompt": prompt}
         )
-        r.raise_for_status()
-        return r.json()["candidates"][0]["content"]["parts"][0]["text"]
+        resp.raise_for_status()
+        data = resp.json()
+        if "imageUrl" in data:
+            return data["imageUrl"]
+        return None
     except Exception as e:
-        return f"[Gemini Error] {e}"
+        print(f"[Gemini Image] Failed: {e}")
+        return None
+
 
-def run_openai_image(prompt):
-    """Generate image with OpenAI DALL·E or fallback."""
+# === OpenAI Image Generation ===
+def run_openai_image(prompt: str) -> Optional[str]:
+    if not OPENAI_API_KEY:
+        return None
     try:
-        r = requests.post(
+        resp = requests.post(
             "https://api.openai.com/v1/images/generations",
-            headers={"Authorization": f"Bearer {OPENAI_API_KEY}"},
+            headers={
+                "Authorization": f"Bearer {OPENAI_API_KEY}",
+                "Content-Type": "application/json"
+            },
             json={"model": "gpt-image-1", "prompt": prompt, "size": "1024x1024"}
         )
-        r.raise_for_status()
-        return r.json()["data"][0]["url"]
+        resp.raise_for_status()
+        return resp.json()["data"][0]["url"]
     except Exception as e:
-        return f"[OpenAI Image Error] {e}"
+        print(f"[OpenAI Image] Failed: {e}")
+        return None
 
-def pubmed_fallback_search(query, retmax=5):
-    """Simple PubMed search for citations."""
+
+# === Hugging Face Fallback Image Generation ===
+def run_hf_image(prompt: str) -> Optional[str]:
+    if not HF_API_KEY:
+        return None
     try:
-        url = f"https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi"
-        params = {"db": "pubmed", "term": query, "retmode": "json", "retmax": retmax, "api_key": NCBI_API_KEY}
-        r = requests.get(url, params=params)
-        r.raise_for_status()
-        ids = r.json()["esearchresult"]["idlist"]
-        return [{"pmid": pid, "url": f"https://pubmed.ncbi.nlm.nih.gov/{pid}/"} for pid in ids]
+        resp = requests.post(
+            "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2",
+            headers={"Authorization": f"Bearer {HF_API_KEY}"},
+            json={"inputs": prompt}
+        )
+        resp.raise_for_status()
+        # HF returns binary image; you need to save or serve it yourself
+        output_file = "hf_image.png"
+        with open(output_file, "wb") as f:
+            f.write(resp.content)
+        return output_file
     except Exception as e:
-        return [{"error": str(e)}]
+        print(f"[HF Image] Failed: {e}")
+        return None