Update app.py

app.py

@@ -55,7 +55,7 @@ def check_rate_limit(request: gr.Request):
 
 # ----------------- JigsawStack API Wrappers ------------------
 
-def web_ai_search(query, ai_overview, safe_search, spell_check, deep_research, max_depth, max_breadth, max_output_tokens, target_output_tokens, request: gr.Request):
+def web_ai_search(query, ai_overview, safe_search, spell_check, max_depth, max_breadth, max_output_tokens, target_output_tokens, request: gr.Request):
     rate_limit_ok, rate_limit_msg = check_rate_limit(request)
     if not rate_limit_ok:
         return f"❌ {rate_limit_msg}", "", [], ""
@@ -66,18 +66,17 @@ def web_ai_search(query, ai_overview, safe_search, spell_check, deep_research, m
         "query": query.strip(),
         "ai_overview": ai_overview,
         "safe_search": safe_search,
-        "spell_check": spell_check
+        "spell_check": spell_check,
+        "deep_research": True
     }
-    if deep_research:
-        payload["deep_research"] = True
-        config = {
-            "max_depth": max_depth if max_depth is not None else 3,
-            "max_breadth": max_breadth if max_breadth is not None else 3,
-            "max_output_tokens": max_output_tokens if max_output_tokens is not None else 32000
-        }
-        if target_output_tokens is not None and target_output_tokens != "":
-            config["target_output_tokens"] = target_output_tokens
-        payload["deep_research_config"] = config
+    config = {
+        "max_depth": max_depth if max_depth is not None else 3,
+        "max_breadth": max_breadth if max_breadth is not None else 3,
+        "max_output_tokens": max_output_tokens if max_output_tokens is not None else 32000
+    }
+    if target_output_tokens is not None and target_output_tokens != "":
+        config["target_output_tokens"] = target_output_tokens
+    payload["deep_research_config"] = config
     try:
         response = requests.post(f"{BASE_URL}/web/search", headers=headers, json=payload)
         if response.status_code != 200:
@@ -114,8 +113,7 @@ with gr.Blocks() as demo:
                 ai_overview = gr.Checkbox(label="AI Overview", value=True)
                 safe_search = gr.Dropdown(choices=["moderate", "strict", "off"], value="moderate", label="Safe Search")
                 spell_check = gr.Checkbox(label="Spell Check", value=True)
-                deep_research = gr.Checkbox(label="Deep Research", value=True)
-                with gr.Group(visible=True) as deep_research_group:
+                with gr.Group() as deep_research_group:
                     max_depth = gr.Number(label="Max Depth", value=3, precision=0)
                     max_breadth = gr.Number(label="Max Breadth", value=3, precision=0)
                     max_output_tokens = gr.Number(label="Max Output Tokens", value=32000, precision=0)
@@ -129,12 +127,10 @@ with gr.Blocks() as demo:
                 search_json_box = gr.Accordion("Raw JSON Response", open=False)
                 with search_json_box:
                     search_json_output = gr.Textbox(show_label=False, lines=20, interactive=False)
-    def toggle_deep_research(checked):
-            return {deep_research_group: gr.update(visible=checked)}
-    deep_research.change(fn=toggle_deep_research, inputs=deep_research, outputs=deep_research_group)
-    def on_search(query, overview, safe, spell, deep, d_depth, d_breadth, d_tokens, d_target, request: gr.Request):
-        return web_ai_search(query, overview, safe, spell, deep, d_depth, d_breadth, d_tokens, d_target, request)
-    search_btn.click(fn=on_search, inputs=[search_query, ai_overview, safe_search, spell_check, deep_research, max_depth, max_breadth, max_output_tokens, target_output_tokens],
+
+    def on_search(query, overview, safe, spell, d_depth, d_breadth, d_tokens, d_target, request: gr.Request):
+        return web_ai_search(query, overview, safe, spell, d_depth, d_breadth, d_tokens, d_target, request)
+    search_btn.click(fn=on_search, inputs=[search_query, ai_overview, safe_search, spell_check, max_depth, max_breadth, max_output_tokens, target_output_tokens],
                         outputs=[search_status, overview_box, results_box, search_json_output])
     def clear_search():
             return "", "", [], ""