Deploy Signal Generator app

src/main.py

@@ -225,6 +225,110 @@ async def get_signals():
         logger.error(f"Error fetching signals: {e}")
         raise HTTPException(status_code=500, detail="Database error")
 
+@app.post("/test-db", dependencies=[Depends(verify_api_secret)])
+async def test_db_connection():
+    """
+    Detailed database connection test
+    """
+    results = {
+        "status": "failed",
+        "details": [],
+        "config": {}
+    }
+    
+    try:
+        # Check env vars (redact password)
+        import os
+        results["config"] = {
+            "host": os.getenv('DB_HOST'),
+            "port": os.getenv('DB_PORT'),
+            "user": os.getenv('DB_USERNAME'),
+            "database": os.getenv('DB_DATABASE'),
+            "ssl_ca_set": bool(os.getenv('DB_SSL_CA'))
+        }
+        
+        db = LocalDatabase()
+        conn = db._create_connection()
+        
+        if conn and conn.is_connected():
+            results["status"] = "success"
+            results["details"].append("Connection successful")
+            results["details"].append(f"Server Info: {conn.get_server_info()}")
+            
+            cursor = conn.cursor()
+            cursor.execute("SELECT VERSION()")
+            version = cursor.fetchone()
+            results["details"].append(f"DB Version: {version[0]}")
+            conn.close()
+        else:
+            results["details"].append("Connection object create but is_connected() returned False")
+            
+    except Exception as e:
+        results["details"].append(f"Exception: {str(e)}")
+        import traceback
+        results["traceback"] = traceback.format_exc()
+        
+    return results
+
+@app.post("/test-ollama", dependencies=[Depends(verify_api_secret)])
+async def test_ollama_connection():
+    """
+    Test Ollama connectivity and model status
+    """
+    import requests
+    results = {
+        "status": "failed",
+        "details": [],
+        "model_found": False
+    }
+    
+    try:
+        # 1. Check if Service is Up
+        base_url = "http://localhost:11434"
+        try:
+            resp = requests.get(f"{base_url}/api/tags", timeout=5)
+            if resp.status_code == 200:
+                results["details"].append("Ollama Service is UP")
+                models = resp.json().get('models', [])
+                model_names = [m.get('name') for m in models]
+                results["details"].append(f"Available Models: {', '.join(model_names)}")
+                
+                if any("llama3.1" in m for m in model_names):
+                    results["model_found"] = True
+                else:
+                    results["details"].append("WARNING: llama3.1 model not found in list!")
+            else:
+                results["details"].append(f"Service returned status {resp.status_code}")
+                return results
+        except Exception as e:
+            results["details"].append(f"Failed to connect to Ollama Service: {e}")
+            return results
+
+        # 2. Test Generation (if service is up)
+        if results["model_found"]:
+            try:
+                payload = {
+                    "model": "llama3.1",
+                    "prompt": "hi",
+                    "stream": False
+                }
+                resp = requests.post(f"{base_url}/api/generate", json=payload, timeout=10)
+                if resp.status_code == 200:
+                    ans = resp.json().get('response', '')
+                    results["details"].append(f"Generation Test Pass: '{ans[:20]}...'")
+                    results["status"] = "success"
+                else:
+                    results["details"].append(f"Generation Failed: {resp.text}")
+            except Exception as e:
+                 results["details"].append(f"Generation Error: {e}")
+        else:
+             results["details"].append("Skipping generation test as model not found.")
+
+    except Exception as e:
+        results["details"].append(f"Unexpected Error: {e}")
+        
+    return results
+
 @app.get("/", response_class=HTMLResponse)
 async def root(request: Request):
     """

src/templates/index.html

@@ -188,9 +188,17 @@
                 <span style="color:var(--text-secondary)">Ollama</span>
                 <span id="ollamaStatus">Checking...</span>
             </div>
-            <button class="btn btn-purple" onclick="runAnalysis()">
-                Run Weekly Analysis
-            </button>
+            <div style="display: flex; gap: 0.5rem;">
+                <button class="btn btn-purple" onclick="runAnalysis()" style="flex: 1;">
+                    Run Analysis
+                </button>
+                <button class="btn" onclick="testDb()" style="flex: 1; background: #475569;">
+                    Test DB
+                </button>
+                <button class="btn" onclick="testOllama()" style="flex: 1; background: #ea580c;">
+                    Test LLM
+                </button>
+            </div>
         </div>
     </div>
 
@@ -347,6 +355,64 @@
             alert('Failed to start analysis');
         }
     }
+
+    async function testDb() {
+        const btn = event.target;
+        const ogText = btn.innerText;
+        btn.innerText = 'Testing...';
+        btn.disabled = true;
+        
+        try {
+            const res = await fetch('/test-db', {
+                method: 'POST',
+                headers: { 'X-API-Secret': API_SECRET }
+            });
+            const data = await res.json();
+            
+            let msg = `Status: ${data.status.toUpperCase()}\n\n`;
+            if (data.details) msg += data.details.join('\n');
+            if (data.config) msg += `\n\nHost: ${data.config.host}\nSSL Set: ${data.config.ssl_ca_set}`;
+            
+            alert(msg);
+            
+            // Refresh health check UI
+            checkHealth();
+            
+        } catch (e) {
+            alert('Test request failed: ' + e);
+        } finally {
+            btn.innerText = ogText;
+            btn.disabled = false;
+        }
+    }
+
+    async function testOllama() {
+        const btn = event.target;
+        const ogText = btn.innerText;
+        btn.innerText = 'Testing...';
+        btn.disabled = true;
+        
+        try {
+            const res = await fetch('/test-ollama', {
+                method: 'POST',
+                headers: { 'X-API-Secret': API_SECRET }
+            });
+            const data = await res.json();
+            
+            let msg = `Status: ${data.status.toUpperCase()}\n\n`;
+            if (data.details) msg += data.details.join('\n');
+            
+            alert(msg);
+            
+            checkHealth();
+            
+        } catch (e) {
+            alert('Test request failed: ' + e);
+        } finally {
+            btn.innerText = ogText;
+            btn.disabled = false;
+        }
+    }
 </script>
 </body>
 </html>