Update app.py

app.py

@@ -796,8 +796,12 @@ def process_input(prompt: str, uploaded_files):
             client = get_openai_client()
             
             # Prepare system prompt
-            sys_prompt = get_idea_system_prompt()
 
+            selected_cat = st.session_state.get("category_focus", "(None)")
+            if selected_cat == "(None)":
+                selected_cat = None
+
+            sys_prompt = get_idea_system_prompt(selected_category=selected_cat)
             # Optional: gather search results
             search_content = None
             if use_web_search: