Update app.py

app.py

@@ -105,7 +105,7 @@ def create_heatmap(contact_matrix, rna_labels, protein_labels, rna_name, protein
     # Apply Threshold threshold
     filtered_matrix = contact_matrix.copy()
     filtered_matrix[filtered_matrix < Threshold] = 0
-
+    ratio = filtered_matrix.shape[1] / filtered_matrix.shape[0]
     fig = go.Figure(data=go.Heatmap(
         z=filtered_matrix,
         x=protein_labels,
@@ -123,13 +123,12 @@ def create_heatmap(contact_matrix, rna_labels, protein_labels, rna_name, protein
             'xanchor': 'center',
             'yanchor': 'top'
         },
-        xaxis_title=f"Protein Residues ({protein_name})",
-        yaxis_title=f"RNA Nucleotides ({rna_name})",
-        width=800,
-        height=600,
+        xaxis_title=f"Residues ({protein_name}, {len(protein_labels)} AA)",
+        yaxis_title=f"Nucleotides ({rna_name}, {len(rna_labels)} nt)",
+        width=max(ratio*300,600),
+        height=300,
         font=dict(size=12)
     )
-
     return fig
 
 
@@ -322,10 +321,21 @@ def create_interface():
         margin-bottom: 8px;
         color: #4a5568;
     }
+    .scrollable-plot {
+        max-width: 100%;
+        overflow-x: auto;    /* 水平滚动条 */
+        overflow-y: auto;    /* 垂直滚动条 */
+        border: 1px solid #ddd;
+        border-radius: 4px;
+    }
+    
+    .scrollable-plot > div {
+        min-width: max-content;  /* 保持原始宽度 */
+    }
     """
 
     with gr.Blocks(title="RNA-Protein Contact Prediction Tool",
-                   theme=gr.themes.Soft(primary_hue="blue", secondary_hue="teal")) as app:
+                   theme=gr.themes.Soft(primary_hue="blue", secondary_hue="teal"),css=None) as app:
         gr.Markdown("""
    <center>
    
@@ -346,28 +356,29 @@ def create_interface():
                     with gr.Group(elem_classes="input-group"):
                         input_method = gr.Radio(
                             choices=["Upload FASTA File", "Enter Sequences Directly"],
-                            value="Upload FASTA File",
+                            value="Enter Sequences Directly",
                             label="Input Method"
                         )
 
                         fasta_input = gr.File(
                             label="FASTA File",
                             file_types=['.fasta', '.fa', '.txt'],
-                            type='binary'
+                            type='binary',
+                            visible = False
                         )
 
                         rna_input = gr.Textbox(
                             label="RNA Sequence",
                             placeholder="Enter RNA sequence (use A,C,G,U)",
                             lines=3,
-                            visible=False
+                            visible=True
                         )
 
                         protein_input = gr.Textbox(
                             label="Protein Sequence",
                             placeholder="Enter protein sequence (standard amino acid codes)",
                             lines=3,
-                            visible=False
+                            visible=True
                         )
 
                     # Example data
@@ -415,7 +426,6 @@ def create_interface():
                     """)
                     # Heatmap display
                     heatmap_plot = gr.Plot(label='Contact Map')
-
                     # Contact pairs table with info header
                     gr.Markdown("### 🎯Contact Pairs")
                     contact_info = gr.Markdown("", elem_classes="contact-info")