debase 0.5.1__tar.gz → 0.6.0__tar.gz

{debase-0.5.1/src/debase.egg-info → debase-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: debase
-Version: 0.5.1
+Version: 0.6.0
 Summary: Enzyme lineage analysis and sequence extraction package
 Home-page: https://github.com/YuemingLong/DEBase
 Author: DEBase Team

{debase-0.5.1 → debase-0.6.0}/src/debase/_version.py

@@ -1,3 +1,3 @@
 """Version information."""
 
-__version__ = "0.5.1"
+__version__ = "0.6.0"

{debase-0.5.1 → debase-0.6.0}/src/debase/lineage_format.py

@@ -35,7 +35,6 @@ import logging
 import os
 import pickle
 import re
-import sqlite3
 import sys
 import time
 from concurrent.futures import ThreadPoolExecutor, as_completed
@@ -137,8 +136,7 @@ SUBSTRATE_CACHE_FILE: Path = CACHE_DIR / "substrate_smiles_cache.pkl"
 CANONICAL_CACHE_FILE: Path = CACHE_DIR / "canonical_smiles_cache.pkl"
 CACHE_DIR.mkdir(parents=True, exist_ok=True)
 
-# Local PubChem DB (optional) --------------------------------------------------------
-PUBCHEM_DB_PATH: Path = Path(__file__).parent.parent.parent / "data" / "iupac2smiles.db"
+# API endpoints for IUPAC to SMILES conversion --------------------------------------
 
 # Gemini API configuration -----------------------------------------------------------
 GEMINI_API_KEY: str = os.environ.get("GEMINI_API_KEY", "")
@@ -323,37 +321,7 @@ SUBSTRATE_CACHE: Dict[str, str] = _load_pickle(SUBSTRATE_CACHE_FILE)
 CANONICAL_CACHE: Dict[str, str] = _load_pickle(CANONICAL_CACHE_FILE)
 
 
-# --- Database lookup ---------------------------------------------------------------
-class PubChemDB:
-    """Very thin wrapper around a local SQLite mapping IUPAC -> SMILES."""
-
-    def __init__(self, path: Path | str) -> None:
-        self.path = Path(path)
-        self._conn: Optional[sqlite3.Connection] = None
-        if not self.path.exists():
-            log.warning("Local PubChem DB not found at %s", self.path)
-
-    def _connect(self) -> sqlite3.Connection:
-        if self._conn is None:
-            self._conn = sqlite3.connect(str(self.path))
-        return self._conn
-
-    def lookup(self, name: str) -> Optional[str]:
-        if not self.path.exists():
-            return None
-        sql = "SELECT smiles FROM x WHERE name = ? LIMIT 1"
-        try:
-            # Create a new connection for thread safety
-            conn = sqlite3.connect(str(self.path))
-            cur = conn.execute(sql, (name.lower(),))
-            row = cur.fetchone()
-            conn.close()
-            return row[0] if row else None
-        except Exception:  # pragma: no cover
-            return None
-
-
-PC_DB = PubChemDB(PUBCHEM_DB_PATH)
+# --- Removed local database - using only online APIs -------------------------------
 
 
 # === 5. SEQUENCE / MUTATION HELPERS ================================================
@@ -481,12 +449,7 @@ def _name_to_smiles(name: str, is_substrate: bool) -> str:
     if not name or name.lower() in ['nan', 'none', 'null', 'n/a', 'na', '']:
         return ""
     
-    # 1. Local DB (fast, offline)
-    db_smiles = PC_DB.lookup(name)
-    if db_smiles:
-        return db_smiles
-
-    # 2. OPSIN (if installed) ---------------------------------------------------
+    # 1. OPSIN (if installed) - fast and reliable for IUPAC names
     try:
         import subprocess
 
@@ -503,12 +466,7 @@ def _name_to_smiles(name: str, is_substrate: bool) -> str:
     except FileNotFoundError:
         pass  # OPSIN not installed
 
-    # 3. Gemini search (for complex compounds) ---------------------------------
-    gemini_smiles = search_smiles_with_gemini(name)
-    if gemini_smiles:
-        return gemini_smiles
-    
-    # 4. PubChem PUG REST (online) ---------------------------------------------
+    # 2. PubChem PUG REST API (online) - comprehensive database
     try:
         import requests
 
@@ -521,6 +479,11 @@ def _name_to_smiles(name: str, is_substrate: bool) -> str:
             return pug_smiles
     except Exception:  # pragma: no cover
         pass
+    
+    # 3. Gemini search (for complex compounds) - AI fallback
+    gemini_smiles = search_smiles_with_gemini(name)
+    if gemini_smiles:
+        return gemini_smiles
 
     # Return empty string if all methods fail
     return ""

{debase-0.5.1 → debase-0.6.0/src/debase.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: debase
-Version: 0.5.1
+Version: 0.6.0
 Summary: Enzyme lineage analysis and sequence extraction package
 Home-page: https://github.com/YuemingLong/DEBase
 Author: DEBase Team