Waiting for response

app.py

@@ -3,12 +3,14 @@ import requests
 import json
 import time
 
-st.title("Transaction Summarizer v2")
+st.title("Transaction Summarizer")
 
-transactions = st.text_area("Enter your transactions (comma-separated)").split(',')
-transactions = [i.strip() for i in transactions if i.strip()]
+# Input for submitting new transactions
+new_transactions = st.text_area("Enter your transactions (comma-separated)").split(',')
+new_transactions = [i.strip() for i in new_transactions if i.strip()]
 
-if st.button("Submit"):
+# Submit button for new transactions
+if st.button("Submit New Transactions"):
     url = "https://api.runpod.ai/v2/0wnm75vx5o77s1/run"
     headers = {
         'Content-Type': 'application/json',
@@ -16,7 +18,7 @@ if st.button("Submit"):
     }
     data = {
         'input': {
-            'transaction': transactions
+            'transaction': new_transactions
         }
     }
     
@@ -30,7 +32,7 @@ if st.button("Submit"):
         # Parse response to get job ID
         result = response.json()
         job_id = result['id']
-        st.write(f"Job ID: {job_id}")
+        st.write(f"New Job ID: {job_id}")
 
         # Keep checking status until it's no longer 'IN_QUEUE'
         status_url = f"https://api.runpod.ai/v2/0wnm75vx5o77s1/status:{job_id}"
@@ -43,12 +45,34 @@ if st.button("Submit"):
             
             st.write(f"Current status: {status}")
 
-        # Once status changes, display final result
+        # Once status changes, display final status
         st.write("Final status:", status_data)
         
     except requests.exceptions.RequestException as e:
         st.error(f"An error occurred: {e}")
 
+# Input and button for fetching status or result by job ID
+st.subheader("Fetch Status or Result by Job ID")
+job_id_input = st.text_input("Enter Job ID")
+
+if st.button("Fetch Status or Result"):
+    if job_id_input:
+        try:
+            status_url = f"https://api.runpod.ai/v2/0wnm75vx5o77s1/status:{job_id_input}"
+            headers = {
+                'Authorization': 'API_KEY'
+            }
+
+            # Fetch status or result based on job ID
+            status_response = requests.get(status_url, headers=headers)
+            status_data = status_response.json()
+            
+            # Display status or result
+            st.write("Status or Result:", status_data)
+        
+        except requests.exceptions.RequestException as e:
+            st.error(f"An error occurred: {e}")
+
 else:
     if st.button("Reset"):
         st.text_area("Enter your transactions (comma-separated)", value="")