Update genesis/providers.py

genesis/providers.py

@@ -1,114 +1,65 @@
+# genesis/providers.py
+"""
+API Providers for GENESIS-AI
+Handles calls to DeepSeek, Gemini, OpenAI, and PubMed.
+"""
+
 import os
 import requests
-from typing import List, Dict
 
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-HF_TOKEN = os.getenv("HF_TOKEN")
+# ENV
+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")
 
-# -------- DeepSeek Summary --------
-def run_deepseek_summary(prompt: str) -> str:
-    """Run a dense scientific summary using DeepSeek API."""
-    try:
-        url = "https://api.deepseek.com/v1/chat/completions"
-        headers = {"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"}
-        payload = {
-            "model": "deepseek-science",
-            "messages": [{"role": "user", "content": prompt}],
-            "temperature": 0.2
-        }
-        r = requests.post(url, headers=headers, json=payload, timeout=60)
-        r.raise_for_status()
-        data = r.json()
-        return data["choices"][0]["message"]["content"]
-    except Exception as e:
-        print(f"[DeepSeek] Failed: {e}")
-        return prompt
-
-# -------- Gemini Polish --------
-def run_gemini_polish(text: str) -> str:
-    """Polish the summary using Gemini."""
-    if not GEMINI_API_KEY:
-        return text
+def run_deepseek_summary(prompt):
+    """Summarize using DeepSeek API."""
     try:
-        url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}"
-        payload = {"contents": [{"parts": [{"text": f"Polish and clarify this research report:\n\n{text}"}]}]}
-        r = requests.post(url, json=payload, timeout=30)
+        r = 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}]}
+        )
         r.raise_for_status()
-        data = r.json()
-        return data["candidates"][0]["content"]["parts"][0]["text"]
+        return r.json()["choices"][0]["message"]["content"]
     except Exception as e:
-        print(f"[Gemini] Failed: {e}")
-        return text
+        return f"[DeepSeek Error] {e}"
 
-# -------- OpenAI Image --------
-def run_openai_image(prompt: str) -> str:
-    """Generate image using OpenAI API."""
-    if not OPENAI_API_KEY:
-        return None
+def run_gemini_polish(text):
+    """Polish text with Gemini."""
     try:
-        url = "https://api.openai.com/v1/images/generations"
-        headers = {"Authorization": f"Bearer {OPENAI_API_KEY}"}
-        payload = {
-            "model": "gpt-image-1",
-            "prompt": f"Highly detailed scientific diagram: {prompt}",
-            "size": "1024x1024"
-        }
-        r = requests.post(url, headers=headers, json=payload, timeout=60)
+        r = requests.post(
+            f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}",
+            json={"contents": [{"parts": [{"text": text}]}]}
+        )
         r.raise_for_status()
-        data = r.json()
-        return data["data"][0]["url"]
+        return r.json()["candidates"][0]["content"]["parts"][0]["text"]
     except Exception as e:
-        print(f"[OpenAI Image] Failed: {e}")
-        return None
+        return f"[Gemini Error] {e}"
 
-# -------- Hugging Face Image (Fallback) --------
-def run_hf_image(prompt: str) -> str:
-    """Generate image using Hugging Face Stable Diffusion."""
-    if not HF_TOKEN:
-        return None
+def run_openai_image(prompt):
+    """Generate image with OpenAI DALL·E or fallback."""
     try:
-        url = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
-        headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-        payload = {"inputs": prompt}
-        r = requests.post(url, headers=headers, json=payload, timeout=60)
+        r = requests.post(
+            "https://api.openai.com/v1/images/generations",
+            headers={"Authorization": f"Bearer {OPENAI_API_KEY}"},
+            json={"model": "gpt-image-1", "prompt": prompt, "size": "1024x1024"}
+        )
         r.raise_for_status()
-        # Hugging Face returns raw image bytes, so we save to file
-        image_path = f"generated_{hash(prompt)}.png"
-        with open(image_path, "wb") as f:
-            f.write(r.content)
-        return image_path
+        return r.json()["data"][0]["url"]
     except Exception as e:
-        print(f"[HF Image] Failed: {e}")
-        return None
+        return f"[OpenAI Image Error] {e}"
 
-# -------- PubMed Fallback --------
-def pubmed_fallback_search(query: str, api_key: str, email: str) -> List[Dict]:
-    """Search PubMed if no citations found."""
-    results = []
-    if not api_key or not email:
-        return results
+def pubmed_fallback_search(query, retmax=5):
+    """Simple PubMed search for citations."""
     try:
-        base_url = "https://eutils.ncbi.nlm.nih.gov/entrez/eutils/esearch.fcgi"
-        params = {
-            "db": "pubmed",
-            "term": query,
-            "retmax": 3,
-            "api_key": api_key,
-            "email": email,
-            "retmode": "json"
-        }
-        r = requests.get(base_url, params=params, timeout=10)
+        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().get("esearchresult", {}).get("idlist", [])
-        for pmid in ids:
-            results.append({
-                "type": "PMID",
-                "id": pmid,
-                "url": f"https://pubmed.ncbi.nlm.nih.gov/{pmid}/"
-            })
+        ids = r.json()["esearchresult"]["idlist"]
+        return [{"pmid": pid, "url": f"https://pubmed.ncbi.nlm.nih.gov/{pid}/"} for pid in ids]
     except Exception as e:
-        print(f"[PubMed] Failed: {e}")
-    return results
+        return [{"error": str(e)}]