Update app.py

app.py

@@ -556,7 +556,6 @@ def create_gradio_app():
     """
     
     with gr.Blocks(css=css, title="🤖 Creative Agentic AI Chat", theme=gr.themes.Ocean()) as app:
-        
         # Header
         gr.HTML("""
         <div class="header">
@@ -622,7 +621,108 @@ def create_gradio_app():
                     type="password",
                     info="Get your API key from: https://console.groq.com/"
                 )
-                
+            with gr.Column(scale=2):
+                model_selection = gr.Radio(
+                    choices=[
+                        "compound-beta",
+                        "compound-beta-mini", 
+                        "openai/gpt-oss-20b",
+                        "llama-3.3-70b-versatile",
+                        "llama-3.1-70b-versatile",
+                        "mixtral-8x7b-32768"
+                    ],
+                    label="🧠 Model Selection",
+                    value="compound-beta",
+                    info="Choose based on your search needs"
+                )
+            with gr.Column(scale=1):
+                connect_btn = gr.Button("🔗 Connect", variant="primary", size="lg")
+        
+        # Status display
+        status_display = gr.Markdown("### 📊 Status: Not connected", elem_classes=["status-box"])
+        
+        # Connect button functionality
+        connect_btn.click(
+            fn=validate_api_key,
+            inputs=[api_key, model_selection],
+            outputs=[status_display]
+        )
+        
+        model_selection.change(
+            fn=update_model,
+            inputs=[model_selection],
+            outputs=[status_display]
+        )
+        
+        # Main Chat Interface
+        with gr.Tab("💬 Chat"):
+            chatbot = gr.Chatbot(
+                label="Creative AI Assistant with Enhanced Search",
+                height=500,
+                show_label=True,
+                bubble_full_width=False,
+                show_copy_button=True
+            )
+            
+            with gr.Row():
+                msg = gr.Textbox(
+                    label="Your Message",
+                    placeholder="Type your message here...",
+                    lines=3
+                )
+                with gr.Column():
+                    send_btn = gr.Button("📤 Send", variant="primary")
+                    clear_btn = gr.Button("🗑️ Clear", variant="secondary")
+            
+        # Search Settings
+        with gr.Accordion("🔍 Search Settings", open=False, elem_id="neuroscope-accordion"):
+            with gr.Row():
+                search_type = gr.Radio(
+                    choices=["auto", "web_search", "browser_search", "none"],
+                    label="🎯 Search Type",
+                    value="auto",
+                    info="Choose search method (auto = model decides)"
+                )
+                force_search = gr.Checkbox(
+                    label="⚡ Force Search",
+                    value=False,
+                    info="Force AI to search even for general questions"
+                )
+            
+            # Update search options when model changes
+            model_selection.change(
+                fn=get_search_options,
+                inputs=[model_selection],
+                outputs=[search_type]
+            )
+            
+        # Domain Filtering Section (only for web search models)
+        with gr.Accordion("🌐 Domain Filtering (Web Search Models Only)", open=False, elem_id="neuroscope-accordion"):
+            gr.Markdown("""
+            <div class="domain-info">
+            <h4>🔍 Domain Filtering Guide</h4>
+            <p><strong>Note:</strong> Domain filtering only works with compound models (compound-beta, compound-beta-mini)</p>
+            <ul>
+                <li><strong>Include Domains:</strong> Only search these domains (comma-separated)</li>
+                <li><strong>Exclude Domains:</strong> Never search these domains (comma-separated)</li>
+                <li><strong>Examples:</strong> arxiv.org, *.edu, github.com, stackoverflow.com</li>
+                <li><strong>Wildcards:</strong> Use *.edu for all educational domains</li>
+            </ul>
+            </div>
+            """)
+            
+            with gr.Row():
+                include_domains = gr.Textbox(
+                    label="✅ Include Domains (comma-separated)",
+                    placeholder="arxiv.org, *.edu, github.com, stackoverflow.com",
+                    info="Only search these domains (compound models only)"
+                )
+                exclude_domains = gr.Textbox(
+                    label="❌ Exclude Domains (comma-separated)", 
+                    placeholder="wikipedia.org, reddit.com, twitter.com",
+                    info="Never search these domains (compound models only)"
+                )
+        
         # Advanced Settings
         with gr.Accordion("⚙️ Advanced Settings", open=False, elem_id="neuroscope-accordion"):
             with gr.Row():
