Update app.py

app.py

@@ -633,18 +633,24 @@ User Query: """
         # Check whether the user device is a mobile or a computer
 
         if not request:
-            return "Unknown device"
-        # Method 1: Check sec-ch-ua-mobile header
+            return "Desktop"  # Default to desktop if no request info
+        
+        # Method 1: Check sec-ch-ua-mobile header (most reliable)
         is_mobile_header = request.headers.get("sec-ch-ua-mobile")
         if is_mobile_header:
             return "Mobile" if "?1" in is_mobile_header else "Desktop"
 
         # Method 2: Check user-agent string
         user_agent = request.headers.get("user-agent", "").lower()
-        mobile_keywords = ["android", "iphone", "ipad", "mobile", "phone"]
-
+        mobile_keywords = ["android", "iphone", "ipad", "mobile", "phone", "tablet"]
+        
+        # More comprehensive mobile detection
         if any(keyword in user_agent for keyword in mobile_keywords):
             return "Mobile"
+        
+        # Check for mobile-specific patterns
+        if "mobile" in user_agent or "android" in user_agent or "iphone" in user_agent:
+            return "Mobile"
 
         # Method 3: Check platform
         platform = request.headers.get("sec-ch-ua-platform", "").lower()
@@ -654,11 +660,42 @@ User Query: """
             elif platform in ['"windows"', '"macos"', '"linux"']:
                 return "Desktop"
 
+        # Method 4: Check viewport width (if available)
+        viewport_width = request.headers.get("viewport-width")
+        if viewport_width:
+            try:
+                width = int(viewport_width)
+                return "Mobile" if width <= 768 else "Desktop"
+            except ValueError:
+                pass
+
         # Default case if no clear indicators
         return "Desktop"
 
     def launch(self, **kwargs):
-        with gr.Blocks(theme="ocean", fill_height=True) as demo:
+        # Custom CSS for mobile optimization
+        custom_css = """
+        @media (max-width: 768px) {
+            .gradio-container {
+                max-width: 100% !important;
+                padding: 10px !important;
+            }
+            .main {
+                padding: 10px !important;
+            }
+            .chatbot {
+                max-height: 60vh !important;
+            }
+            .textbox {
+                font-size: 16px !important; /* Prevents zoom on iOS */
+            }
+            .button {
+                min-height: 44px !important; /* Better touch targets */
+            }
+        }
+        """
+        
+        with gr.Blocks(theme="ocean", fill_height=True, css=custom_css) as demo:
             # Different layouts for mobile and computer devices
             @gr.render()
             def layout(request: gr.Request):
@@ -869,18 +906,19 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                             [text_input, launch_research_btn],
                         )
 
-                # Render simple layout
+                # Render simple layout for mobile
                 else:
