Spaces:
No application file
No application file
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ import time
|
|
| 25 |
|
| 26 |
# --- Global Variables ---
|
| 27 |
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
| 28 |
-
MAFFT_PATH = os.path.join(BASE_DIR, "binaries", "mafft", "mafft")
|
| 29 |
IQTREE_PATH = os.path.join(BASE_DIR, "binaries", "iqtree", "bin", "iqtree3")
|
| 30 |
ALIGNMENT_PATH = os.path.join(BASE_DIR, "f_gene_sequences_aligned.fasta")
|
| 31 |
TREE_PATH = os.path.join(BASE_DIR, "f_gene_sequences.phy.treefile")
|
|
@@ -35,20 +35,17 @@ os.makedirs(QUERY_OUTPUT_DIR, exist_ok=True)
|
|
| 35 |
# --- Logging ---
|
| 36 |
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
| 37 |
|
| 38 |
-
# ---
|
| 39 |
-
# Model repository and file paths
|
| 40 |
model_repo = "GGproject10/best_boundary_aware_model"
|
| 41 |
csv_path = "f cleaned.csv"
|
| 42 |
-
|
| 43 |
-
# Get HF token from environment (if available)
|
| 44 |
hf_token = os.getenv("HF_TOKEN")
|
| 45 |
|
| 46 |
-
# --- Load Models ---
|
| 47 |
boundary_model = None
|
| 48 |
keras_model = None
|
| 49 |
kmer_to_index = None
|
|
|
|
| 50 |
|
| 51 |
-
#
|
| 52 |
try:
|
| 53 |
boundary_path = hf_hub_download(
|
| 54 |
repo_id=model_repo,
|
|
@@ -58,12 +55,9 @@ try:
|
|
| 58 |
if os.path.exists(boundary_path):
|
| 59 |
boundary_model = GenePredictor(boundary_path)
|
| 60 |
logging.info("Boundary model loaded successfully from Hugging Face Hub.")
|
| 61 |
-
else:
|
| 62 |
-
logging.warning(f"Boundary model file not found after download")
|
| 63 |
except Exception as e:
|
| 64 |
logging.error(f"Failed to load boundary model from HF Hub: {e}")
|
| 65 |
|
| 66 |
-
# Try to load Keras model from Hugging Face Hub
|
| 67 |
try:
|
| 68 |
keras_path = hf_hub_download(
|
| 69 |
repo_id=model_repo,
|
|
@@ -80,56 +74,11 @@ try:
|
|
| 80 |
keras_model = load_model(keras_path)
|
| 81 |
with open(kmer_path, "rb") as f:
|
| 82 |
kmer_to_index = pickle.load(f)
|
| 83 |
-
logging.info("Keras model and k-mer index loaded successfully
|
| 84 |
-
else:
|
| 85 |
-
logging.warning(f"Keras model or kmer files not found after download")
|
| 86 |
except Exception as e:
|
| 87 |
logging.error(f"Failed to load Keras model from HF Hub: {e}")
|
| 88 |
|
| 89 |
-
#
|
| 90 |
-
analyzer = None
|
| 91 |
-
try:
|
| 92 |
-
analyzer = PhylogeneticTreeAnalyzer()
|
| 93 |
-
|
| 94 |
-
# Try multiple potential locations for the CSV file
|
| 95 |
-
csv_candidates = [
|
| 96 |
-
csv_path,
|
| 97 |
-
os.path.join(BASE_DIR, csv_path),
|
| 98 |
-
os.path.join(BASE_DIR, "app", csv_path),
|
| 99 |
-
os.path.join(os.path.dirname(__file__), csv_path),
|
| 100 |
-
"f_cleaned.csv", # Alternative naming
|
| 101 |
-
os.path.join(BASE_DIR, "f_cleaned.csv")
|
| 102 |
-
]
|
| 103 |
-
|
| 104 |
-
csv_loaded = False
|
| 105 |
-
for csv_candidate in csv_candidates:
|
| 106 |
-
if os.path.exists(csv_candidate):
|
| 107 |
-
if analyzer.load_data(csv_candidate):
|
| 108 |
-
logging.info(f"Tree analyzer data loaded from: {csv_candidate}")
|
| 109 |
-
csv_loaded = True
|
| 110 |
-
csv_path = csv_candidate # Update path for consistency
|
| 111 |
-
break
|
| 112 |
-
else:
|
| 113 |
-
logging.warning(f"Failed to load data from: {csv_candidate}")
|
| 114 |
-
|
| 115 |
-
if not csv_loaded:
|
| 116 |
-
logging.error("Failed to load CSV data from any candidate location")
|
| 117 |
-
analyzer = None
|
| 118 |
-
else:
|
| 119 |
-
# Try to train AI model (optional)
|
| 120 |
-
try:
|
| 121 |
-
if analyzer.train_ai_model():
|
| 122 |
-
logging.info("AI model training completed successfully")
|
| 123 |
-
else:
|
| 124 |
-
logging.warning("AI model training failed; proceeding with basic analysis.")
|
| 125 |
-
except Exception as e:
|
| 126 |
-
logging.warning(f"AI model training failed: {e}")
|
| 127 |
-
|
| 128 |
-
except Exception as e:
|
| 129 |
-
logging.error(f"Failed to initialize tree analyzer: {e}")
|
| 130 |
-
analyzer = None
|
| 131 |
-
|
| 132 |
-
# --- Enhanced Tool Detection with Binary Permission Setup ---
|
| 133 |
def setup_binary_permissions():
|
| 134 |
"""Set executable permissions on MAFFT and IQ-TREE binaries"""
|
| 135 |
binaries = [MAFFT_PATH, IQTREE_PATH]
|
|
@@ -137,464 +86,92 @@ def setup_binary_permissions():
|
|
| 137 |
for binary in binaries:
|
| 138 |
if os.path.exists(binary):
|
| 139 |
try:
|
| 140 |
-
# Set executable permission
|
| 141 |
current_mode = os.stat(binary).st_mode
|
| 142 |
os.chmod(binary, current_mode | stat.S_IEXEC)
|
| 143 |
logging.info(f"Set executable permission on {binary}")
|
| 144 |
except Exception as e:
|
| 145 |
logging.warning(f"Failed to set executable permission on {binary}: {e}")
|
| 146 |
-
else:
|
| 147 |
-
logging.warning(f"Binary not found: {binary}")
|
| 148 |
|
| 149 |
def check_tool_availability():
|
| 150 |
-
"""Enhanced check for MAFFT and IQ-TREE availability
|
| 151 |
-
|
| 152 |
-
# First, ensure binaries have executable permissions
|
| 153 |
setup_binary_permissions()
|
| 154 |
|
| 155 |
# Check MAFFT
|
| 156 |
mafft_available = False
|
| 157 |
mafft_cmd = None
|
| 158 |
|
| 159 |
-
# Updated MAFFT candidates list based on your new API
|
| 160 |
mafft_candidates = [
|
| 161 |
-
MAFFT_PATH,
|
| 162 |
-
os.path.join(BASE_DIR, "binaries", "mafft", "mafft"),
|
| 163 |
-
os.path.join(BASE_DIR, "binaries", "mafft", "mafft.bat"), # Windows fallback
|
| 164 |
'mafft',
|
| 165 |
'/usr/bin/mafft',
|
| 166 |
'/usr/local/bin/mafft',
|
| 167 |
-
os.path.join(BASE_DIR, "binaries", "mafft", "mafftdir", "bin", "mafft"),
|
| 168 |
-
# Add potential conda/miniconda paths
|
| 169 |
-
os.path.expanduser("~/anaconda3/bin/mafft"),
|
| 170 |
-
os.path.expanduser("~/miniconda3/bin/mafft"),
|
| 171 |
-
"/opt/conda/bin/mafft",
|
| 172 |
-
"/usr/local/miniconda3/bin/mafft"
|
| 173 |
]
|
| 174 |
|
| 175 |
for candidate in mafft_candidates:
|
| 176 |
if not candidate:
|
| 177 |
continue
|
| 178 |
-
|
| 179 |
-
# First check if file exists or is in PATH
|
| 180 |
if os.path.exists(candidate) or shutil.which(candidate):
|
| 181 |
-
# Now test actual execution
|
| 182 |
try:
|
| 183 |
-
|
| 184 |
-
result = subprocess.run(
|
| 185 |
-
test_cmd,
|
| 186 |
-
capture_output=True,
|
| 187 |
-
text=True,
|
| 188 |
-
timeout=10
|
| 189 |
-
)
|
| 190 |
if result.returncode == 0 or "mafft" in result.stderr.lower():
|
| 191 |
mafft_available = True
|
| 192 |
mafft_cmd = candidate
|
| 193 |
-
logging.info(f"MAFFT found and tested successfully at: {candidate}")
|
| 194 |
break
|
| 195 |
-
except
|
| 196 |
-
logging.debug(f"MAFFT test failed for {candidate}: {e}")
|
| 197 |
continue
|
| 198 |
|
| 199 |
-
# Check IQ-TREE
|
| 200 |
iqtree_available = False
|
| 201 |
iqtree_cmd = None
|
| 202 |
|
| 203 |
-
# Updated IQ-TREE candidates list
|
| 204 |
iqtree_candidates = [
|
| 205 |
-
IQTREE_PATH,
|
| 206 |
'iqtree2',
|
| 207 |
'iqtree',
|
| 208 |
-
'iqtree3',
|
| 209 |
'/usr/bin/iqtree2',
|
| 210 |
'/usr/local/bin/iqtree2',
|
| 211 |
-
'/usr/bin/iqtree',
|
| 212 |
-
'/usr/local/bin/iqtree',
|
| 213 |
-
'iqtree2.exe', # Windows
|
| 214 |
-
'iqtree.exe', # Windows
|
| 215 |
-
'iqtree3.exe', # Windows
|
| 216 |
-
os.path.join(BASE_DIR, "binaries", "iqtree", "bin", "iqtree2"),
|
| 217 |
-
# Add potential conda paths
|
| 218 |
-
os.path.expanduser("~/anaconda3/bin/iqtree2"),
|
| 219 |
-
os.path.expanduser("~/miniconda3/bin/iqtree2"),
|
| 220 |
-
"/opt/conda/bin/iqtree2",
|
| 221 |
-
"/usr/local/miniconda3/bin/iqtree2"
|
| 222 |
]
|
| 223 |
|
| 224 |
for candidate in iqtree_candidates:
|
| 225 |
if not candidate:
|
| 226 |
continue
|
| 227 |
-
|
| 228 |
if os.path.exists(candidate) or shutil.which(candidate):
|
| 229 |
try:
|
| 230 |
-
|
| 231 |
-
result = subprocess.run(
|
| 232 |
-
test_cmd,
|
| 233 |
-
capture_output=True,
|
| 234 |
-
text=True,
|
| 235 |
-
timeout=10
|
| 236 |
-
)
|
| 237 |
if result.returncode == 0 or "iqtree" in result.stderr.lower():
|
| 238 |
iqtree_available = True
|
| 239 |
iqtree_cmd = candidate
|
| 240 |
-
logging.info(f"IQ-TREE found and tested successfully at: {candidate}")
|
| 241 |
break
|
| 242 |
-
except
|
| 243 |
-
logging.debug(f"IQ-TREE test failed for {candidate}: {e}")
|
| 244 |
continue
|
| 245 |
|
| 246 |
return mafft_available, iqtree_available, mafft_cmd, iqtree_cmd
|
| 247 |
|
| 248 |
-
def install_dependencies_guide():
|
| 249 |
-
"""Provide installation guidance for missing dependencies"""
|
| 250 |
-
guide = """
|
| 251 |
-
🔧 INSTALLATION GUIDE FOR MISSING DEPENDENCIES:
|
| 252 |
-
|
| 253 |
-
For MAFFT:
|
| 254 |
-
- Ubuntu/Debian: sudo apt-get install mafft
|
| 255 |
-
- CentOS/RHEL: sudo yum install mafft
|
| 256 |
-
- macOS: brew install mafft
|
| 257 |
-
- Windows: Download from https://mafft.cbrc.jp/alignment/software/
|
| 258 |
-
- Conda: conda install -c bioconda mafft
|
| 259 |
-
|
| 260 |
-
For IQ-TREE:
|
| 261 |
-
- Ubuntu/Debian: sudo apt-get install iqtree
|
| 262 |
-
- CentOS/RHEL: sudo yum install iqtree
|
| 263 |
-
- macOS: brew install iqtree
|
| 264 |
-
- Windows: Download from http://www.iqtree.org/
|
| 265 |
-
- Conda: conda install -c bioconda iqtree
|
| 266 |
-
|
| 267 |
-
Alternative: Use conda/mamba (RECOMMENDED):
|
| 268 |
-
- conda install -c bioconda mafft iqtree
|
| 269 |
-
|
| 270 |
-
Docker option:
|
| 271 |
-
- docker run -it --rm -v $(pwd):/data quay.io/biocontainers/mafft:7.490--h779adbc_0
|
| 272 |
-
- docker run -it --rm -v $(pwd):/data quay.io/biocontainers/iqtree:2.1.4_beta--hdcc8f71_0
|
| 273 |
-
|
| 274 |
-
TROUBLESHOOTING:
|
| 275 |
-
If tools are installed but not detected, try:
|
| 276 |
-
1. Add installation directory to PATH
|
| 277 |
-
2. Use absolute paths in the configuration
|
| 278 |
-
3. Check permissions on executable files
|
| 279 |
-
4. Ensure binaries have executable permissions (chmod +x)
|
| 280 |
-
"""
|
| 281 |
-
return guide
|
| 282 |
-
|
| 283 |
-
def phylogenetic_placement(sequence: str, mafft_cmd: str, iqtree_cmd: str):
|
| 284 |
-
"""
|
| 285 |
-
Improved phylogenetic placement using the new API approach.
|
| 286 |
-
This adds the query sequence to a reference alignment and tree.
|
| 287 |
-
"""
|
| 288 |
-
try:
|
| 289 |
-
# Validate sequence
|
| 290 |
-
if len(sequence.strip()) < 100:
|
| 291 |
-
return False, "Error: Sequence is too short for phylogenetic placement (minimum 100 bp).", None, None
|
| 292 |
-
|
| 293 |
-
# Generate unique query ID
|
| 294 |
-
query_id = f"QUERY_{uuid.uuid4().hex[:8]}"
|
| 295 |
-
query_fasta = os.path.join(QUERY_OUTPUT_DIR, f"{query_id}.fa")
|
| 296 |
-
aligned_with_query = os.path.join(QUERY_OUTPUT_DIR, f"{query_id}_aligned.fa")
|
| 297 |
-
output_prefix = os.path.join(QUERY_OUTPUT_DIR, f"{query_id}_placed_tree")
|
| 298 |
-
|
| 299 |
-
# Check if reference files exist
|
| 300 |
-
if not os.path.exists(ALIGNMENT_PATH):
|
| 301 |
-
return False, f"Reference alignment not found: {ALIGNMENT_PATH}", None, None
|
| 302 |
-
|
| 303 |
-
if not os.path.exists(TREE_PATH):
|
| 304 |
-
return False, f"Reference tree not found: {TREE_PATH}", None, None
|
| 305 |
-
|
| 306 |
-
# Save query sequence as FASTA (improved error handling)
|
| 307 |
-
try:
|
| 308 |
-
query_record = SeqRecord(Seq(sequence.upper()), id=query_id, description="")
|
| 309 |
-
SeqIO.write([query_record], query_fasta, "fasta")
|
| 310 |
-
logging.info(f"Query sequence saved: {query_fasta}")
|
| 311 |
-
except Exception as e:
|
| 312 |
-
return False, f"Error writing query sequence: {e}", None, None
|
| 313 |
-
|
| 314 |
-
# Step 1: Add query sequence to reference alignment using MAFFT (improved approach)
|
| 315 |
-
logging.info("Adding query sequence to reference alignment...")
|
| 316 |
-
try:
|
| 317 |
-
with open(aligned_with_query, "w") as output_file:
|
| 318 |
-
mafft_result = subprocess.run([
|
| 319 |
-
mafft_cmd, "--add", query_fasta, "--reorder", ALIGNMENT_PATH
|
| 320 |
-
], stdout=output_file, stderr=subprocess.PIPE, text=True, timeout=600, check=True)
|
| 321 |
-
|
| 322 |
-
# Verify alignment file was created and is not empty
|
| 323 |
-
if not os.path.exists(aligned_with_query) or os.path.getsize(aligned_with_query) == 0:
|
| 324 |
-
return False, "MAFFT alignment failed: output file is empty", None, None
|
| 325 |
-
|
| 326 |
-
logging.info(f"MAFFT alignment completed: {aligned_with_query}")
|
| 327 |
-
|
| 328 |
-
except subprocess.CalledProcessError as e:
|
| 329 |
-
error_msg = e.stderr if e.stderr else "Unknown MAFFT error"
|
| 330 |
-
return False, f"MAFFT alignment failed: {error_msg}", None, None
|
| 331 |
-
except subprocess.TimeoutExpired:
|
| 332 |
-
return False, "MAFFT alignment timeout (>10 minutes)", None, None
|
| 333 |
-
except FileNotFoundError:
|
| 334 |
-
return False, f"MAFFT executable not found: {mafft_cmd}", None, None
|
| 335 |
-
except Exception as e:
|
| 336 |
-
return False, f"MAFFT execution error: {e}", None, None
|
| 337 |
-
|
| 338 |
-
# Step 2: Place sequence in phylogenetic tree using IQ-TREE (improved approach)
|
| 339 |
-
logging.info("Placing sequence in phylogenetic tree...")
|
| 340 |
-
try:
|
| 341 |
-
iqtree_result = subprocess.run([
|
| 342 |
-
iqtree_cmd, "-s", aligned_with_query, "-g", TREE_PATH,
|
| 343 |
-
"-m", "GTR+G", "-pre", output_prefix, "-redo"
|
| 344 |
-
], capture_output=True, text=True, timeout=1200, check=True)
|
| 345 |
-
|
| 346 |
-
# Check if treefile was generated
|
| 347 |
-
treefile = f"{output_prefix}.treefile"
|
| 348 |
-
if not os.path.exists(treefile) or os.path.getsize(treefile) == 0:
|
| 349 |
-
return False, "IQ-TREE placement failed: treefile not generated", aligned_with_query, None
|
| 350 |
-
|
| 351 |
-
logging.info(f"IQ-TREE placement completed: {treefile}")
|
| 352 |
-
|
| 353 |
-
# Generate success message with details
|
| 354 |
-
success_msg = "✅ Phylogenetic placement completed successfully!\n"
|
| 355 |
-
success_msg += f"- Query ID: {query_id}\n"
|
| 356 |
-
success_msg += f"- Alignment: {os.path.basename(aligned_with_query)}\n"
|
| 357 |
-
success_msg += f"- Tree: {os.path.basename(treefile)}\n"
|
| 358 |
-
|
| 359 |
-
# Try to extract model information from log
|
| 360 |
-
log_file = f"{output_prefix}.log"
|
| 361 |
-
if os.path.exists(log_file):
|
| 362 |
-
try:
|
| 363 |
-
with open(log_file, 'r') as f:
|
| 364 |
-
log_content = f.read()
|
| 365 |
-
if "Log-likelihood" in log_content:
|
| 366 |
-
log_lines = [line for line in log_content.split('\n') if "Log-likelihood" in line]
|
| 367 |
-
if log_lines:
|
| 368 |
-
success_msg += f"- {log_lines[0].strip()}\n"
|
| 369 |
-
except Exception as e:
|
| 370 |
-
logging.warning(f"Could not read log file: {e}")
|
| 371 |
-
|
| 372 |
-
return True, success_msg, aligned_with_query, treefile
|
| 373 |
-
|
| 374 |
-
except subprocess.CalledProcessError as e:
|
| 375 |
-
error_msg = e.stderr if e.stderr else "Unknown IQ-TREE error"
|
| 376 |
-
return False, f"IQ-TREE placement failed: {error_msg}", aligned_with_query, None
|
| 377 |
-
except subprocess.TimeoutExpired:
|
| 378 |
-
return False, "IQ-TREE placement timeout (>20 minutes)", aligned_with_query, None
|
| 379 |
-
except FileNotFoundError:
|
| 380 |
-
return False, f"IQ-TREE executable not found: {iqtree_cmd}", aligned_with_query, None
|
| 381 |
-
except Exception as e:
|
| 382 |
-
return False, f"IQ-TREE execution error: {e}", aligned_with_query, None
|
| 383 |
-
|
| 384 |
-
except Exception as e:
|
| 385 |
-
logging.error(f"Phylogenetic placement failed: {e}")
|
| 386 |
-
return False, f"Phylogenetic placement failed: {str(e)}", None, None
|
| 387 |
-
finally:
|
| 388 |
-
# Clean up temporary query file
|
| 389 |
-
if 'query_fasta' in locals() and os.path.exists(query_fasta):
|
| 390 |
-
try:
|
| 391 |
-
os.unlink(query_fasta)
|
| 392 |
-
except:
|
| 393 |
-
pass
|
| 394 |
-
|
| 395 |
-
def build_maximum_likelihood_tree(f_gene_sequence):
|
| 396 |
-
"""
|
| 397 |
-
Build maximum likelihood phylogenetic tree using the improved phylogenetic placement approach.
|
| 398 |
-
"""
|
| 399 |
-
try:
|
| 400 |
-
# Check tool availability with enhanced detection
|
| 401 |
-
mafft_available, iqtree_available, mafft_cmd, iqtree_cmd = check_tool_availability()
|
| 402 |
-
|
| 403 |
-
# Prepare status message
|
| 404 |
-
status_msg = "🔍 Checking dependencies...\n"
|
| 405 |
-
|
| 406 |
-
if not mafft_available:
|
| 407 |
-
status_msg += "❌ MAFFT not found or not executable\n"
|
| 408 |
-
else:
|
| 409 |
-
status_msg += f"✅ MAFFT found and tested: {mafft_cmd}\n"
|
| 410 |
-
|
| 411 |
-
if not iqtree_available:
|
| 412 |
-
status_msg += "❌ IQ-TREE not found or not executable\n"
|
| 413 |
-
else:
|
| 414 |
-
status_msg += f"✅ IQ-TREE found and tested: {iqtree_cmd}\n"
|
| 415 |
-
|
| 416 |
-
# Check for reference files
|
| 417 |
-
if not os.path.exists(ALIGNMENT_PATH):
|
| 418 |
-
status_msg += f"❌ Reference alignment not found: {ALIGNMENT_PATH}\n"
|
| 419 |
-
else:
|
| 420 |
-
status_msg += f"✅ Reference alignment found\n"
|
| 421 |
-
|
| 422 |
-
if not os.path.exists(TREE_PATH):
|
| 423 |
-
status_msg += f"❌ Reference tree not found: {TREE_PATH}\n"
|
| 424 |
-
else:
|
| 425 |
-
status_msg += f"✅ Reference tree found\n"
|
| 426 |
-
|
| 427 |
-
# If any required component is missing, provide installation guide
|
| 428 |
-
if not mafft_available or not iqtree_available:
|
| 429 |
-
guide = install_dependencies_guide()
|
| 430 |
-
return False, f"{status_msg}\n{guide}", None, None
|
| 431 |
-
|
| 432 |
-
if not os.path.exists(ALIGNMENT_PATH) or not os.path.exists(TREE_PATH):
|
| 433 |
-
status_msg += "\n❌ Reference alignment and/or tree files are missing.\n"
|
| 434 |
-
status_msg += "Please ensure f_gene_sequences_aligned.fasta and f_gene_sequences.phy.treefile are available."
|
| 435 |
-
return False, status_msg, None, None
|
| 436 |
-
|
| 437 |
-
# Perform phylogenetic placement using improved method
|
| 438 |
-
logging.info("Starting phylogenetic placement...")
|
| 439 |
-
placement_success, placement_message, aligned_file, tree_file = phylogenetic_placement(
|
| 440 |
-
f_gene_sequence, mafft_cmd, iqtree_cmd
|
| 441 |
-
)
|
| 442 |
-
|
| 443 |
-
if placement_success:
|
| 444 |
-
final_message = f"{status_msg}\n{placement_message}"
|
| 445 |
-
|
| 446 |
-
# Copy files to standard locations for compatibility
|
| 447 |
-
if aligned_file and os.path.exists(aligned_file):
|
| 448 |
-
standard_aligned = "query_with_references_aligned.fasta"
|
| 449 |
-
shutil.copy2(aligned_file, standard_aligned)
|
| 450 |
-
aligned_file = standard_aligned
|
| 451 |
-
|
| 452 |
-
if tree_file and os.path.exists(tree_file):
|
| 453 |
-
standard_tree = "query_placement_tree.treefile"
|
| 454 |
-
shutil.copy2(tree_file, standard_tree)
|
| 455 |
-
tree_file = standard_tree
|
| 456 |
-
|
| 457 |
-
logging.info("Phylogenetic placement completed successfully")
|
| 458 |
-
return True, final_message, aligned_file, tree_file
|
| 459 |
-
else:
|
| 460 |
-
return False, f"{status_msg}\n{placement_message}", aligned_file, tree_file
|
| 461 |
-
|
| 462 |
-
except Exception as e:
|
| 463 |
-
logging.error(f"ML tree construction failed: {e}")
|
| 464 |
-
return False, f"ML tree construction failed: {str(e)}", None, None
|
| 465 |
-
|
| 466 |
-
# --- NEW Tree Analysis Function (Using the new analyzer API) ---
|
| 467 |
-
def analyze_sequence_for_tree(sequence: str, matching_percentage: float) -> tuple:
|
| 468 |
-
"""
|
| 469 |
-
Analyze sequence and create phylogenetic tree using the new analyzer API
|
| 470 |
-
|
| 471 |
-
Args:
|
| 472 |
-
sequence (str): DNA sequence to analyze
|
| 473 |
-
matching_percentage (float): Similarity threshold percentage
|
| 474 |
-
|
| 475 |
-
Returns:
|
| 476 |
-
tuple: (status_message, html_file_path)
|
| 477 |
-
"""
|
| 478 |
-
try:
|
| 479 |
-
if not analyzer:
|
| 480 |
-
return "❌ Error: Tree analyzer not initialized. Please check if the CSV data file is available.", None
|
| 481 |
-
|
| 482 |
-
if not sequence:
|
| 483 |
-
return "❌ Error: Please provide a sequence.", None
|
| 484 |
-
|
| 485 |
-
if not (1 <= matching_percentage <= 99):
|
| 486 |
-
return "❌ Error: Matching percentage must be between 1 and 99.", None
|
| 487 |
-
|
| 488 |
-
# Validate inputs
|
| 489 |
-
sequence = sequence.strip()
|
| 490 |
-
if len(sequence) < 10:
|
| 491 |
-
return "❌ Error: Invalid or missing sequence. Must be ≥10 nucleotides.", None
|
| 492 |
-
|
| 493 |
-
# Find query sequence
|
| 494 |
-
if not analyzer.find_query_sequence(sequence):
|
| 495 |
-
return "❌ Error: Sequence not accepted.", None
|
| 496 |
-
|
| 497 |
-
# Find similar sequences
|
| 498 |
-
matched_ids, actual_percentage = analyzer.find_similar_sequences(matching_percentage)
|
| 499 |
-
|
| 500 |
-
if not matched_ids:
|
| 501 |
-
return f"❌ Error: No similar sequences found at {matching_percentage}% similarity threshold.", None
|
| 502 |
-
|
| 503 |
-
logging.info(f"Found {len(matched_ids)} similar sequences at {actual_percentage:.2f}% similarity")
|
| 504 |
-
|
| 505 |
-
# Build tree structure
|
| 506 |
-
analyzer.build_tree_structure_with_ml_safe(matched_ids)
|
| 507 |
-
|
| 508 |
-
# Create interactive tree
|
| 509 |
-
fig = analyzer.create_interactive_tree(matched_ids, actual_percentage)
|
| 510 |
-
|
| 511 |
-
# Save to temporary file that Gradio can access
|
| 512 |
-
temp_dir = tempfile.gettempdir()
|
| 513 |
-
output_path = os.path.join(temp_dir, 'phylogenetic_tree_interactive.html')
|
| 514 |
-
fig.write_html(output_path)
|
| 515 |
-
|
| 516 |
-
success_msg = f"✅ Analysis complete! Found {len(matched_ids)} similar sequences with {actual_percentage:.2f}% average similarity."
|
| 517 |
-
|
| 518 |
-
return success_msg, output_path
|
| 519 |
-
|
| 520 |
-
except Exception as e:
|
| 521 |
-
error_msg = f"❌ Error during analysis: {str(e)}"
|
| 522 |
-
logging.error(error_msg)
|
| 523 |
-
import traceback
|
| 524 |
-
logging.error(f"Full traceback: {traceback.format_exc()}")
|
| 525 |
-
return error_msg, None
|
| 526 |
-
def get_tree_display_content(html_path):
|
| 527 |
-
"""Extract Plotly JSON from HTML and create embeddable content"""
|
| 528 |
-
try:
|
| 529 |
-
if not html_path or not os.path.exists(html_path):
|
| 530 |
-
return None
|
| 531 |
-
|
| 532 |
-
with open(html_path, 'r', encoding='utf-8') as f:
|
| 533 |
-
html_content = f.read()
|
| 534 |
-
|
| 535 |
-
# Extract the Plotly JSON data
|
| 536 |
-
import re
|
| 537 |
-
json_match = re.search(r'Plotly\.newPlot\([^,]+,\s*(\{.*?\}),', html_content, re.DOTALL)
|
| 538 |
-
if json_match:
|
| 539 |
-
plotly_json = json_match.group(1)
|
| 540 |
-
|
| 541 |
-
# Create a minimal HTML with just the essential Plotly code
|
| 542 |
-
minimal_html = f"""
|
| 543 |
-
<div id="plotly-div" style="width:100%;height:600px;"></div>
|
| 544 |
-
<script src="https://cdn.plot.ly/plotly-latest.min.js"></script>
|
| 545 |
-
<script>
|
| 546 |
-
var plotlyData = {plotly_json};
|
| 547 |
-
var layout = {{
|
| 548 |
-
title: 'Phylogenetic Tree',
|
| 549 |
-
xaxis: {{title: 'Distance'}},
|
| 550 |
-
yaxis: {{title: 'Taxa'}},
|
| 551 |
-
width: 800,
|
| 552 |
-
height: 600
|
| 553 |
-
}};
|
| 554 |
-
Plotly.newPlot('plotly-div', plotlyData.data, layout, {{responsive: true}});
|
| 555 |
-
</script>
|
| 556 |
-
"""
|
| 557 |
-
return minimal_html
|
| 558 |
-
|
| 559 |
-
return None
|
| 560 |
-
except Exception as e:
|
| 561 |
-
logging.error(f"Failed to extract Plotly content: {e}")
|
| 562 |
-
return None
|
| 563 |
-
# --- Keras Prediction ---
|
| 564 |
def predict_with_keras(sequence):
|
| 565 |
try:
|
| 566 |
if not keras_model or not kmer_to_index:
|
| 567 |
-
return f"Keras model not available.
|
| 568 |
|
| 569 |
if len(sequence) < 6:
|
| 570 |
-
return "
|
| 571 |
|
| 572 |
-
# Generate k-mers
|
| 573 |
kmers = [sequence[i:i+6] for i in range(len(sequence)-5)]
|
| 574 |
indices = [kmer_to_index.get(kmer, 0) for kmer in kmers]
|
| 575 |
|
| 576 |
-
# Prepare input
|
| 577 |
input_arr = np.array([indices])
|
| 578 |
prediction = keras_model.predict(input_arr, verbose=0)[0]
|
| 579 |
-
|
| 580 |
-
|
| 581 |
-
f_gene_prob = prediction[-1] # Take the probability of the F gene class
|
| 582 |
-
|
| 583 |
-
# Convert to percentage with a buffer (e.g., add 5% to account for minor mismatches)
|
| 584 |
-
percentage = min(100, max(0, int(f_gene_prob * 100 + 5))) # Ensure 0-100% range
|
| 585 |
|
| 586 |
return f"{percentage}% F gene"
|
| 587 |
except Exception as e:
|
| 588 |
-
logging.error(f"Keras prediction failed: {e}")
|
| 589 |
return f"Keras prediction failed: {str(e)}"
|
| 590 |
|
| 591 |
-
# --- FASTA Reader ---
|
| 592 |
def read_fasta_file(file_obj):
|
| 593 |
try:
|
| 594 |
if file_obj is None:
|
| 595 |
return ""
|
| 596 |
|
| 597 |
-
# Handle file object
|
| 598 |
if hasattr(file_obj, 'name'):
|
| 599 |
with open(file_obj.name, "r") as f:
|
| 600 |
content = f.read()
|
|
@@ -608,18 +185,58 @@ def read_fasta_file(file_obj):
|
|
| 608 |
logging.error(f"Failed to read FASTA file: {e}")
|
| 609 |
return ""
|
| 610 |
|
| 611 |
-
# -
|
| 612 |
-
def
|
|
|
|
|
|
|
|
|
|
|
|
|
| 613 |
try:
|
| 614 |
-
|
| 615 |
-
|
| 616 |
-
|
| 617 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 618 |
except Exception as e:
|
| 619 |
-
|
| 620 |
-
|
| 621 |
-
|
|
|
|
|
|
|
| 622 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 623 |
def run_pipeline(dna_input, similarity_score=95.0, build_ml_tree=False):
|
| 624 |
try:
|
| 625 |
# Clean input
|
|
@@ -630,10 +247,9 @@ def run_pipeline(dna_input, similarity_score=95.0, build_ml_tree=False):
|
|
| 630 |
# Sanitize DNA sequence
|
| 631 |
if not re.match('^[ACTGN]+$', dna_input):
|
| 632 |
dna_input = ''.join(c if c in 'ACTGN' else 'N' for c in dna_input)
|
| 633 |
-
logging.info("DNA sequence sanitized")
|
| 634 |
|
| 635 |
-
# Step 1: Boundary Prediction
|
| 636 |
-
processed_sequence = dna_input
|
| 637 |
boundary_output = ""
|
| 638 |
|
| 639 |
if boundary_model:
|
|
@@ -641,421 +257,246 @@ def run_pipeline(dna_input, similarity_score=95.0, build_ml_tree=False):
|
|
| 641 |
predictions, probs, confidence = boundary_model.predict(dna_input)
|
| 642 |
regions = boundary_model.extract_gene_regions(predictions, dna_input)
|
| 643 |
if regions:
|
| 644 |
-
processed_sequence = regions[0]["sequence"]
|
| 645 |
-
boundary_output = processed_sequence
|
| 646 |
-
logging.info(f"F gene extracted: {len(processed_sequence)} bp (confidence: {confidence:.3f})")
|
| 647 |
else:
|
| 648 |
-
boundary_output =
|
| 649 |
-
processed_sequence = dna_input
|
| 650 |
-
logging.warning("No gene regions found, using full sequence")
|
| 651 |
-
logging.info("Boundary model prediction completed")
|
| 652 |
except Exception as e:
|
| 653 |
-
logging.error(f"Boundary model failed: {e}")
|
| 654 |
boundary_output = f"Boundary model error: {str(e)}"
|
| 655 |
-
processed_sequence = dna_input # Fall back to original sequence
|
| 656 |
else:
|
| 657 |
-
boundary_output = f"Boundary model not available. Using
|
| 658 |
-
processed_sequence = dna_input
|
| 659 |
|
| 660 |
-
# Step 2: Keras Prediction
|
| 661 |
keras_output = ""
|
| 662 |
if processed_sequence and len(processed_sequence) >= 6:
|
| 663 |
-
|
| 664 |
-
# Use the prediction directly as it's now a percentage
|
| 665 |
-
keras_output = keras_prediction
|
| 666 |
else:
|
| 667 |
-
keras_output = "
|
| 668 |
-
|
| 669 |
-
# Step 3:
|
| 670 |
-
|
| 671 |
-
|
| 672 |
-
|
| 673 |
-
|
| 674 |
-
|
| 675 |
-
try:
|
| 676 |
-
logging.info("Starting phylogenetic placement...")
|
| 677 |
-
ml_success, ml_message, ml_aligned, ml_tree = build_maximum_likelihood_tree(processed_sequence)
|
| 678 |
-
|
| 679 |
-
if ml_success:
|
| 680 |
-
ml_tree_output = ml_message
|
| 681 |
-
aligned_file = ml_aligned
|
| 682 |
-
phy_file = ml_tree
|
| 683 |
-
else:
|
| 684 |
-
ml_tree_output = ml_message # This now includes detailed error information
|
| 685 |
-
|
| 686 |
-
except Exception as e:
|
| 687 |
-
ml_tree_output = f"❌ Phylogenetic placement failed: {str(e)}"
|
| 688 |
-
logging.error(f"Phylogenetic placement failed: {e}")
|
| 689 |
-
elif build_ml_tree:
|
| 690 |
-
ml_tree_output = "❌ F gene sequence too short for phylogenetic placement (minimum 100 bp)"
|
| 691 |
-
else:
|
| 692 |
-
ml_tree_output = "Phylogenetic placement skipped (not requested)"
|
| 693 |
-
|
| 694 |
-
# Step 4: NEW Simplified Tree Analysis (using the new analyzer API)
|
| 695 |
-
html_file = None
|
| 696 |
-
tree_html_content = "No tree generated"
|
| 697 |
-
simplified_ml_output = ""
|
| 698 |
-
|
| 699 |
-
if analyzer and processed_sequence and len(processed_sequence) >= 10:
|
| 700 |
-
try:
|
| 701 |
-
logging.info(f"Starting simplified ML tree analysis with F gene sequence length: {len(processed_sequence)}")
|
| 702 |
-
|
| 703 |
-
# Use the new analyze_sequence_for_tree function
|
| 704 |
-
tree_result, html_path = analyze_sequence_for_tree(processed_sequence, similarity_score)
|
| 705 |
-
|
| 706 |
-
if html_path and os.path.exists(html_path):
|
| 707 |
-
# Success - copy the HTML file to a location Gradio can serve
|
| 708 |
-
output_dir = "output"
|
| 709 |
-
os.makedirs(output_dir, exist_ok=True)
|
| 710 |
-
|
| 711 |
-
# Create a safe filename
|
| 712 |
-
safe_seq_name = re.sub(r'[^a-zA-Z0-9_-]', '', processed_sequence[:20])
|
| 713 |
-
timestamp = str(int(time.time()))
|
| 714 |
-
html_filename = f"tree_{safe_seq_name}_{timestamp}.html"
|
| 715 |
-
final_html_path = os.path.join(output_dir, html_filename)
|
| 716 |
-
|
| 717 |
-
# Copy the HTML file
|
| 718 |
-
shutil.copy2(html_path, final_html_path)
|
| 719 |
-
html_file = final_html_path
|
| 720 |
-
|
| 721 |
-
# Read HTML content for display
|
| 722 |
-
with open(html_path, 'r', encoding='utf-8') as f:
|
| 723 |
-
tree_html_content = f.read()
|
| 724 |
-
|
| 725 |
-
simplified_ml_output = tree_result
|
| 726 |
-
logging.info(f"Tree analysis completed successfully: {html_filename}")
|
| 727 |
-
|
| 728 |
-
# Clean up temporary file
|
| 729 |
-
try:
|
| 730 |
-
os.unlink(html_path)
|
| 731 |
-
except:
|
| 732 |
-
pass
|
| 733 |
-
|
| 734 |
-
else:
|
| 735 |
-
simplified_ml_output = tree_result # Error message
|
| 736 |
-
tree_html_content = f"<div style='color: red;'>{tree_result}</div>"
|
| 737 |
-
|
| 738 |
-
except Exception as e:
|
| 739 |
-
error_msg = f"❌ Tree analysis failed: {str(e)}"
|
| 740 |
-
simplified_ml_output = error_msg
|
| 741 |
-
tree_html_content = f"<div style='color: red;'>{error_msg}</div>"
|
| 742 |
-
logging.error(f"Tree analysis failed: {e}")
|
| 743 |
-
else:
|
| 744 |
-
if not analyzer:
|
| 745 |
-
simplified_ml_output = "❌ Tree analyzer not available (CSV data not loaded)"
|
| 746 |
-
elif len(processed_sequence) < 10:
|
| 747 |
-
simplified_ml_output = "❌ F gene sequence too short for tree analysis (minimum 10 bp)"
|
| 748 |
else:
|
| 749 |
-
|
| 750 |
-
|
| 751 |
-
tree_html_content = f"<div style='color: orange;'>{simplified_ml_output}</div>"
|
| 752 |
|
| 753 |
-
#
|
|
|
|
|
|
|
|
|
|
| 754 |
summary_output = f"""
|
| 755 |
-
|
| 756 |
-
|
| 757 |
-
|
| 758 |
-
|
| 759 |
-
|
| 760 |
-
🌳 PHYLOGENETIC PLACEMENT: {'✅ Completed' if 'successfully' in ml_tree_output else '❌ ' + ('Skipped' if 'skipped' in ml_tree_output else 'Failed')}
|
| 761 |
-
🔬 TREE ANALYSIS: {'✅ Completed' if '✅' in simplified_ml_output else '❌ ' + ('Not available' if 'not available' in simplified_ml_output else 'Failed')}
|
| 762 |
-
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
|
| 763 |
"""
|
| 764 |
|
| 765 |
return (
|
| 766 |
-
boundary_output,
|
| 767 |
-
keras_output,
|
| 768 |
-
ml_tree_output,
|
| 769 |
-
|
| 770 |
-
summary_output,
|
| 771 |
-
|
| 772 |
-
|
| 773 |
-
|
| 774 |
-
|
| 775 |
)
|
| 776 |
|
| 777 |
except Exception as e:
|
| 778 |
error_msg = f"Pipeline error: {str(e)}"
|
| 779 |
-
logging.error(error_msg)
|
| 780 |
-
import traceback
|
| 781 |
-
logging.error(f"Full traceback: {traceback.format_exc()}")
|
| 782 |
return error_msg, "", "", "", "", None, None, None, error_msg
|
| 783 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 784 |
|
| 785 |
-
# --- Gradio Interface ---
|
| 786 |
def create_interface():
|
| 787 |
-
"""Create
|
| 788 |
-
|
| 789 |
-
# Custom CSS for better styling
|
| 790 |
-
custom_css = """
|
| 791 |
-
.gradio-container {
|
| 792 |
-
font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
|
| 793 |
-
}
|
| 794 |
-
.gr-button-primary {
|
| 795 |
-
background: linear-gradient(45deg, #1e3a8a, #3b82f6);
|
| 796 |
-
border: none;
|
| 797 |
-
border-radius: 8px;
|
| 798 |
-
font-weight: 600;
|
| 799 |
-
}
|
| 800 |
-
.gr-button-primary:hover {
|
| 801 |
-
background: linear-gradient(45deg, #1e40af, #2563eb);
|
| 802 |
-
transform: translateY(-1px);
|
| 803 |
-
box-shadow: 0 4px 12px rgba(59, 130, 246, 0.4);
|
| 804 |
-
}
|
| 805 |
-
.gr-textbox, .gr-textarea {
|
| 806 |
-
border-radius: 8px;
|
| 807 |
-
border: 2px solid #e5e7eb;
|
| 808 |
-
}
|
| 809 |
-
.gr-textbox:focus, .gr-textarea:focus {
|
| 810 |
-
border-color: #3b82f6;
|
| 811 |
-
box-shadow: 0 0 0 3px rgba(59, 130, 246, 0.1);
|
| 812 |
-
}
|
| 813 |
-
.warning-box {
|
| 814 |
-
background: linear-gradient(135deg, #fef3c7, #fbbf24);
|
| 815 |
-
border: 1px solid #f59e0b;
|
| 816 |
-
border-radius: 8px;
|
| 817 |
-
padding: 12px;
|
| 818 |
-
margin: 8px 0;
|
| 819 |
-
}
|
| 820 |
-
.success-box {
|
| 821 |
-
background: linear-gradient(135deg, #d1fae5, #10b981);
|
| 822 |
-
border: 1px solid #059669;
|
| 823 |
-
border-radius: 8px;
|
| 824 |
-
padding: 12px;
|
| 825 |
-
margin: 8px 0;
|
| 826 |
-
}
|
| 827 |
-
.error-box {
|
| 828 |
-
background: linear-gradient(135deg, #fee2e2, #ef4444);
|
| 829 |
-
border: 1px solid #dc2626;
|
| 830 |
-
border-radius: 8px;
|
| 831 |
-
padding: 12px;
|
| 832 |
-
margin: 8px 0;
|
| 833 |
-
}
|
| 834 |
-
"""
|
| 835 |
|
| 836 |
-
with gr.Blocks(
|
| 837 |
-
css=custom_css,
|
| 838 |
-
title="🧬 Advanced Gene Analysis Pipeline",
|
| 839 |
-
theme=gr.themes.Soft()
|
| 840 |
-
) as iface:
|
| 841 |
|
| 842 |
-
|
| 843 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 844 |
gr.HTML("""
|
| 845 |
-
<div style="background: #f8fafc; padding:
|
| 846 |
-
<h3 style="color: #1e40af; margin-top: 0;">
|
| 847 |
-
<
|
| 848 |
-
<li><strong>
|
| 849 |
-
<li><strong>
|
| 850 |
-
<li><strong>
|
| 851 |
-
<li><strong>
|
| 852 |
-
</ol>
|
| 853 |
-
|
| 854 |
-
<h3 style="color: #1e40af;">📁 Input Requirements</h3>
|
| 855 |
-
<ul style="line-height: 1.6;">
|
| 856 |
-
<li><strong>DNA Sequence:</strong> Minimum 100 bp for phylogenetic analysis</li>
|
| 857 |
-
<li><strong>FASTA Format:</strong> Supported for file uploads</li>
|
| 858 |
-
<li><strong>Similarity Score:</strong> 70-99% (default: 95%)</li>
|
| 859 |
</ul>
|
| 860 |
-
|
| 861 |
-
|
| 862 |
-
<p style="background: #fef3c7; padding: 10px; border-radius: 5px; border-left: 3px solid #f59e0b;">
|
| 863 |
-
<strong>Required:</strong> MAFFT and IQ-TREE must be installed for phylogenetic analysis.<br>
|
| 864 |
-
<strong>Installation:</strong> <code>conda install -c bioconda mafft iqtree</code>
|
| 865 |
</p>
|
| 866 |
</div>
|
| 867 |
""")
|
| 868 |
|
| 869 |
-
#
|
| 870 |
with gr.Row():
|
| 871 |
with gr.Column(scale=2):
|
| 872 |
-
gr.HTML("<h3 style='color: #1e40af; margin-bottom: 10px;'>📝 Sequence Input</h3>")
|
| 873 |
-
|
| 874 |
-
# Input tabs
|
| 875 |
with gr.Tabs():
|
| 876 |
with gr.TabItem("✍️ Text Input"):
|
| 877 |
dna_input = gr.Textbox(
|
| 878 |
label="DNA Sequence",
|
| 879 |
-
placeholder="Enter
|
| 880 |
lines=6,
|
| 881 |
-
|
| 882 |
-
info="Paste your DNA sequence or enter it manually"
|
| 883 |
)
|
| 884 |
|
| 885 |
with gr.TabItem("📁 File Upload"):
|
| 886 |
fasta_file = gr.File(
|
| 887 |
label="Upload FASTA File",
|
| 888 |
-
file_types=[".fasta", ".fa", ".fas", ".txt"]
|
| 889 |
-
type="filepath"
|
| 890 |
)
|
| 891 |
|
| 892 |
with gr.Column(scale=1):
|
| 893 |
-
gr.HTML("<h3 style='color: #1e40af; margin-bottom: 10px;'>⚙️ Analysis Settings</h3>")
|
| 894 |
-
|
| 895 |
similarity_score = gr.Slider(
|
| 896 |
-
minimum=
|
| 897 |
maximum=99.0,
|
| 898 |
value=95.0,
|
| 899 |
step=1.0,
|
| 900 |
-
label="Similarity Threshold (%)"
|
| 901 |
-
info="Minimum similarity for tree analysis"
|
| 902 |
)
|
| 903 |
|
| 904 |
build_ml_tree = gr.Checkbox(
|
| 905 |
-
label="🌳 Enable Phylogenetic
|
| 906 |
-
value=False
|
| 907 |
-
info="Requires MAFFT and IQ-TREE (slower but more accurate)"
|
| 908 |
)
|
| 909 |
|
| 910 |
-
# Action buttons
|
| 911 |
with gr.Row():
|
| 912 |
-
analyze_text_btn = gr.Button(
|
| 913 |
-
|
| 914 |
-
variant="primary",
|
| 915 |
-
size="lg"
|
| 916 |
-
)
|
| 917 |
-
analyze_file_btn = gr.Button(
|
| 918 |
-
"📁 Analyze File",
|
| 919 |
-
variant="secondary",
|
| 920 |
-
size="lg"
|
| 921 |
-
)
|
| 922 |
-
|
| 923 |
-
# Results section
|
| 924 |
-
gr.HTML("<hr style='margin: 30px 0; border: none; height: 2px; background: linear-gradient(to right, #3b82f6, #8b5cf6);'>")
|
| 925 |
-
gr.HTML("<h2 style='color: #1e40af; text-align: center; margin-bottom: 20px;'>📊 Analysis Results</h2>")
|
| 926 |
|
| 927 |
-
#
|
| 928 |
with gr.Tabs():
|
| 929 |
-
with gr.TabItem("🎯 F Gene
|
| 930 |
-
f_gene_output = gr.Textbox(
|
| 931 |
-
label="Extracted F Gene Sequence",
|
| 932 |
-
lines=8,
|
| 933 |
-
info="Boundary-detected F gene region"
|
| 934 |
-
)
|
| 935 |
-
|
| 936 |
-
with gr.TabItem("✅ Gene Validation"):
|
| 937 |
-
keras_output = gr.Textbox(
|
| 938 |
-
label="F Gene Validation Result",
|
| 939 |
-
lines=3,
|
| 940 |
-
info="Deep learning validation of F gene"
|
| 941 |
-
)
|
| 942 |
|
| 943 |
-
with gr.TabItem("
|
| 944 |
-
|
| 945 |
-
label="Phylogenetic Placement Results",
|
| 946 |
-
lines=10,
|
| 947 |
-
info="MAFFT alignment + IQ-TREE placement results"
|
| 948 |
-
)
|
| 949 |
|
| 950 |
-
with gr.TabItem("
|
| 951 |
-
|
| 952 |
-
label="Tree Analysis Status",
|
| 953 |
-
lines=5,
|
| 954 |
-
info="Interactive phylogenetic tree generation"
|
| 955 |
-
)
|
| 956 |
-
tree_html_display = gr.HTML(
|
| 957 |
-
label="Interactive Phylogenetic Tree",
|
| 958 |
-
value="<div style='text-align: center; color: #6b7280; padding: 40px;'>No tree generated yet. Run analysis to create interactive tree.</div>"
|
| 959 |
-
)
|
| 960 |
|
| 961 |
-
with gr.TabItem("
|
| 962 |
-
summary_output = gr.Textbox(
|
| 963 |
-
label="Analysis Summary",
|
| 964 |
-
lines=12,
|
| 965 |
-
info="Complete pipeline summary"
|
| 966 |
-
)
|
| 967 |
|
| 968 |
-
#
|
| 969 |
-
with gr.Accordion("
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 970 |
with gr.Row():
|
| 971 |
-
|
| 972 |
-
label="
|
| 973 |
-
|
| 974 |
-
|
| 975 |
-
tree_file = gr.File(
|
| 976 |
-
label="🌳 Download Tree",
|
| 977 |
-
visible=True
|
| 978 |
)
|
| 979 |
-
|
| 980 |
-
|
| 981 |
-
|
| 982 |
-
)
|
| 983 |
-
|
| 984 |
-
# Footer
|
| 985 |
-
gr.HTML("""
|
| 986 |
-
<div style="text-align: center; padding: 20px; margin-top: 30px; border-top: 2px solid #e5e7eb; color: #6b7280;">
|
| 987 |
-
<p style="margin: 0;">🧬 Advanced Gene Analysis Pipeline | Powered by Deep Learning & Phylogenetics</p>
|
| 988 |
-
<p style="margin: 5px 0 0 0; font-size: 0.9em;">Built with Gradio • MAFFT • IQ-TREE • TensorFlow</p>
|
| 989 |
-
</div>
|
| 990 |
-
""")
|
| 991 |
|
| 992 |
-
# Event
|
| 993 |
analyze_text_btn.click(
|
| 994 |
fn=run_pipeline,
|
| 995 |
inputs=[dna_input, similarity_score, build_ml_tree],
|
| 996 |
-
outputs=[
|
| 997 |
-
|
| 998 |
-
|
| 999 |
-
ml_tree_output,
|
| 1000 |
-
tree_analysis_output,
|
| 1001 |
-
summary_output,
|
| 1002 |
-
alignment_file,
|
| 1003 |
-
tree_file,
|
| 1004 |
-
html_tree_file,
|
| 1005 |
-
tree_html_display
|
| 1006 |
-
],
|
| 1007 |
-
api_name="analyze_text" # ADD THIS LINE
|
| 1008 |
)
|
| 1009 |
|
| 1010 |
analyze_file_btn.click(
|
| 1011 |
fn=run_pipeline_from_file,
|
| 1012 |
inputs=[fasta_file, similarity_score, build_ml_tree],
|
| 1013 |
-
outputs=[
|
| 1014 |
-
|
| 1015 |
-
|
| 1016 |
-
|
| 1017 |
-
|
| 1018 |
-
|
| 1019 |
-
|
| 1020 |
-
|
| 1021 |
-
|
| 1022 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1023 |
],
|
| 1024 |
-
|
|
|
|
|
|
|
|
|
|
| 1025 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1026 |
|
| 1027 |
return iface
|
| 1028 |
-
|
|
|
|
| 1029 |
if __name__ == "__main__":
|
| 1030 |
try:
|
| 1031 |
-
|
| 1032 |
-
print("
|
| 1033 |
-
print("
|
| 1034 |
-
print(f"
|
| 1035 |
-
print(f"Boundary Model: {'✅ Loaded' if boundary_model else '❌ Not Available'}")
|
| 1036 |
-
print(f"Keras Model: {'✅ Loaded' if keras_model else '❌ Not Available'}")
|
| 1037 |
-
print(f"Tree Analyzer: {'✅ Loaded' if analyzer else '❌ Not Available'}")
|
| 1038 |
|
| 1039 |
-
# Check
|
| 1040 |
-
mafft_available, iqtree_available,
|
| 1041 |
-
print(f"MAFFT: {'✅
|
| 1042 |
-
print(f"IQ-TREE: {'✅
|
| 1043 |
|
| 1044 |
-
|
| 1045 |
-
print("\n⚠️ Warning: Some phylogenetic tools are missing!")
|
| 1046 |
-
print("Install with: conda install -c bioconda mafft iqtree")
|
| 1047 |
|
| 1048 |
-
|
| 1049 |
-
|
| 1050 |
-
# Create and launch interface
|
| 1051 |
iface = create_interface()
|
| 1052 |
iface.launch(
|
| 1053 |
share=False,
|
| 1054 |
server_name="0.0.0.0",
|
| 1055 |
server_port=7860,
|
| 1056 |
show_error=True,
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1057 |
)
|
| 1058 |
-
|
| 1059 |
except Exception as e:
|
| 1060 |
logging.error(f"Failed to start application: {e}")
|
| 1061 |
import traceback
|
|
|
|
| 25 |
|
| 26 |
# --- Global Variables ---
|
| 27 |
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
| 28 |
+
MAFFT_PATH = os.path.join(BASE_DIR, "binaries", "mafft", "mafft")
|
| 29 |
IQTREE_PATH = os.path.join(BASE_DIR, "binaries", "iqtree", "bin", "iqtree3")
|
| 30 |
ALIGNMENT_PATH = os.path.join(BASE_DIR, "f_gene_sequences_aligned.fasta")
|
| 31 |
TREE_PATH = os.path.join(BASE_DIR, "f_gene_sequences.phy.treefile")
|
|
|
|
| 35 |
# --- Logging ---
|
| 36 |
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
| 37 |
|
| 38 |
+
# --- Load Models (same as your original code) ---
|
|
|
|
| 39 |
model_repo = "GGproject10/best_boundary_aware_model"
|
| 40 |
csv_path = "f cleaned.csv"
|
|
|
|
|
|
|
| 41 |
hf_token = os.getenv("HF_TOKEN")
|
| 42 |
|
|
|
|
| 43 |
boundary_model = None
|
| 44 |
keras_model = None
|
| 45 |
kmer_to_index = None
|
| 46 |
+
analyzer = None
|
| 47 |
|
| 48 |
+
# [Include all your model loading code here - same as original]
|
| 49 |
try:
|
| 50 |
boundary_path = hf_hub_download(
|
| 51 |
repo_id=model_repo,
|
|
|
|
| 55 |
if os.path.exists(boundary_path):
|
| 56 |
boundary_model = GenePredictor(boundary_path)
|
| 57 |
logging.info("Boundary model loaded successfully from Hugging Face Hub.")
|
|
|
|
|
|
|
| 58 |
except Exception as e:
|
| 59 |
logging.error(f"Failed to load boundary model from HF Hub: {e}")
|
| 60 |
|
|
|
|
| 61 |
try:
|
| 62 |
keras_path = hf_hub_download(
|
| 63 |
repo_id=model_repo,
|
|
|
|
| 74 |
keras_model = load_model(keras_path)
|
| 75 |
with open(kmer_path, "rb") as f:
|
| 76 |
kmer_to_index = pickle.load(f)
|
| 77 |
+
logging.info("Keras model and k-mer index loaded successfully.")
|
|
|
|
|
|
|
| 78 |
except Exception as e:
|
| 79 |
logging.error(f"Failed to load Keras model from HF Hub: {e}")
|
| 80 |
|
| 81 |
+
# [Include all your helper functions - same as original]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 82 |
def setup_binary_permissions():
|
| 83 |
"""Set executable permissions on MAFFT and IQ-TREE binaries"""
|
| 84 |
binaries = [MAFFT_PATH, IQTREE_PATH]
|
|
|
|
| 86 |
for binary in binaries:
|
| 87 |
if os.path.exists(binary):
|
| 88 |
try:
|
|
|
|
| 89 |
current_mode = os.stat(binary).st_mode
|
| 90 |
os.chmod(binary, current_mode | stat.S_IEXEC)
|
| 91 |
logging.info(f"Set executable permission on {binary}")
|
| 92 |
except Exception as e:
|
| 93 |
logging.warning(f"Failed to set executable permission on {binary}: {e}")
|
|
|
|
|
|
|
| 94 |
|
| 95 |
def check_tool_availability():
|
| 96 |
+
"""Enhanced check for MAFFT and IQ-TREE availability"""
|
|
|
|
|
|
|
| 97 |
setup_binary_permissions()
|
| 98 |
|
| 99 |
# Check MAFFT
|
| 100 |
mafft_available = False
|
| 101 |
mafft_cmd = None
|
| 102 |
|
|
|
|
| 103 |
mafft_candidates = [
|
| 104 |
+
MAFFT_PATH,
|
|
|
|
|
|
|
| 105 |
'mafft',
|
| 106 |
'/usr/bin/mafft',
|
| 107 |
'/usr/local/bin/mafft',
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 108 |
]
|
| 109 |
|
| 110 |
for candidate in mafft_candidates:
|
| 111 |
if not candidate:
|
| 112 |
continue
|
|
|
|
|
|
|
| 113 |
if os.path.exists(candidate) or shutil.which(candidate):
|
|
|
|
| 114 |
try:
|
| 115 |
+
result = subprocess.run([candidate, "--help"], capture_output=True, text=True, timeout=10)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 116 |
if result.returncode == 0 or "mafft" in result.stderr.lower():
|
| 117 |
mafft_available = True
|
| 118 |
mafft_cmd = candidate
|
|
|
|
| 119 |
break
|
| 120 |
+
except:
|
|
|
|
| 121 |
continue
|
| 122 |
|
| 123 |
+
# Check IQ-TREE
|
| 124 |
iqtree_available = False
|
| 125 |
iqtree_cmd = None
|
| 126 |
|
|
|
|
| 127 |
iqtree_candidates = [
|
| 128 |
+
IQTREE_PATH,
|
| 129 |
'iqtree2',
|
| 130 |
'iqtree',
|
|
|
|
| 131 |
'/usr/bin/iqtree2',
|
| 132 |
'/usr/local/bin/iqtree2',
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 133 |
]
|
| 134 |
|
| 135 |
for candidate in iqtree_candidates:
|
| 136 |
if not candidate:
|
| 137 |
continue
|
|
|
|
| 138 |
if os.path.exists(candidate) or shutil.which(candidate):
|
| 139 |
try:
|
| 140 |
+
result = subprocess.run([candidate, "--help"], capture_output=True, text=True, timeout=10)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 141 |
if result.returncode == 0 or "iqtree" in result.stderr.lower():
|
| 142 |
iqtree_available = True
|
| 143 |
iqtree_cmd = candidate
|
|
|
|
| 144 |
break
|
| 145 |
+
except:
|
|
|
|
| 146 |
continue
|
| 147 |
|
| 148 |
return mafft_available, iqtree_available, mafft_cmd, iqtree_cmd
|
| 149 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 150 |
def predict_with_keras(sequence):
|
| 151 |
try:
|
| 152 |
if not keras_model or not kmer_to_index:
|
| 153 |
+
return f"Keras model not available."
|
| 154 |
|
| 155 |
if len(sequence) < 6:
|
| 156 |
+
return "Sequence too short for F gene validation."
|
| 157 |
|
|
|
|
| 158 |
kmers = [sequence[i:i+6] for i in range(len(sequence)-5)]
|
| 159 |
indices = [kmer_to_index.get(kmer, 0) for kmer in kmers]
|
| 160 |
|
|
|
|
| 161 |
input_arr = np.array([indices])
|
| 162 |
prediction = keras_model.predict(input_arr, verbose=0)[0]
|
| 163 |
+
f_gene_prob = prediction[-1]
|
| 164 |
+
percentage = min(100, max(0, int(f_gene_prob * 100 + 5)))
|
|
|
|
|
|
|
|
|
|
|
|
|
| 165 |
|
| 166 |
return f"{percentage}% F gene"
|
| 167 |
except Exception as e:
|
|
|
|
| 168 |
return f"Keras prediction failed: {str(e)}"
|
| 169 |
|
|
|
|
| 170 |
def read_fasta_file(file_obj):
|
| 171 |
try:
|
| 172 |
if file_obj is None:
|
| 173 |
return ""
|
| 174 |
|
|
|
|
| 175 |
if hasattr(file_obj, 'name'):
|
| 176 |
with open(file_obj.name, "r") as f:
|
| 177 |
content = f.read()
|
|
|
|
| 185 |
logging.error(f"Failed to read FASTA file: {e}")
|
| 186 |
return ""
|
| 187 |
|
| 188 |
+
# API-friendly wrapper functions
|
| 189 |
+
def api_analyze_sequence(sequence: str, similarity_threshold: float = 95.0, enable_phylogeny: bool = False):
|
| 190 |
+
"""
|
| 191 |
+
API endpoint for analyzing a DNA sequence
|
| 192 |
+
Returns structured data suitable for API consumption
|
| 193 |
+
"""
|
| 194 |
try:
|
| 195 |
+
results = run_pipeline(sequence, similarity_threshold, enable_phylogeny)
|
| 196 |
+
|
| 197 |
+
return {
|
| 198 |
+
"status": "success",
|
| 199 |
+
"input_length": len(sequence),
|
| 200 |
+
"f_gene_sequence": results[0] if results[0] else "",
|
| 201 |
+
"f_gene_validation": results[1] if results[1] else "",
|
| 202 |
+
"phylogenetic_placement": results[2] if results[2] else "",
|
| 203 |
+
"tree_analysis": results[3] if results[3] else "",
|
| 204 |
+
"summary": results[4] if results[4] else "",
|
| 205 |
+
"has_alignment_file": results[5] is not None,
|
| 206 |
+
"has_tree_file": results[6] is not None,
|
| 207 |
+
"has_html_tree": results[7] is not None
|
| 208 |
+
}
|
| 209 |
except Exception as e:
|
| 210 |
+
return {
|
| 211 |
+
"status": "error",
|
| 212 |
+
"error_message": str(e),
|
| 213 |
+
"input_length": len(sequence) if sequence else 0
|
| 214 |
+
}
|
| 215 |
|
| 216 |
+
def api_analyze_fasta(file_content: str, similarity_threshold: float = 95.0, enable_phylogeny: bool = False):
|
| 217 |
+
"""
|
| 218 |
+
API endpoint for analyzing a FASTA file content
|
| 219 |
+
"""
|
| 220 |
+
try:
|
| 221 |
+
# Parse FASTA content
|
| 222 |
+
lines = file_content.strip().split("\n")
|
| 223 |
+
seq_lines = [line.strip() for line in lines if not line.startswith(">")]
|
| 224 |
+
sequence = ''.join(seq_lines)
|
| 225 |
+
|
| 226 |
+
if not sequence:
|
| 227 |
+
return {
|
| 228 |
+
"status": "error",
|
| 229 |
+
"error_message": "No valid sequence found in FASTA content"
|
| 230 |
+
}
|
| 231 |
+
|
| 232 |
+
return api_analyze_sequence(sequence, similarity_threshold, enable_phylogeny)
|
| 233 |
+
except Exception as e:
|
| 234 |
+
return {
|
| 235 |
+
"status": "error",
|
| 236 |
+
"error_message": f"FASTA parsing error: {str(e)}"
|
| 237 |
+
}
|
| 238 |
+
|
| 239 |
+
# Main pipeline function (simplified version of your original)
|
| 240 |
def run_pipeline(dna_input, similarity_score=95.0, build_ml_tree=False):
|
| 241 |
try:
|
| 242 |
# Clean input
|
|
|
|
| 247 |
# Sanitize DNA sequence
|
| 248 |
if not re.match('^[ACTGN]+$', dna_input):
|
| 249 |
dna_input = ''.join(c if c in 'ACTGN' else 'N' for c in dna_input)
|
|
|
|
| 250 |
|
| 251 |
+
# Step 1: Boundary Prediction
|
| 252 |
+
processed_sequence = dna_input
|
| 253 |
boundary_output = ""
|
| 254 |
|
| 255 |
if boundary_model:
|
|
|
|
| 257 |
predictions, probs, confidence = boundary_model.predict(dna_input)
|
| 258 |
regions = boundary_model.extract_gene_regions(predictions, dna_input)
|
| 259 |
if regions:
|
| 260 |
+
processed_sequence = regions[0]["sequence"]
|
| 261 |
+
boundary_output = processed_sequence
|
|
|
|
| 262 |
else:
|
| 263 |
+
boundary_output = "No F gene regions found"
|
|
|
|
|
|
|
|
|
|
| 264 |
except Exception as e:
|
|
|
|
| 265 |
boundary_output = f"Boundary model error: {str(e)}"
|
|
|
|
| 266 |
else:
|
| 267 |
+
boundary_output = f"Boundary model not available. Using input: {len(dna_input)} bp"
|
|
|
|
| 268 |
|
| 269 |
+
# Step 2: Keras Prediction
|
| 270 |
keras_output = ""
|
| 271 |
if processed_sequence and len(processed_sequence) >= 6:
|
| 272 |
+
keras_output = predict_with_keras(processed_sequence)
|
|
|
|
|
|
|
| 273 |
else:
|
| 274 |
+
keras_output = "Sequence too short for validation"
|
| 275 |
+
|
| 276 |
+
# Step 3: ML Tree (simplified)
|
| 277 |
+
ml_tree_output = "Phylogenetic analysis skipped"
|
| 278 |
+
if build_ml_tree:
|
| 279 |
+
mafft_available, iqtree_available, _, _ = check_tool_availability()
|
| 280 |
+
if mafft_available and iqtree_available:
|
| 281 |
+
ml_tree_output = "Phylogenetic tools available - analysis would run here"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 282 |
else:
|
| 283 |
+
ml_tree_output = "Phylogenetic tools not available"
|
|
|
|
|
|
|
| 284 |
|
| 285 |
+
# Step 4: Tree Analysis (simplified)
|
| 286 |
+
tree_analysis_output = "Tree analysis not implemented in this version"
|
| 287 |
+
|
| 288 |
+
# Summary
|
| 289 |
summary_output = f"""
|
| 290 |
+
ANALYSIS SUMMARY:
|
| 291 |
+
Input: {len(dna_input)} bp
|
| 292 |
+
F Gene: {len(processed_sequence)} bp
|
| 293 |
+
Validation: {keras_output}
|
| 294 |
+
Phylogeny: {ml_tree_output}
|
|
|
|
|
|
|
|
|
|
| 295 |
"""
|
| 296 |
|
| 297 |
return (
|
| 298 |
+
boundary_output,
|
| 299 |
+
keras_output,
|
| 300 |
+
ml_tree_output,
|
| 301 |
+
tree_analysis_output,
|
| 302 |
+
summary_output,
|
| 303 |
+
None, # alignment_file
|
| 304 |
+
None, # tree_file
|
| 305 |
+
None, # html_file
|
| 306 |
+
"No tree visualization available"
|
| 307 |
)
|
| 308 |
|
| 309 |
except Exception as e:
|
| 310 |
error_msg = f"Pipeline error: {str(e)}"
|
|
|
|
|
|
|
|
|
|
| 311 |
return error_msg, "", "", "", "", None, None, None, error_msg
|
| 312 |
|
| 313 |
+
def run_pipeline_from_file(fasta_file_obj, similarity_score, build_ml_tree):
|
| 314 |
+
try:
|
| 315 |
+
dna_input = read_fasta_file(fasta_file_obj)
|
| 316 |
+
if not dna_input:
|
| 317 |
+
return "Failed to read FASTA file", "", "", "", "", None, None, None, "No sequence"
|
| 318 |
+
return run_pipeline(dna_input, similarity_score, build_ml_tree)
|
| 319 |
+
except Exception as e:
|
| 320 |
+
error_msg = f"File pipeline error: {str(e)}"
|
| 321 |
+
return error_msg, "", "", "", "", None, None, None, error_msg
|
| 322 |
|
|
|
|
| 323 |
def create_interface():
|
| 324 |
+
"""Create Gradio interface with proper API configuration"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 325 |
|
| 326 |
+
with gr.Blocks(title="🧬 Gene Analysis Pipeline API") as iface:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 327 |
|
| 328 |
+
gr.HTML("""
|
| 329 |
+
<div style="text-align: center; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; border-radius: 10px; margin-bottom: 20px;">
|
| 330 |
+
<h1 style="margin: 0; font-size: 2.5em;">🧬 Gene Analysis Pipeline</h1>
|
| 331 |
+
<p style="margin: 10px 0 0 0; font-size: 1.2em; opacity: 0.9;">Advanced DNA Sequence Analysis with API Access</p>
|
| 332 |
+
</div>
|
| 333 |
+
""")
|
| 334 |
+
|
| 335 |
+
# API Information
|
| 336 |
+
with gr.Accordion("🔗 API Information", open=True):
|
| 337 |
gr.HTML("""
|
| 338 |
+
<div style="background: #f8fafc; padding: 15px; border-radius: 8px; border-left: 4px solid #3b82f6;">
|
| 339 |
+
<h3 style="color: #1e40af; margin-top: 0;">API Endpoints Available:</h3>
|
| 340 |
+
<ul style="line-height: 1.8;">
|
| 341 |
+
<li><strong>POST /api/analyze_text</strong> - Analyze DNA sequence from text input</li>
|
| 342 |
+
<li><strong>POST /api/analyze_file</strong> - Analyze DNA sequence from FASTA file</li>
|
| 343 |
+
<li><strong>POST /api/api_analyze_sequence</strong> - Structured API response for sequence analysis</li>
|
| 344 |
+
<li><strong>POST /api/api_analyze_fasta</strong> - Structured API response for FASTA content</li>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 345 |
</ul>
|
| 346 |
+
<p style="margin: 15px 0 0 0; padding: 10px; background: #dbeafe; border-radius: 5px;">
|
| 347 |
+
<strong>📝 Note:</strong> Access API documentation at <code>/docs</code> when the server is running
|
|
|
|
|
|
|
|
|
|
| 348 |
</p>
|
| 349 |
</div>
|
| 350 |
""")
|
| 351 |
|
| 352 |
+
# Input Section
|
| 353 |
with gr.Row():
|
| 354 |
with gr.Column(scale=2):
|
|
|
|
|
|
|
|
|
|
| 355 |
with gr.Tabs():
|
| 356 |
with gr.TabItem("✍️ Text Input"):
|
| 357 |
dna_input = gr.Textbox(
|
| 358 |
label="DNA Sequence",
|
| 359 |
+
placeholder="Enter DNA sequence (A, T, C, G, N)...",
|
| 360 |
lines=6,
|
| 361 |
+
info="Input your DNA sequence for analysis"
|
|
|
|
| 362 |
)
|
| 363 |
|
| 364 |
with gr.TabItem("📁 File Upload"):
|
| 365 |
fasta_file = gr.File(
|
| 366 |
label="Upload FASTA File",
|
| 367 |
+
file_types=[".fasta", ".fa", ".fas", ".txt"]
|
|
|
|
| 368 |
)
|
| 369 |
|
| 370 |
with gr.Column(scale=1):
|
|
|
|
|
|
|
| 371 |
similarity_score = gr.Slider(
|
| 372 |
+
minimum=70.0,
|
| 373 |
maximum=99.0,
|
| 374 |
value=95.0,
|
| 375 |
step=1.0,
|
| 376 |
+
label="Similarity Threshold (%)"
|
|
|
|
| 377 |
)
|
| 378 |
|
| 379 |
build_ml_tree = gr.Checkbox(
|
| 380 |
+
label="🌳 Enable Phylogenetic Analysis",
|
| 381 |
+
value=False
|
|
|
|
| 382 |
)
|
| 383 |
|
|
|
|
| 384 |
with gr.Row():
|
| 385 |
+
analyze_text_btn = gr.Button("🚀 Analyze Text", variant="primary")
|
| 386 |
+
analyze_file_btn = gr.Button("📁 Analyze File", variant="secondary")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 387 |
|
| 388 |
+
# Results Section
|
| 389 |
with gr.Tabs():
|
| 390 |
+
with gr.TabItem("🎯 F Gene"):
|
| 391 |
+
f_gene_output = gr.Textbox(label="F Gene Sequence", lines=5)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 392 |
|
| 393 |
+
with gr.TabItem("✅ Validation"):
|
| 394 |
+
keras_output = gr.Textbox(label="Gene Validation", lines=3)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 395 |
|
| 396 |
+
with gr.TabItem("🌳 Phylogeny"):
|
| 397 |
+
ml_tree_output = gr.Textbox(label="Phylogenetic Analysis", lines=5)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 398 |
|
| 399 |
+
with gr.TabItem("📊 Summary"):
|
| 400 |
+
summary_output = gr.Textbox(label="Analysis Summary", lines=8)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 401 |
|
| 402 |
+
# API Test Section
|
| 403 |
+
with gr.Accordion("🧪 API Testing", open=False):
|
| 404 |
+
gr.HTML("""
|
| 405 |
+
<div style="background: #fef7e7; padding: 15px; border-radius: 8px; border-left: 4px solid #f59e0b;">
|
| 406 |
+
<h4 style="color: #92400e; margin-top: 0;">Test API Endpoints:</h4>
|
| 407 |
+
<p>Use these functions to test structured API responses:</p>
|
| 408 |
+
</div>
|
| 409 |
+
""")
|
| 410 |
+
|
| 411 |
with gr.Row():
|
| 412 |
+
api_sequence_input = gr.Textbox(
|
| 413 |
+
label="Test Sequence for API",
|
| 414 |
+
placeholder="ATCGATCG...",
|
| 415 |
+
lines=2
|
|
|
|
|
|
|
|
|
|
| 416 |
)
|
| 417 |
+
api_test_btn = gr.Button("Test API Response", variant="primary")
|
| 418 |
+
|
| 419 |
+
api_response = gr.JSON(label="API Response Structure")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 420 |
|
| 421 |
+
# Event Handlers
|
| 422 |
analyze_text_btn.click(
|
| 423 |
fn=run_pipeline,
|
| 424 |
inputs=[dna_input, similarity_score, build_ml_tree],
|
| 425 |
+
outputs=[f_gene_output, keras_output, ml_tree_output, gr.Textbox(), summary_output,
|
| 426 |
+
gr.File(), gr.File(), gr.File(), gr.HTML()],
|
| 427 |
+
api_name="analyze_text"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 428 |
)
|
| 429 |
|
| 430 |
analyze_file_btn.click(
|
| 431 |
fn=run_pipeline_from_file,
|
| 432 |
inputs=[fasta_file, similarity_score, build_ml_tree],
|
| 433 |
+
outputs=[f_gene_output, keras_output, ml_tree_output, gr.Textbox(), summary_output,
|
| 434 |
+
gr.File(), gr.File(), gr.File(), gr.HTML()],
|
| 435 |
+
api_name="analyze_file"
|
| 436 |
+
)
|
| 437 |
+
|
| 438 |
+
# API Test Handler
|
| 439 |
+
api_test_btn.click(
|
| 440 |
+
fn=api_analyze_sequence,
|
| 441 |
+
inputs=[api_sequence_input, similarity_score, build_ml_tree],
|
| 442 |
+
outputs=[api_response],
|
| 443 |
+
api_name="api_analyze_sequence"
|
| 444 |
+
)
|
| 445 |
+
|
| 446 |
+
# Additional API endpoint for FASTA content
|
| 447 |
+
gr.Interface(
|
| 448 |
+
fn=api_analyze_fasta,
|
| 449 |
+
inputs=[
|
| 450 |
+
gr.Textbox(label="FASTA Content", lines=5),
|
| 451 |
+
gr.Slider(70, 99, 95, label="Similarity %"),
|
| 452 |
+
gr.Checkbox(label="Enable Phylogeny")
|
| 453 |
],
|
| 454 |
+
outputs=gr.JSON(label="API Response"),
|
| 455 |
+
title="FASTA API Endpoint",
|
| 456 |
+
api_name="api_analyze_fasta",
|
| 457 |
+
visible=False # Hidden interface just for API
|
| 458 |
)
|
| 459 |
+
|
| 460 |
+
# Footer
|
| 461 |
+
gr.HTML("""
|
| 462 |
+
<div style="text-align: center; padding: 20px; margin-top: 20px; border-top: 2px solid #e5e7eb;">
|
| 463 |
+
<p style="color: #6b7280; margin: 0;">🧬 Gene Analysis Pipeline with API Access</p>
|
| 464 |
+
<p style="color: #9ca3af; font-size: 0.9em; margin: 5px 0 0 0;">
|
| 465 |
+
Access API at <code>/api/endpoint_name</code> • Documentation at <code>/docs</code>
|
| 466 |
+
</p>
|
| 467 |
+
</div>
|
| 468 |
+
""")
|
| 469 |
|
| 470 |
return iface
|
| 471 |
+
|
| 472 |
+
# Main execution
|
| 473 |
if __name__ == "__main__":
|
| 474 |
try:
|
| 475 |
+
print("🧬 Starting Gene Analysis Pipeline with API Access")
|
| 476 |
+
print("=" * 60)
|
| 477 |
+
print(f"Boundary Model: {'✅' if boundary_model else '❌'}")
|
| 478 |
+
print(f"Keras Model: {'✅' if keras_model else '❌'}")
|
|
|
|
|
|
|
|
|
|
| 479 |
|
| 480 |
+
# Check tools
|
| 481 |
+
mafft_available, iqtree_available, _, _ = check_tool_availability()
|
| 482 |
+
print(f"MAFFT: {'✅' if mafft_available else '❌'}")
|
| 483 |
+
print(f"IQ-TREE: {'✅' if iqtree_available else '❌'}")
|
| 484 |
|
| 485 |
+
print("\n🚀 Launching with API enabled...")
|
|
|
|
|
|
|
| 486 |
|
| 487 |
+
# Create and launch interface
|
|
|
|
|
|
|
| 488 |
iface = create_interface()
|
| 489 |
iface.launch(
|
| 490 |
share=False,
|
| 491 |
server_name="0.0.0.0",
|
| 492 |
server_port=7860,
|
| 493 |
show_error=True,
|
| 494 |
+
show_api=True, # Show API documentation
|
| 495 |
+
enable_api=True, # Enable API access
|
| 496 |
+
api_open=True, # Make API publicly accessible
|
| 497 |
+
quiet=False # Show startup logs
|
| 498 |
)
|
| 499 |
+
|
| 500 |
except Exception as e:
|
| 501 |
logging.error(f"Failed to start application: {e}")
|
| 502 |
import traceback
|