Update app.py

app.py

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