Revert App.py

app.py

@@ -1,3 +1,5 @@
+Please review the following code. nothing to do right now but familarize yourself with it:
+ 
 # app.py for HF Spaces: Gradio UI for Multi-Book Paranormal Text Repository
 # (With Codex Tree v0.5: Hierarchical categorization)
 import gradio as gr
@@ -148,8 +150,8 @@ semantic_pipeline.add_component("retriever_embedding", retriever_embedding_sem)
 semantic_pipeline.connect("embedder.embedding", "retriever_embedding.query_embedding")
 logger.info(f"Semantic pipeline components: {semantic_pipeline.graph.nodes.keys()}")
 # Helper: Run search (updated for book filter via meta['book'], added debug logs for doc counts)
-def search_stories(query, source_filter=None, type_filter=None, search_mode="Both", top_k=1000, min_score=0.2, category_filter="All"):
-    logger.info(f"Searching for query: {query}, source: {source_filter}, type: {type_filter}, mode: {search_mode}, min_score: {min_score}, category_filter: {category_filter}")
+def search_stories(query, source_filter=None, type_filter=None, search_mode="Both", top_k=1000, min_score=0.2):
+    logger.info(f"Searching for query: {query}, source: {source_filter}, type: {type_filter}, mode: {search_mode}, min_score: {min_score}")
     filters = {"operator": "AND", "conditions": []}
     if source_filter and source_filter != "All Sources":
         filters["conditions"].append({"field": "book", "operator": "==", "value": source_filter})
@@ -216,13 +218,6 @@ def search_stories(query, source_filter=None, type_filter=None, search_mode="Bot
             elif title in grouped:
                 grouped[title]["score"] = max(grouped[title]["score"], doc.score)
     sorted_results = sorted(grouped.values(), key=lambda x: x["score"], reverse=True)
-    if category_filter != "All":
-        filtered_results = []
-        for r in sorted_results:
-            has_assignment = r['title'] in stories_dict
-            if (category_filter == "Assigned" and has_assignment) or (category_filter == "Unassigned" and not has_assignment):
-                filtered_results.append(r)
-        sorted_results = filtered_results
     if search_mode == "Exact":
         sorted_results = [{**r, "search_query": query} for r in sorted_results]
     logger.info(f"Search returned {len(sorted_results)} results for query: {query}")
@@ -863,14 +858,7 @@ def remove_category(l1, l2, l3, l4, l5, l6, tree, sel):
     path = [p for p in [l1, l2, l3, l4, l5, l6] if p]
     if not path:
         return tree
-    title = sel['title']
-    tree = remove_from_path(tree, path, title)
-    paths = find_paths_for_title(tree, title)
-    if not paths:
-        global stories_dict
-        if title in stories_dict:
-            del stories_dict[title]
-    return tree
+    return remove_from_path(tree, path, sel['title'])
 # Helper: Update current categories display
 def update_current_categories(selected, tree):
     if not selected:
@@ -914,7 +902,7 @@ def save_and_status(tree):
     save_codex_tree(tree)
     return "Changes saved. Space may restart shortly."
 # Helper: Update results
-def update_results(query, source, type_filter, search_mode, min_score, category_filter):
+def update_results(query, source, type_filter, search_mode, min_score):
     try:
         mode_map = {
             "Keywords (Exact/Phrase Matches)": "Keywords",
@@ -923,7 +911,7 @@ def update_results(query, source, type_filter, search_mode, min_score, category_
             "Exact (Word/Phrase)": "Exact"
         }
         mode = mode_map.get(search_mode, "Both")
-        results = search_stories(query, source, type_filter, mode, min_score=min_score, category_filter=category_filter)
+        results = search_stories(query, source, type_filter, mode, min_score=min_score)
         choices = [
             f"{r['title']} (Score: {r['score']:.2f}) (Book: {r['book_slug'].replace('_', ' ').title()}) (Pages: {r['pages']}, Keywords: {r['keywords']})"
             for r in results
@@ -1079,7 +1067,6 @@ def gradio_ui():
                         source_filter = gr.Dropdown(label="Source", choices=sources, value="All Sources")
                         type_filter = gr.Dropdown(label="Type", choices=["Story", "Non-Story", "Both"], value="Both")
                         search_mode = gr.Dropdown(label="Search Mode", choices=["Keywords (Exact/Phrase Matches)", "Semantic (Conceptual Similarity)", "Both (Hybrid)", "Exact (Word/Phrase)"], value="Both (Hybrid)")
-                        category_filter = gr.Dropdown(label="Category Filter", choices=["All", "Assigned", "Unassigned"], value="All")
                         min_score_input = gr.Number(label="Min Score Threshold", value=0.1, minimum=0.0, step=0.05)
                         search_btn = gr.Button("Search")
                     # Main Row: Three Columns
@@ -1126,7 +1113,7 @@ def gradio_ui():
                             export_list_btn = gr.Dropdown(label="Export List", choices=["MD", "PDF", "Word"], interactive=True)
                             export_list_output = gr.HTML(value="", show_label=False)
                     # Event Handlers (updated for book_slug)
-                    search_btn.click(update_results, [query, source_filter, type_filter, search_mode, min_score_input, category_filter], [results_state, results_radio, stories_count])
+                    search_btn.click(update_results, [query, source_filter, type_filter, search_mode, min_score_input], [results_state, results_radio, stories_count])
                     results_radio.change(
                         select_story, [results_state, results_radio], [selected_story, edited_start, edited_end, edited_keywords]
                     ).then(