Waiting for response

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import json
+import time
 
 st.title("Transaction Summarizer")
 
@@ -22,17 +23,32 @@ if st.button("Submit"):
     json_data = json.dumps(data)
 
     try:
-        # Send POST request
+        # Send POST request to start processing
         response = requests.post(url, headers=headers, data=json_data)
         response.raise_for_status()  # Raise an error for bad status codes
 
-        # Parse and display response
+        # Parse response to get job ID
         result = response.json()
-        st.write("Response:", result)
+        job_id = result['id']
+        st.write(f"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}"
+        status = "IN_QUEUE"
+        while status == "IN_QUEUE":
+            status_response = requests.get(status_url, headers=headers)
+            status_data = status_response.json()
+            status = status_data.get('status', '')
+            time.sleep(5)  # Adjust interval as needed
+            
+            st.write(f"Current status: {status}")
+
+        # Once status changes, display final result
+        st.write("Final status:", 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="")
-