Streamlit poc

Author: letiesperon

README.md

@@ -1,97 +1,37 @@
 # Sequence Extractor
 
-Implemented as a simple UI application built with Python and [Tkinter](https://docs.python.org/3/library/tkinter.html#module-tkinter).
+A streamlit application for processing Excel files.
 
-## Requirements
+## Description
 
-- Python 3.13 or later
-- [uv](https://github.com/astral-sh/uv) - Fast Python package installer and resolver
-
-## Installing uv
-
-Before you can use this project, you need to install uv on your Mac:
-
-```bash
-# Using Homebrew (recommended)
-brew install uv
-
-# Or using pip
-pip install uv
-```
-
-For more installation options, see the [uv installation guide](https://github.com/astral-sh/uv#installation).
+This application allows users to:
+- Upload a single "RS totales" Excel file
+- Upload multiple "Variant tables" Excel files
+- Process these files and view statistics on the data
 
 ## Installation
 
-1. Clone this repository:
-   ```bash
-   git clone <repository-url>
-   cd sequence_extractor
-   ```
-
-2. Create a virtual environment using uv:
-   ```bash
-   uv venv
-   ```
-
-3. Activate the virtual environment:
-   ```bash
-   source .venv/bin/activate
-   ```
-
-4. Install dependencies:
-   ```bash
-   uv pip install -e .
-   ```
-
-## Running the Application
+### Requirements
+- Python 3.13 or higher
 
-After installation, you can run the application:
+### Setup with UV
 
+Declare new dependencies:
 ```bash
-uv run main.py
+uv add streamlit pandas openpyxl
+uv add --dev pytest
 ```
 
-## Managing Dependencies
+## Usage
 
-### Adding New Dependencies
-
-To add a new dependency:
-
-1. Add it to the `dependencies` list in `pyproject.toml`:
-   ```toml
-   dependencies = [
-       "package-name>=1.0.0",
-   ]
-   ```
-
-2. Install the updated dependencies:
-   ```bash
-   uv pip install -e .
-   ```
-
-### Adding Development Dependencies
-
-For development-only dependencies:
-
-1. Add a `[project.optional-dependencies]` section to `pyproject.toml`:
-   ```toml
-   [project.optional-dependencies]
-   dev = [
-       "pytest>=7.0.0",
-       "black>=23.0.0",
-   ]
-   ```
-
-2. Install development dependencies:
-   ```bash
-   uv pip install -e ".[dev]"
-   ```
-
-## Running Tests
+Run the application:
+```bash
+uv run -m streamlit run main.py
+```
 
-If you've added pytest as a development dependency:
+## Development
 
+Run tests:
 ```bash
-uv run pytest
+pytest
 ```

main.py

@@ -1,16 +1,50 @@
-from tkinter import *
-from tkinter import ttk
+import streamlit as st
+import pandas as pd
+import time
 
-# def main():
-#     print("Hello from sequence-extractor!")
+def read_excel_file(file):
+    """Read an Excel file and return the DataFrame."""
+    return pd.read_excel(file)
 
+def count_total_rows(dataframes):
+    """Count total rows across all dataframes."""
+    return sum(len(df) for df in dataframes if df is not None)
 
-# if __name__ == "__main__":
-#     main()
+def main():
+    st.title("Excel File Sequence Extractor")
 
-root = Tk()
-frm = ttk.Frame(root, padding=10)
-frm.grid()
-ttk.Label(frm, text="Hello World!").grid(column=0, row=0)
-ttk.Button(frm, text="Quit", command=root.destroy).grid(column=1, row=0)
-root.mainloop()
+    # First input for single file
+    st.subheader("Input Files")
+    rs_totales_file = st.file_uploader("RS totales", type=["xlsx"])
+
+    # Second input for multiple files
+    variant_tables_files = st.file_uploader("Variant tables", type=["xlsx"], accept_multiple_files=True)
+
+    # Submit button
+    if st.button("Submit"):
+        if rs_totales_file is None or not variant_tables_files:
+            st.error("Please upload all required files.")
+        else:
+            # Show spinner while processing
+            with st.spinner("Processing files..."):
+                # Read the RS totales file
+                rs_df = read_excel_file(rs_totales_file)
+
+                # Read all variant tables files
+                variant_dfs = []
+                for file in variant_tables_files:
+                    variant_dfs.append(read_excel_file(file))
+
+                # Combine all dataframes for counting
+                all_dfs = [rs_df] + variant_dfs
+
+                # Simulate processing time
+                time.sleep(1)
+
+            # Display results
+            total_rows = count_total_rows(all_dfs)
+            st.success(f"Processing complete!")
+            st.metric("Total rows parsed across all files", total_rows)
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -4,7 +4,11 @@ version = "0.1.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"
-dependencies = []
+dependencies = [
+    "openpyxl>=3.1.5",
+    "pandas>=2.2.3",
+    "streamlit>=1.43.2",
+]
 
 [project.optional-dependencies]
 dev = [
@@ -13,3 +17,8 @@ dev = [
     "black>=23.9.1",
     "flake8>=6.1.0",
 ]
+
+[dependency-groups]
+dev = [
+    "pytest>=8.3.5",
+]

tests/test_main.py

@@ -1,6 +1,17 @@
 import pytest
+import io
+from main import read_excel_file, count_total_rows
 
 def test_basic_sum():
     """A simple placeholder test that demonstrates how to write a test."""
     # This is just a placeholder - replace with actual tests
     assert 1 + 1 == 2
+
+def test_count_total_rows():
+    """Test the function that counts total rows across dataframes."""
+    df1 = pd.DataFrame({'A': [1, 2, 3]})
+    df2 = pd.DataFrame({'B': [1, 2, 3, 4, 5]})
+    df3 = pd.DataFrame({'C': [1, 2]})
+
+    total = count_total_rows([df1, df2, df3])
+    assert total == 10