Update genesis/trials.py

genesis/trials.py

@@ -1,60 +1,25 @@
 # genesis/trials.py
+"""
+Clinical Trials API integration for GENESIS-AI
+Fetches trials from ClinicalTrials.gov.
+"""
+
 import requests
-from typing import List, Dict
 
-CLINICAL_TRIALS_API = "https://clinicaltrials.gov/api/query/study_fields"
+BASE_URL = "https://clinicaltrials.gov/api/query/study_fields"
 
-def search_clinical_trials(query: str, max_results: int = 5) -> List[Dict]:
-    """
-    Search ClinicalTrials.gov for trials matching the query.
-    
-    Returns a list of dictionaries with:
-    - title
-    - status
-    - nct_id
-    - locations
-    - url
-    """
+def fetch_clinical_trials(query, max_results=10):
+    """Fetch clinical trial summaries for a given query."""
     try:
         params = {
             "expr": query,
-            "fields": "NCTId,BriefTitle,OverallStatus,LocationCity,LocationCountry",
+            "fields": "NCTId,BriefTitle,OverallStatus,StartDate,CompletionDate,Phase,LocationCountry",
             "min_rnk": 1,
             "max_rnk": max_results,
             "fmt": "json"
         }
-        r = requests.get(CLINICAL_TRIALS_API, params=params, timeout=15)
+        r = requests.get(BASE_URL, params=params)
         r.raise_for_status()
-        data = r.json()
-
-        studies = data.get("StudyFieldsResponse", {}).get("StudyFields", [])
-        results = []
-        for s in studies:
-            nct_id = s.get("NCTId", [""])[0]
-            title = s.get("BriefTitle", [""])[0]
-            status = s.get("OverallStatus", [""])[0]
-            cities = s.get("LocationCity", [])
-            countries = s.get("LocationCountry", [])
-            locations = [f"{c}, {countries[i]}" if i < len(countries) else c 
-                         for i, c in enumerate(cities)]
-            results.append({
-                "title": title,
-                "status": status,
-                "nct_id": nct_id,
-                "locations": locations,
-                "url": f"https://clinicaltrials.gov/study/{nct_id}"
-            })
-        return results
+        return r.json().get("StudyFieldsResponse", {}).get("StudyFields", [])
     except Exception as e:
-        print(f"[ClinicalTrials] Failed: {e}")
-        return []
-
-def format_trials_markdown(trials: List[Dict]) -> str:
-    """Convert trials list into markdown table."""
-    if not trials:
-        return "No clinical trials found."
-    md = "| Title | Status | NCT ID | Locations |\n|-------|--------|--------|-----------|\n"
-    for t in trials:
-        loc_str = ", ".join(t["locations"]) if t["locations"] else "N/A"
-        md += f"| [{t['title']}]({t['url']}) | {t['status']} | [{t['nct_id']}]({t['url']}) | {loc_str} |\n"
-    return md
+        return {"error": str(e)}