Added secrets

app.py

@@ -12,9 +12,13 @@ new_transactions = [i.strip() for i in new_transactions if i.strip()]
 # Submit button for new transactions
 if st.button("Submit New Transactions"):
     url = "https://api.runpod.ai/v2/0wnm75vx5o77s1/run"
+    
+    # Retrieve API key from Streamlit secrets
+    api_key = st.secrets["api_key"]
+    
     headers = {
         'Content-Type': 'application/json',
-        'Authorization': 'API_KEY'
+        'Authorization': api_key
     }
     data = {
         'input': {
@@ -41,9 +45,7 @@ if st.button("Submit New Transactions"):
             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}")
+            time.sleep(2)  # Adjust interval as needed
 
         # Once status changes, display final status
         st.write("Final status:", status_data)
@@ -51,27 +53,6 @@ if st.button("Submit New Transactions"):
     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"):