Adding the ability to upload csv files

.idea/.gitignore

@@ -0,0 +1,8 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/log-summarizer-tinyllama.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="jdk" jdkName="Python 3.12" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/log-summarizer-tinyllama.iml" filepath="$PROJECT_DIR$/.idea/log-summarizer-tinyllama.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
+import pandas as pd
 import requests
 import json
 import time
+import io
 
 st.title("Transaction Summarizer")
 
@@ -12,41 +14,38 @@ model_selection = st.radio("Select model to use:", ["Tinyllama", "Gemma"])
 new_transactions_input = st.text_area("Enter your transactions (comma-separated)", key="input_area")
 submit_button = st.button("Submit New Transactions", type="primary")
 
+# File uploader for CSV files
+uploaded_file = st.file_uploader("Upload a CSV file of transactions", type=["csv"])
+
 # Session state to keep track of 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 submit_button:
-    # Split transactions and strip whitespace
-    new_transactions = [i.strip() for i in new_transactions_input.split(',') if i.strip()]
-else:
-    new_transactions = []
-
 # Determine URL based on model selection
 if model_selection == "Gemma":
     base_url = "https://api.runpod.ai/v2/lld3iiy6fx7hcf/"
 elif model_selection == "Tinyllama":
     base_url = "https://api.runpod.ai/v2/0wnm75vx5o77s1/"
 
-# When submit button is clicked and there are transactions to process
-if submit_button and new_transactions:
+
+def process_transactions(transactions):
     url = base_url + "runsync"
-    
+
     # Retrieve API key from Streamlit secrets
     api_key = st.secrets["api_key"]
-    
+
     headers = {
         'Content-Type': 'application/json',
         'Authorization': api_key
     }
     data = {
         'input': {
-            'transaction': new_transactions
+            'transaction': transactions
         }
     }
-    
+
     json_data = json.dumps(data)
 
     # Show a spinner while waiting for the response
@@ -72,11 +71,58 @@ if submit_button and new_transactions:
                     break
                 time.sleep(2)  # Adjust interval as needed
 
-            # Once status changes, display final status
-            st.write("Final status:", status_data)
-            
+            # Once status changes, retrieve and display the result
+            result_url = f"{base_url}results/{st.session_state.job_id}"
+            result_response = requests.get(result_url, headers=headers)
+            result_response.raise_for_status()
+            result_data = result_response.json()
+            return result_data
+
         except requests.exceptions.RequestException as e:
             st.error(f"An error occurred: {e}")
+            return None
+
+
+if submit_button and new_transactions_input:
+    # Split transactions and strip whitespace
+    new_transactions = [i.strip() for i in new_transactions_input.split(',') if i.strip()]
+    result_data = process_transactions(new_transactions)
+    if result_data:
+        st.write("Transaction Summaries:")
+        st.write(result_data)
+
+if uploaded_file is not None:
+    # Read the uploaded CSV file
+    try:
+        df = pd.read_csv(uploaded_file)
+        st.write("Uploaded Data:")
+        st.write(df)
+
+        # Process the transactions in the CSV file
+        transactions = df['transaction'].tolist()
+        result_data = process_transactions(transactions)
+
+        if result_data:
+            # Assuming result_data is a list of summaries matching the input transactions
+            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"
+            )
+
+    except Exception as e:
+        st.error(f"An error occurred while processing the CSV file: {e}")
 
 # Cancel button
 if st.session_state.job_id and st.session_state.job_status == "IN_QUEUE":