Update app.py

app.py

@@ -65,26 +65,39 @@ def generate_heatmap(protein_sequence, start_pos=1, end_pos=None):
     plt.close()
     return temp_file
 
-def heatmap_interface(sequence, start, end):
-    # Ensure start and end positions are within bounds
-    if start < 1 or end > len(sequence):
-        return "Start or end position is out of bounds."
+def heatmap_interface(sequence, start, end, state):
+    # If end is None or greater than sequence length, set it to sequence length
+    if end is None or end > len(sequence):
+        end = len(sequence)
+        state['end'] = end  # Update state
+
+    # Ensure start is within bounds
+    if start < 1:
+        return "Start position is out of bounds."
 
     # Generate heatmap
     heatmap_path = generate_heatmap(sequence, start, end)
-    return heatmap_path
+    return heatmap_path, state
+
+# Initialize state with None for end position
+initial_state = {'end': None}
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=heatmap_interface,
     inputs=[
-        gr.Textbox(lines=2, placeholder="MAPLRKTYVLKLYVAGNTPNSVRALKTLNNILEKEFKGVYALKVIDVLKNPQLAEEDKILATPTLAKVLPPPVRRIIGDLSNREKVLIGLDLLYEEIGDQAEDDLGLE"),
-        gr.Number(label="Start Position", value=1),  # Use 'value' instead of 'default'
-        gr.Number(label="End Position", value=len("MAPLRKTYVLKLYVAGNTPNSVRALKTLNNILEKEFKGVYALKVIDVLKNPQLAEEDKILATPTLAKVLPPPVRRIIGDLSNREKVLIGLDLLYEEIGDQAEDDLGLE"))  # Example default end position
+        gr.Textbox(lines=2, placeholder="Enter Protein Sequence Here..."),
+        gr.Number(label="Start Position", value=1),
+        gr.Number(label="End Position")
     ],
-    outputs="image",
+    outputs=[
+        "image",
+        gr.State()
+    ],
+    state=initial_state,
     live=True
 )
 
 # Run the Gradio app
 iface.launch()
+