Added tabs for better organization

app.py

@@ -10,26 +10,19 @@ st.title("Transaction Summarizer")
 # Input for selecting model
 model_selection = st.radio("Select model to use:", ["Tinyllama", "Gemma"])
 
-# Input for submitting new transactions
-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
-
 # 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/"
 
+# Initialize session state for 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
 
+# Define the transaction processing function
 def process_transactions(transactions):
     url = base_url + "runsync"
 
@@ -81,64 +74,85 @@ def process_transactions(transactions):
             st.error(f"An error occurred: {e}")
             return None
 
+# Creating tabs for different pages
+tab1, tab2, tab3 = st.tabs(["Submit Transactions", "Upload CSV", "Manage Jobs"])
 
-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)
+# Tab 1: Submit Transactions
+with tab1:
+    st.header("Submit New Transactions")
 
-        # Process the transactions in the CSV file
-        transactions = df['transaction'].tolist()
-        result_data = process_transactions(transactions)
+    # Input for submitting new transactions
+    new_transactions_input = st.text_area("Enter your transactions (comma-separated)", key="input_area")
+    submit_button = st.button("Submit New Transactions", type="primary")
 
+    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:
-            # Assuming result_data is a list of summaries matching the input transactions
-            df['summary'] = result_data
-            st.write("Summarized Data:")
-            st.write(df)
+            st.write("Transaction Summaries:")
+            st.write(result_data)
+
+# Tab 2: Upload CSV
+with tab2:
+    st.header("Upload a CSV File of Transactions")
+
+    # File uploader for CSV files
+    uploaded_file = st.file_uploader("Upload a CSV file of transactions", type=["csv"])
 
-            # 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":
-    cancel_button = st.button("Cancel Request")
-    if cancel_button:
-        cancel_url = f"{base_url}cancel/{st.session_state.job_id}"
+    if uploaded_file is not None:
+        # Read the uploaded CSV file
         try:
-            cancel_response = requests.post(cancel_url, headers=headers)
-            cancel_response.raise_for_status()
-            cancel_result = cancel_response.json()
-            st.session_state.job_status = "CANCELLED"
-            st.write(f"Job {st.session_state.job_id} has been cancelled.")
-        except requests.exceptions.RequestException as e:
-            st.error(f"An error occurred while cancelling: {e}")
+            df = pd.read_csv(uploaded_file)
+            st.write("Uploaded Data:")
+            st.write(df)
 
-# Reset button
-if st.button("Reset", key="reset_button"):
-    st.session_state.job_id = None
-    st.session_state.job_status = None
-    st.text_area("Enter your transactions (comma-separated)", value="", key="reset_area")
+            # 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}")
+
+# Tab 3: Manage Jobs
+with tab3:
+    st.header("Manage Jobs")
+
+    # Cancel button
+    if st.session_state.job_id and st.session_state.job_status == "IN_QUEUE":
+        cancel_button = st.button("Cancel Request")
+        if cancel_button:
+            cancel_url = f"{base_url}cancel/{st.session_state.job_id}"
+            try:
+                cancel_response = requests.post(cancel_url, headers=headers)
+                cancel_response.raise_for_status()
+                cancel_result = cancel_response.json()
+                st.session_state.job_status = "CANCELLED"
+                st.write(f"Job {st.session_state.job_id} has been cancelled.")
+            except requests.exceptions.RequestException as e:
+                st.error(f"An error occurred while cancelling: {e}")
+
+    # Reset button
+    if st.button("Reset", key="reset_button"):
+        st.session_state.job_id = None
+        st.session_state.job_status = None
+        st.text_area("Enter your transactions (comma-separated)", value="", key="reset_area")