@@ -851,105 +951,4 @@ if __name__ == "__main__":
         share=True,
         server_name="0.0.0.0",
         server_port=7860
-    )
-            with gr.Column(scale=2):
-                model_selection = gr.Radio(
-                    choices=[
-                        "compound-beta",
-                        "compound-beta-mini", 
-                        "openai/gpt-oss-20b",
-                        "llama-3.3-70b-versatile",
-                        "llama-3.1-70b-versatile",
-                        "mixtral-8x7b-32768"
-                    ],
-                    label="🧠 Model Selection",
-                    value="compound-beta",
-                    info="Choose based on your search needs"
-                )
-            with gr.Column(scale=1):
-                connect_btn = gr.Button("🔗 Connect", variant="primary", size="lg")
-        
-        # Status display
-        status_display = gr.Markdown("### 📊 Status: Not connected", elem_classes=["status-box"])
-        
-        # Connect button functionality
-        connect_btn.click(
-            fn=validate_api_key,
-            inputs=[api_key, model_selection],
-            outputs=[status_display]
-        )
-        
-        model_selection.change(
-            fn=update_model,
-            inputs=[model_selection],
-            outputs=[status_display]
-        )
-        
-        # Main Chat Interface
-        with gr.Tab("💬 Chat"):
-            chatbot = gr.Chatbot(
-                label="Creative AI Assistant with Enhanced Search",
-                height=500,
-                show_label=True,
-                bubble_full_width=False,
-                show_copy_button=True
-            )
-            
-            with gr.Row():
-                msg = gr.Textbox(
-                    label="Your Message",
-                    placeholder="Type your message here...",
-                    lines=3
-                )
-                with gr.Column():
-                    send_btn = gr.Button("📤 Send", variant="primary")
-                    clear_btn = gr.Button("🗑️ Clear", variant="secondary")
-            
-        # Search Settings
-        with gr.Accordion("🔍 Search Settings", open=False, elem_id="neuroscope-accordion"):
-            with gr.Row():
-                search_type = gr.Radio(
-                    choices=["auto", "web_search", "browser_search", "none"],
-                    label="🎯 Search Type",
-                    value="auto",
-                    info="Choose search method (auto = model decides)"
-                )
-                force_search = gr.Checkbox(
-                    label="⚡ Force Search",
-                    value=False,
-                    info="Force AI to search even for general questions"
-                )
-            
-            # Update search options when model changes
-            model_selection.change(
-                fn=get_search_options,
-                inputs=[model_selection],
-                outputs=[search_type]
-            )
-            
-        # Domain Filtering Section (only for web search models)
-        with gr.Accordion("🌐 Domain Filtering (Web Search Models Only)", open=False, elem_id="neuroscope-accordion"):
-            gr.Markdown("""
-            <div class="domain-info">
-            <h4>🔍 Domain Filtering Guide</h4>
-            <p><strong>Note:</strong> Domain filtering only works with compound models (compound-beta, compound-beta-mini)</p>
-            <ul>
-                <li><strong>Include Domains:</strong> Only search these domains (comma-separated)</li>
-                <li><strong>Exclude Domains:</strong> Never search these domains (comma-separated)</li>
-                <li><strong>Examples:</strong> arxiv.org, *.edu, github.com, stackoverflow.com</li>
-                <li><strong>Wildcards:</strong> Use *.edu for all educational domains</li>
-            </ul>
-            </div>
-            """)
-            
-            with gr.Row():
-                include_domains = gr.Textbox(
-                    label="✅ Include Domains (comma-separated)",
-                    placeholder="arxiv.org, *.edu, github.com, stackoverflow.com",
-                    info="Only search these domains (compound models only)"
-                )
-                exclude_domains = gr.Textbox(
-                    label="❌ Exclude Domains (comma-separated)", 
-                    placeholder="wikipedia.org, reddit.com, twitter.com",
-                    info="Never search these domains (compound models only)"
-                )
+    )