Fixed an error in processing ids

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import requests
 import json
 import time
+import io
 
 st.title("Transaction Summarizer")
 
@@ -15,20 +16,13 @@ if model_selection == "Gemma":
 elif model_selection == "Tinyllama":
     base_url = "https://api.runpod.ai/v2/0wnm75vx5o77s1/"
 
-# Initialize session state for job ID and status
+# Initialize session state for jobs list
 if "jobs" not in st.session_state:
     st.session_state.jobs = []
 
 
-# Function to update the status of a job
-def update_job_status(job_id, status):
-    for job in st.session_state.jobs:
-        if job["id"] == job_id:
-            job["status"] = status
-
-
 # Define the transaction processing function
-def process_transactions(transactions, job_id):
+def process_transactions(transactions):
     url = base_url + "runsync"
 
     # Retrieve API key from Streamlit secrets
@@ -54,7 +48,15 @@ def process_transactions(transactions, job_id):
         # Parse response to get job ID
         result = response.json()
         job_id = result['id']
-        update_job_status(job_id, "IN_QUEUE")
+
+        # Add the job to session state and set initial status
+        st.session_state.jobs.append({
+            "id": job_id,
+            "status": "IN_QUEUE",
+            "transactions": transactions,
+            "result": None
+        })
+
         status_url = f"{base_url}status/{job_id}"
 
         # Polling the job status
@@ -62,7 +64,11 @@ def process_transactions(transactions, job_id):
             status_response = requests.get(status_url, headers=headers)
             status_data = status_response.json()
             status = status_data.get('status', '')
-            update_job_status(job_id, status)
+
+            # Update job status in session state
+            for job in st.session_state.jobs:
+                if job["id"] == job_id:
+                    job["status"] = status
 
             if status == "COMPLETED":
                 break
@@ -76,7 +82,7 @@ def process_transactions(transactions, job_id):
         result_response.raise_for_status()
         result_data = result_response.json().get('output')
 
-        update_job_status(job_id, "COMPLETED")
+        # Update job result in session state
         for job in st.session_state.jobs:
             if job["id"] == job_id:
                 job["result"] = result_data
@@ -85,7 +91,6 @@ def process_transactions(transactions, job_id):
 
     except requests.exceptions.RequestException as e:
         st.error(f"An error occurred: {e}")
-        update_job_status(job_id, "ERROR")
         return None
 
 
@@ -104,13 +109,9 @@ with tab1:
         # Split transactions and strip whitespace
         new_transactions = [i.strip() for i in new_transactions_input.split(',') if i.strip()]
 
-        # Add the job to session state and set initial status
-        job_id = str(len(st.session_state.jobs) + 1)
-        st.session_state.jobs.append({"id": job_id, "status": "IN_QUEUE", "transactions": new_transactions})
-
         with st.spinner("Processing..."):
             # Process the transactions and display the results
-            result_data = process_transactions(new_transactions, job_id)
+            result_data = process_transactions(new_transactions)
 
             if result_data:
                 st.write("Transaction Summaries:")
@@ -135,17 +136,28 @@ with tab2:
             # Process the transactions in the CSV file
             transactions = df['transaction'].tolist()
 
-            # Add the job to session state and set initial status
-            job_id = str(len(st.session_state.jobs) + 1)
-            st.session_state.jobs.append({"id": job_id, "status": "IN_QUEUE", "transactions": transactions})
-
             with st.spinner("Processing..."):
                 # Process the transactions and display the results
-                result_data = process_transactions(transactions, job_id)
+                result_data = process_transactions(transactions)
 
                 if result_data:
-                    st.write("Transaction Summaries:")
-                    st.write(result_data)
+                    df['summary'] = result_data
+                    st.write("Summarized Data:")
+                    st.write(df)
+
+                    # Prepare the summarized data for download
+                    csv_buffer = io.StringIO()
+                    df.to_csv(csv_buffer, index=False)
+                    csv_buffer.seek(0)
+
+                    # Download link for the summarized CSV
+                    st.download_button(
+                        label="Download Summarized CSV",
+                        data=csv_buffer,
+                        file_name="summarized_transactions.csv",
+                        mime="text/csv"
+                    )
+
                 else:
                     st.write("The job was cancelled or encountered an error.")