Update app.py

app.py

@@ -33,7 +33,22 @@ model_promoter.eval()
 
 # D. Multi-Feature LightGBM 
 lgbm_path = hf_hub_download(repo_id="Geonomic/Genomic-Oracle-Weights", filename="dnabert_lightgbm_model_feature_type.pkl")
-lightgbm_model = joblib.load(lgbm_path)
+raw_lgbm = joblib.load(lgbm_path)
+
+# If it's a dictionary, print the keys to the log and try to extract the model
+if isinstance(raw_lgbm, dict):
+    print(f"💡 DEBUG: LightGBM Dictionary Keys: {raw_lgbm.keys()}")
+    # We will try the most common names for saved models
+    if "model" in raw_lgbm:
+        lightgbm_model = raw_lgbm["model"]
+    elif "classifier" in raw_lgbm:
+        lightgbm_model = raw_lgbm["classifier"]
+    else:
+        # Fallback: just grab the very first thing in the dictionary
+        first_key = list(raw_lgbm.keys())[0]
+        lightgbm_model = raw_lgbm[first_key]
+else:
+    lightgbm_model = raw_lgbm
 
 # E. Custom Lean/Obese Phenotype BERT (🚨 Forced Native Architecture via Colab Fix!)
 tokenizer_pheno = BertTokenizer.from_pretrained("Geonomic/Genomic-Oracle-Weights", do_lower_case=False)
@@ -231,14 +246,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="🧬 The Genomic Oracle 🧬") as
     
     with gr.Row():
         with gr.Column(scale=1):
-            dna_input = gr.Textbox(label="Enter DNA Sequence", placeholder="e.g., ATGCGATCGATCGATCG...", lines=6)
+            dna_input = gr.Textbox(label="Enter DNA Sequence", placeholder="e.g., ATGCGATCGATCGATCG...", lines=15)
             run_mapping_cb = gr.Checkbox(value=False, label="Query NCBI BLAST for spatial mapping (Takes 1–3 mins)")
             submit_btn = gr.Button("🚀 Initialize Deep Scan", variant="primary")
 
         with gr.Column(scale=2):
             output_summary = gr.Textbox(label="✅ Classification Summary", lines=4)
             stats_panel = gr.Textbox(label="📊 Internal Pipeline Statistics", lines=4)
-            mapping_section = gr.Accordion("📝 Genomic Context (BLAST/Ensembl)", open=False)
+            mapping_section = gr.Accordion("🌐 Genomic Context (BLAST/Ensembl)", open=False)
             with mapping_section:
                 context_output = gr.Textbox(label="Mapping Results", lines=5, placeholder="Results will appear here...")