-                    with gr.Blocks(
-                        fill_height=True,
-                    ):
-                        # Project title and repository link at the top
-                        gr.Markdown("""# Open Deep Research Vulnerability Intelligence""")
-                        gr.Markdown("""<img src="https://github.githubassets.com/images/modules/logos_page/GitHub-Mark.png" width="20" height="20" style="display: inline-block; vertical-align: middle; margin-right: 8px;"> <a href="https://github.com/mcdaqc/open-deep-research-vulnerability-intelligence" target="_blank">Github Repository</a>""")
-                        
-                        # About section for mobile
-                        with gr.Accordion("ℹ️ About", open=False):
-                            gr.Markdown("""**What it does:**
+                    try:
+                        with gr.Blocks(
+                            fill_height=True,
+                        ):
+                            # Project title and repository link at the top
+                            gr.Markdown("""# Open Deep Research Vulnerability Intelligence""")
+                            gr.Markdown("""<img src="https://github.githubassets.com/images/modules/logos_page/GitHub-Mark.png" width="20" height="20" style="display: inline-block; vertical-align: middle; margin-right: 8px;"> <a href="https://github.com/mcdaqc/open-deep-research-vulnerability-intelligence" target="_blank">Github Repository</a>""")
+                            
+                            # About section for mobile
+                            with gr.Accordion("ℹ️ About", open=False):
+                                gr.Markdown("""**What it does:**
 This AI agent specializes in automated vulnerability research and analysis, built on <a href="https://huggingface.co/blog/open-deep-research" target="_blank">Hugging Face's Open Deep Research</a> architecture. It can search across multiple security databases to provide comprehensive vulnerability intelligence reports.
 
 **Available Tools & APIs:**
@@ -892,7 +930,13 @@ This AI agent specializes in automated vulnerability research and analysis, buil
 
 **Model Configuration:**
 - **Default Model**: Qwen/Qwen2.5-Coder-32B-Instruct (recommended)
-- **Alternative**: You can also use Ollama with local models for privacy
+- **Local Models**: For privacy, you can use Ollama with local models:
+  1. Install Ollama: https://ollama.ai/
+  2. Pull a model: `ollama pull qwen2.5-coder:7b`
+  3. Set in your `.env` file:
+     - `MODEL_ID=qwen2.5-coder:7b`
+     - `OPENAI_API_BASE=http://localhost:11434/v1`
+     - `OPENAI_API_KEY=ollama`
 
 **How to use:**
 1. Enter your Hugging Face API key below
@@ -945,44 +989,7 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                             scale=1,
                         )
                         
-                        # Mobile API Key setup event
-                        mobile_setup_api_btn.click(
-                            self.setup_api_key,
-                            [mobile_api_key_input, mobile_max_steps_slider],
-                            [mobile_api_key_status]
-                        )
-                        
-
-                        
-                        # Mobile Example button events
-                        mobile_example_btn_1 = gr.Button("🔍 MobaXterm 24.0 vulnerabilities", size="sm", variant="secondary")
-                        mobile_example_btn_2 = gr.Button("🔍 Chrome 120.0.6099.109 security analysis", size="sm", variant="secondary")
-                        mobile_example_btn_3 = gr.Button("🔍 Apache Tomcat 9.0.65 KEV check", size="sm", variant="secondary")
-                        
-                        # Mobile Example button events
-                        mobile_example_btn_1.click(
-                            lambda: "Analyze MobaXterm 24.0 for vulnerabilities as of today",
-                            None,
-                            [text_input]
-                        )
-                        mobile_example_btn_2.click(
-                            lambda: "Check Chrome 120.0.6099.109 for current security issues",
-                            None,
-                            [text_input]
-                        )
-                        mobile_example_btn_3.click(
-                            lambda: "Is Apache Tomcat 9.0.65 in KEV database as of today?",
-                            None,
-                            [text_input]
-                        )
-
-                        # Powered by smolagents for mobile
-                        with gr.Row():
-                            gr.HTML("""<div style="display: flex; align-items: center; gap: 8px; font-family: system-ui, -apple-system, sans-serif;">Powered by
-                        <img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/smolagents/mascot_smol.png" style="width: 32px; height: 32px; object-fit: contain;" alt="logo">
-                        <a target="_blank" href="https://github.com/huggingface/smolagents"><b>hf/smolagents</b></a>
-                        </div>""")
-
+                        # Input section for mobile
                         text_input = gr.Textbox(
                             lines=1,
                             label="Your request",
@@ -1001,6 +1008,20 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                             mobile_example_btn_2 = gr.Button("🔍 Chrome 120.0.6099.109 security analysis", size="sm", variant="secondary")
                             mobile_example_btn_3 = gr.Button("🔍 Apache Tomcat 9.0.65 KEV check", size="sm", variant="secondary")
                         
+                        # Powered by smolagents for mobile
+                        with gr.Row():
+                            gr.HTML("""<div style="display: flex; align-items: center; gap: 8px; font-family: system-ui, -apple-system, sans-serif;">Powered by
+                        <img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/smolagents/mascot_smol.png" style="width: 32px; height: 32px; object-fit: contain;" alt="logo">
+                        <a target="_blank" href="https://github.com/huggingface/smolagents"><b>hf/smolagents</b></a>
+                        </div>""")
+
+                        # Mobile API Key setup event
+                        mobile_setup_api_btn.click(
+                            self.setup_api_key,
+                            [mobile_api_key_input, mobile_max_steps_slider],
+                            [mobile_api_key_status]
+                        )
+                        
                         # Mobile Example button events
                         mobile_example_btn_1.click(
                             lambda: "Analyze MobaXterm 24.0 for vulnerabilities as of today",
@@ -1018,13 +1039,7 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                             [text_input]
                         )
 
-                        # Powered by smolagents for mobile
-                        with gr.Row():
-                            gr.HTML("""<div style="display: flex; align-items: center; gap: 8px; font-family: system-ui, -apple-system, sans-serif;">Powered by
-                        <img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/smolagents/mascot_smol.png" style="width: 32px; height: 32px; object-fit: contain;" alt="logo">
-                        <a target="_blank" href="https://github.com/huggingface/smolagents"><b>hf/smolagents</b></a>
-                        </div>""")
-
+                        # Mobile chat events
                         text_input.submit(
                             self.log_user_message,
                             [text_input, file_uploads_log],
@@ -1063,10 +1078,62 @@ This AI agent specializes in automated vulnerability research and analysis, buil
                             None,
                             [text_input, launch_research_btn],
                         )
-
+                    except Exception as e:
+                        # Fallback to desktop layout if mobile layout fails
+                        logger.error(f"Mobile layout failed: {e}")
+                        # Re-render desktop layout as fallback
+                        with gr.Blocks(fill_height=True):
+                            gr.Markdown("""# Open Deep Research Vulnerability Intelligence""")
+                            gr.Markdown("""<img src="https://github.githubassets.com/images/modules/logos_page/GitHub-Mark.png" width="20" height="20" style="display: inline-block; vertical-align: middle; margin-right: 8px;"> <a href="https://github.com/mcdaqc/open-deep-research-vulnerability-intelligence" target="_blank">Github Repository</a>""")
+                            gr.Markdown("⚠️ Mobile layout failed, using desktop layout as fallback.")
+                            
+                            # Simple fallback interface
+                            stored_messages = gr.State([])
+                            file_uploads_log = gr.State([])
+                            chatbot = gr.Chatbot(
+                                label="open-Deep-Research",
+                                type="messages",
+                                avatar_images=(
+                                    None,
+                                    "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/smolagents/mascot_smol.png",
+                                ),
+                            )
+                            
+                            text_input = gr.Textbox(
+                                lines=1,
+                                label="Your request",
+                                placeholder="Enter your prompt here and press the button",
+                            )
+                            launch_research_btn = gr.Button("Run", variant="primary")
+                            
+                            # Fallback events
+                            text_input.submit(
+                                self.log_user_message,
+                                [text_input, file_uploads_log],
+                                [stored_messages, text_input, launch_research_btn],
+                            ).then(
+                                self.interact_with_agent,
+                                [stored_messages, chatbot],
+                                [chatbot],
+                            )
+                            launch_research_btn.click(
+                                self.log_user_message,
+                                [text_input, file_uploads_log],
+                                [stored_messages, text_input, launch_research_btn],
+                            ).then(
+                                self.interact_with_agent,
+                                [stored_messages, chatbot],
+                                [chatbot],
+                            )
 
         
-        demo.launch(debug=True, **kwargs)
+        demo.launch(
+            debug=True,
+            server_name="0.0.0.0",
+            server_port=7860,
+            share=False,
+            **kwargs
+        )
 
 # can this fix ctrl-c no response? no
 try: