Update app.py

app.py

@@ -1,202 +1,102 @@
 import gradio as gr
-import json
 import os
+import json
+import glob
 
-# File to store all lists data
-DATA_FILE = "lists_data.json"
-
-# Initialize with preset links
-def initialize_data():
-    preset_links = [
-        {
-            "name": "Learn Git Branching",
-            "description": "Interactive Git tutorial",
-            "link": "https://learngitbranching.js.org/?locale=en_US"
-        },
-        {
-            "name": "Terminus",
-            "description": "Command line tutorial game",
-            "link": "https://web.mit.edu/mprat/Public/web/Terminus/Web/main.html"
-        },
-        {
-            "name": "VisuAlgo",
-            "description": "Algorithm visualization",
-            "link": "https://visualgo.net/en"
-        },
-        {
-            "name": "Linux Survival",
-            "description": "Linux tutorial",
-            "link": "https://linuxsurvival.com/"
-        }
-    ]
-    
-    return {
-        "lists": {
-            "default": {
-                "name": "Default List",
-                "items": preset_links
-            }
-        }, 
-        "current_list": "default"
-    }
+# Directory to store JSON list files
+LISTS_DIR = "lists"
 
-# Load or initialize data
-def load_data():
-    if os.path.exists(DATA_FILE):
-        with open(DATA_FILE, "r") as f:
-            return json.load(f)
-    data = initialize_data()
-    save_data(data)
-    return data
+# Ensure the lists directory exists
+os.makedirs(LISTS_DIR, exist_ok=True)
 
-def save_data(data):
-    with open(DATA_FILE, "w") as f:
-        json.dump(data, f)
 
-# Initialize data
-data = load_data()
+# Get list of available JSON files
+def get_list_files():
+    files = glob.glob(os.path.join(LISTS_DIR, "*.json"))
+    return [os.path.basename(f)[:-5] for f in files]  # Remove .json extension
 
-# Function to create a new list
-def create_list(list_name):
-    if not list_name.strip():
-        return gr.update(), "Please enter a list name", data["current_list"], get_list_items()
-    
-    list_id = list_name.lower().replace(" ", "_")
-    
-    if list_id in data["lists"]:
-        return gr.update(), f"List '{list_name}' already exists", data["current_list"], get_list_items()
-    
-    data["lists"][list_id] = {
-        "name": list_name,
-        "items": []
-    }
-    
-    data["current_list"] = list_id
-    save_data(data)
-
-    return gr.update(choices=get_list_choices(), value=list_id), f"Created list: {list_name}", list_name, []
-
-# Function to switch between lists
-def switch_list(list_id):
-    if list_id:
-        data["current_list"] = list_id
-        save_data(data)
-        list_name = data["lists"][list_id]["name"]
-        return list_name, get_list_items()
-    return "", []
-
-# Function to add an item to the current list
-def add_item(name, description, link):
-    if not data["current_list"]:
-        return "Please select or create a list first", get_list_items()
-    
-    if not name.strip():
-        return "Please enter a name for the item", get_list_items()
-    
-    current_list = data["current_list"]
-    
-    new_item = {
-        "name": name,
-        "description": description,
-        "link": link
-    }
-    
-    data["lists"][current_list]["items"].append(new_item)
-    save_data(data)
-    
-    return f"Added item: {name}", get_list_items()
+# Load data from a JSON file
+def load_list_data(list_name):
+    file_path = os.path.join(LISTS_DIR, f"{list_name}.json")
+    if os.path.exists(file_path):
+        with open(file_path, "r") as f:
+            return json.load(f)
+    return {"name": "Empty List", "items": []}
 
-# Function to get current list items for display
-def get_list_items():
-    if not data["current_list"] or data["current_list"] not in data["lists"]:
-        return []
-    
+# Get list items in table format
+def get_list_items(list_name):
+    data = load_list_data(list_name)
     return [[item["name"], item["description"], item["link"]] 
-            for item in data["lists"][data["current_list"]]["items"]]
+            for item in data["items"]]
 
-# Function to remove an item
-def remove_item(evt: gr.SelectData):
-    if not data["current_list"]:
-        return "No list selected", get_list_items()
-    
-    row_index = evt.index[0]
-    current_list = data["current_list"]
-    
-    if 0 <= row_index < len(data["lists"][current_list]["items"]):
-        removed_item = data["lists"][current_list]["items"].pop(row_index)
-        save_data(data)
-        return f"Removed: {removed_item['name']}", get_list_items()
-    
-    return "Error removing item", get_list_items()
+# Get list title
+def get_list_title(list_name):
+    data = load_list_data(list_name)
+    return data["name"]
 
-# Get list choices for dropdown
-def get_list_choices():
-    return [(list_id, data["lists"][list_id]["name"]) for list_id in data["lists"]]
+# Function to switch between lists
+def switch_list(list_name):
+    return get_list_title(list_name), get_list_items(list_name)
 
 # Build the Gradio interface
-with gr.Blocks(title="Simple List Manager") as app:
-    gr.Markdown("# Simple List Manager")
+with gr.Blocks(title="Static List Manager") as app:
+    gr.Markdown("# Static List Manager")
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown("### Create New List")
-            new_list_name = gr.Textbox(label="New List Name")
-            create_btn = gr.Button("Create List")
-            
-            gr.Markdown("### Switch Lists")
-            list_dropdown = gr.Dropdown(
-                choices=get_list_choices(),
-                label="Select List",
-                value=data["current_list"] if data["current_list"] else None,
-                interactive=True
-            )
-        
-        with gr.Column(scale=2):
-            current_list_name = gr.Markdown(
-                f"### Current List: {data['lists'][data['current_list']]['name'] if data['current_list'] in data['lists'] else 'None'}"
-            )
-            status_message = gr.Markdown("")
+    # Get available lists
+    list_names = get_list_files()
+    if not list_names:
+        list_names = ["learn"]  # Fallback to default
+    
+    # Current list state
+    current_list = gr.State(value=list_names[0])
     
-    gr.Markdown("### Add New Item")
     with gr.Row():
-        item_name = gr.Textbox(label="Name")
-        item_description = gr.Textbox(label="Description")
-        item_link = gr.Textbox(label="Link")
+        # Create tabs for each list
+        with gr.Tabs() as tabs:
+            tab_list = []
+            for list_name in list_names:
+                tab = gr.Tab(label=get_list_title(list_name))
+                tab_list.append((tab, list_name))
     
-    add_btn = gr.Button("Add Item")
+    list_title = gr.Markdown(f"## {get_list_title(list_names[0])}")
     
-    gr.Markdown("### List Items (click row to delete)")
+    # Display table of list items
     items_table = gr.Dataframe(
         headers=["Name", "Description", "Link"],
         datatype=["str", "str", "str"],
-        value=get_list_items(),
+        value=get_list_items(list_names[0]),
         interactive=False
     )
     
-    # Connect events
-    create_btn.click(
-        create_list,
-        inputs=[new_list_name],
-        outputs=[list_dropdown, status_message, current_list_name, items_table]
-    )
-    
-    list_dropdown.change(
-        switch_list,
-        inputs=[list_dropdown],
-        outputs=[current_list_name, items_table]
-    )
-    
-    add_btn.click(
-        add_item,
-        inputs=[item_name, item_description, item_link],
-        outputs=[status_message, items_table]
-    )
-    
-    items_table.select(
-        remove_item,
-        None,
-        [status_message, items_table]
-    )
+    # Connect tab events to switch lists
+    for i, (tab, list_name) in enumerate(tab_list):
+        tab.select(
+            lambda name=list_name: switch_list(name), 
+            inputs=[], 
+            outputs=[list_title, items_table]
+        )
+
+    # Add instructions
+    gr.Markdown("""
+    ### How to Add More Lists
+    
+    1. Create a JSON file in the 'lists' folder with the format: listname.json
+    2. Use the following structure in your JSON file:
+    ```json
+    {
+        "name": "Your List Title",
+        "items": [
+            {
+                "name": "Item Name",
+                "description": "Item Description",
+                "link": "https://example.com"
+            },
+            ...
+        ]
+    }
+    ```
+    3. Restart the app to see your new list
+    """)
 
 # Launch the app
-app.launch()
+app.launch()