update inventory

app.py

@@ -8,7 +8,8 @@ from ui import (
     #general_blog,
     meta,
     twitter,
-    price_matching
+    price_matching,
+    inventory
 )
 
 from server import (
@@ -16,7 +17,8 @@ from server import (
     #general_blog as general_blog_srv,
     meta as meta_srv,
     twitter as twitter_srv,
-    price_matching as price_matching_srv
+    price_matching as price_matching_srv,
+    inventory as inventory_srv
 )
 
 
@@ -27,6 +29,7 @@ ui = ui.page_fluid(
         meta.ui,
         twitter.ui,
         price_matching.ui,
+        inventory.ui,
         title="SEO Blog Writer",
         header=ui.tags.head(
             ui.tags.link(rel='stylesheet', type='text/css', href='style.css')
@@ -40,5 +43,6 @@ def server(input, output, session):
     meta_srv.server(input, output, session)
     twitter_srv.server(input, output, session)
     price_matching_srv.server(input, output, session)
+    inventory_srv.server(input, output, session)
 
 app = App(ui, server)

server/inventory.py

@@ -0,0 +1,102 @@
+import os
+import json
+import duckdb
+from shiny import render, reactive, ui
+from google.oauth2 import service_account
+from googleapiclient.discovery import build
+from dotenv import load_dotenv
+
+load_dotenv()
+
+SERVICE_ACCOUNT_JSON = json.loads(os.getenv("SERVICE_ACCOUNT_JSON"))
+GOOGLE_FOLDER_ID = os.getenv("GOOGLE_DRIVE_FOLDER_ID")
+SCOPES = [
+    "https://www.googleapis.com/auth/drive.readonly",
+    "https://www.googleapis.com/auth/spreadsheets.readonly"
+]
+
+credentials = service_account.Credentials.from_service_account_info(SERVICE_ACCOUNT_JSON, scopes=SCOPES)
+drive_service = build("drive", "v3", credentials=credentials)
+sheets_service = build("sheets", "v4", credentials=credentials)
+
+def list_sheets_in_folder(folder_id):
+    query = (
+        f"'{folder_id}' in parents and "
+        "mimeType = 'application/vnd.google-apps.spreadsheet' and trashed = false"
+    )
+    results = drive_service.files().list(q=query, fields="files(id, name)").execute()
+    return results.get("files", [])
+
+def load_sheet_to_duckdb(spreadsheet_id, sheet_tab="Sheet1", table_name="active_sheet"):
+    values = (
+        sheets_service.spreadsheets()
+        .values()
+        .get(spreadsheetId=spreadsheet_id, range=sheet_tab)
+        .execute()
+        .get("values", [])
+    )
+
+    if not values:
+        raise ValueError("Google Sheet is empty or not accessible.")
+
+    headers = values[0]
+    rows = [r + [''] * (len(headers) - len(r)) for r in values[1:]]  # pad short rows
+
+    con = duckdb.connect()
+    insert_rows = ", ".join(str(tuple(r)) for r in rows)
+    insert_cols = ", ".join(f'"{h}"' for h in headers)
+
+    con.execute(f"""
+        CREATE OR REPLACE TABLE {table_name} AS
+        SELECT * FROM (VALUES {insert_rows}) AS t({insert_cols});
+    """)
+    return con
+
+sheet_index = {}
+
+def server(input, output, session):
+    con = {"conn": None}
+
+    @reactive.Effect
+    def init_dropdown_from_folder():
+        global sheet_index
+        try:
+            sheets = list_sheets_in_folder(GOOGLE_FOLDER_ID)
+            sheet_index = {s['name']: s['id'] for s in sheets}
+            sheet_names = list(sheet_index.keys())
+            print(f"[DEBUG] Found {len(sheet_names)} sheets in folder: {sheet_names}")
+
+            ui.update_select("sheet_dropdown", choices=sheet_names)
+
+        except Exception as e:
+            print(f"[ERROR] Failed to list folder contents or populate dropdown: {e}")
+
+    @reactive.Effect
+    def load_selected_sheet():
+        sheet_name = input.sheet_dropdown()
+        if sheet_name not in sheet_index:
+            return
+        spreadsheet_id = sheet_index[sheet_name]
+        print(f"[DEBUG] Loading sheet: {sheet_name} (ID: {spreadsheet_id})")
+        con["conn"] = load_sheet_to_duckdb(spreadsheet_id)
+
+    @output
+    @render.table
+    def results():
+        sheet_name = input.sheet_dropdown()  # force reactivity on sheet change
+
+        try:
+            if con["conn"] is None:
+                return duckdb.query("SELECT 'Waiting for sheet selection' AS status").to_df()
+
+            name_query = input.name_filter()
+            if name_query:
+                return con["conn"].execute(
+                    "SELECT * FROM active_sheet WHERE item_name ILIKE '%' || ? || '%'",
+                    (name_query,)
+                ).fetchdf()
+            else:
+                return con["conn"].execute("SELECT * FROM active_sheet").fetchdf()
+
+        except Exception as e:
+            return duckdb.query(f"SELECT 'Error: {str(e).replace('\'','')}' AS error").to_df()

ui/inventory.py

@@ -0,0 +1,20 @@
+from shiny import ui
+
+ui = ui.nav_panel(
+    "Inventory",
+
+    ui.input_select(
+        "sheet_dropdown",
+        "Select a Sheet from Folder",
+        choices=[],
+        selected=None
+    ),
+
+    ui.input_text(
+        "name_filter",
+        "Search item_name",
+        placeholder="Type to filter by item_name"
+    ),
+
+    ui.output_table("results", class_="mt-3")
+)