Managing jobs in tab 3

app.py

@@ -17,10 +17,9 @@ elif model_selection == "Tinyllama":
     base_url = "https://api.runpod.ai/v2/0wnm75vx5o77s1/"
 
 # Initialize session state for job ID and status
-if "job_id" not in st.session_state:
-    st.session_state.job_id = None
-if "job_status" not in st.session_state:
-    st.session_state.job_status = None
+if "jobs" not in st.session_state:
+    st.session_state.jobs = []
+
 
 # Define the transaction processing function
 def process_transactions(transactions):
@@ -50,17 +49,23 @@ def process_transactions(transactions):
 
             # Parse response to get job ID
             result = response.json()
-            st.session_state.job_id = result['id']
-            st.write(f"New Job ID: {st.session_state.job_id}")
+            job_id = result['id']
+            st.session_state.jobs.append({"id": job_id, "status": "IN_QUEUE"})
+            st.write(f"New Job ID: {job_id}")
 
             # Keep checking status until it's no longer 'IN_QUEUE' or cancelled
-            status_url = f"{base_url}status/{st.session_state.job_id}"
-            st.session_state.job_status = "IN_QUEUE"
-            while st.session_state.job_status != "COMPLETED":
+            status_url = f"{base_url}status/{job_id}"
+            while True:
                 status_response = requests.get(status_url, headers=headers)
                 status_data = status_response.json()
-                st.session_state.job_status = status_data.get('status', '')
-                if st.session_state.job_status == "COMPLETED":
+                status = status_data.get('status', '')
+
+                # Update the job status in session state
+                for job in st.session_state.jobs:
+                    if job["id"] == job_id:
+                        job["status"] = status
+
+                if status == "COMPLETED":
                     break
                 time.sleep(2)  # Adjust interval as needed
 
@@ -74,6 +79,7 @@ def process_transactions(transactions):
             st.error(f"An error occurred: {e}")
             return None
 
+
 # Creating tabs for different pages
 tab1, tab2, tab3 = st.tabs(["Submit Transactions", "Upload CSV", "Manage Jobs"])
 
@@ -137,22 +143,25 @@ with tab2:
 with tab3:
     st.header("Manage Jobs")
 
-    # Cancel button
-    if st.session_state.job_id and st.session_state.job_status == "IN_QUEUE":
-        cancel_button = st.button("Cancel Request")
-        if cancel_button:
-            cancel_url = f"{base_url}cancel/{st.session_state.job_id}"
-            try:
-                cancel_response = requests.post(cancel_url, headers=headers)
-                cancel_response.raise_for_status()
-                cancel_result = cancel_response.json()
-                st.session_state.job_status = "CANCELLED"
-                st.write(f"Job {st.session_state.job_id} has been cancelled.")
-            except requests.exceptions.RequestException as e:
-                st.error(f"An error occurred while cancelling: {e}")
+    # Display job details and provide cancel option
+    if st.session_state.jobs:
+        for job in st.session_state.jobs:
+            st.write(f"Job ID: {job['id']} - Status: {job['status']}")
+
+            # Cancel button for each job
+            if job["status"] == "IN_QUEUE":
+                if st.button(f"Cancel Job {job['id']}"):
+                    cancel_url = f"{base_url}cancel/{job['id']}"
+                    try:
+                        cancel_response = requests.post(cancel_url, headers=headers)
+                        cancel_response.raise_for_status()
+                        job["status"] = "CANCELLED"
+                        st.write(f"Job {job['id']} has been cancelled.")
+                    except requests.exceptions.RequestException as e:
+                        st.error(f"An error occurred while cancelling: {e}")
+    else:
+        st.write("No active jobs.")
 
     # Reset button
-    if st.button("Reset", key="reset_button"):
-        st.session_state.job_id = None
-        st.session_state.job_status = None
-        st.text_area("Enter your transactions (comma-separated)", value="", key="reset_area")
+    if st.button("Reset All Jobs"):
+        st.session_state.jobs = []