Update app.py

app.py

@@ -8,6 +8,8 @@ import matplotlib.pyplot as plt
 import matplotlib.colors as mcolors
 import io
 from PIL import Image
+from scipy.interpolate import interp1d
+import numpy as np
 
 ###############################################################################
 # 1. MODEL DEFINITION
@@ -474,9 +476,6 @@ def analyze_subregion(state, header, region_start, region_end):
 
     return (region_info, heatmap_img, hist_img)
 
-# Add these imports at the top of the file, after existing imports
-from scipy.interpolate import interp1d
-import numpy as np
 
 ###############################################################################
 # NEW SECTION: COMPARATIVE ANALYSIS FUNCTIONS
@@ -638,66 +637,6 @@ Negative values (blue) indicate regions where Sequence 1 is more "human-like"
     
     return comparison_text, heatmap_img, hist_img
 
-###############################################################################
-# NEW TAB TO GRADIO 
-###############################################################################
-
-# Inside the Gradio interface definition, add this new tab:
-    with gr.Tab("3) Comparative Analysis"):
-        gr.Markdown("""
-        **Compare Two Sequences**  
-        Upload or paste two FASTA sequences to compare their SHAP patterns.
-        The sequences will be normalized to the same length for comparison.
-        
-        **Color Scale**:  
-        - Red: Sequence 2 is more human-like in this region
-        - Blue: Sequence 1 is more human-like in this region
-        - White: No substantial difference
-        """)
-        
-        with gr.Row():
-            with gr.Column(scale=1):
-                file_input1 = gr.File(
-                    label="Upload first FASTA file",
-                    file_types=[".fasta", ".fa", ".txt"],
-                    type="filepath"
-                )
-                text_input1 = gr.Textbox(
-                    label="Or paste first FASTA sequence",
-                    placeholder=">sequence1\nACGTACGT...",
-                    lines=5
-                )
-            
-            with gr.Column(scale=1):
-                file_input2 = gr.File(
-                    label="Upload second FASTA file",
-                    file_types=[".fasta", ".fa", ".txt"],
-                    type="filepath"
-                )
-                text_input2 = gr.Textbox(
-                    label="Or paste second FASTA sequence",
-                    placeholder=">sequence2\nACGTACGT...",
-                    lines=5
-                )
-        
-        compare_btn = gr.Button("Compare Sequences", variant="primary")
-        
-        comparison_text = gr.Textbox(
-            label="Comparison Results",
-            lines=12,
-            interactive=False
-        )
-        
-        with gr.Row():
-            diff_heatmap = gr.Image(label="SHAP Difference Heatmap")
-            diff_hist = gr.Image(label="Distribution of SHAP Differences")
-        
-        compare_btn.click(
-            analyze_sequence_comparison,
-            inputs=[file_input1, file_input2, text_input1, text_input2],
-            outputs=[comparison_text, diff_heatmap, diff_hist]
-        )
-        
 ###############################################################################
 # 9. BUILD GRADIO INTERFACE
 ###############################################################################
@@ -788,7 +727,61 @@ with gr.Blocks(css=css) as iface:
             inputs=[seq_state, header_state, region_start, region_end],
             outputs=[subregion_info, subregion_img, subregion_hist_img]
         )
-    
+
+    with gr.Tab("3) Comparative Analysis"):
+        gr.Markdown("""
+        **Compare Two Sequences**  
+        Upload or paste two FASTA sequences to compare their SHAP patterns.
+        The sequences will be normalized to the same length for comparison.
+        
+        **Color Scale**:  
+        - Red: Sequence 2 is more human-like in this region
+        - Blue: Sequence 1 is more human-like in this region
+        - White: No substantial difference
+        """)
+        
+        with gr.Row():
+            with gr.Column(scale=1):
+                file_input1 = gr.File(
+                    label="Upload first FASTA file",
+                    file_types=[".fasta", ".fa", ".txt"],
+                    type="filepath"
+                )
+                text_input1 = gr.Textbox(
+                    label="Or paste first FASTA sequence",
+                    placeholder=">sequence1\nACGTACGT...",
+                    lines=5
+                )
+            
+            with gr.Column(scale=1):
+                file_input2 = gr.File(
+                    label="Upload second FASTA file",
+                    file_types=[".fasta", ".fa", ".txt"],
+                    type="filepath"
+                )
+                text_input2 = gr.Textbox(
+                    label="Or paste second FASTA sequence",
+                    placeholder=">sequence2\nACGTACGT...",
+                    lines=5
+                )
+        
+        compare_btn = gr.Button("Compare Sequences", variant="primary")
+        
+        comparison_text = gr.Textbox(
+            label="Comparison Results",
+            lines=12,
+            interactive=False
+        )
+        
+        with gr.Row():
+            diff_heatmap = gr.Image(label="SHAP Difference Heatmap")
+            diff_hist = gr.Image(label="Distribution of SHAP Differences")
+        
+        compare_btn.click(
+            analyze_sequence_comparison,
+            inputs=[file_input1, file_input2, text_input1, text_input2],
+            outputs=[comparison_text, diff_heatmap, diff_hist]
+        )   
     gr.Markdown("""
     ### Interface Features
     - **Overall Classification** (human vs non-human) using k-mer frequencies.