Spaces:

Papaflessas
/

gotti_signal_gen

Running

App Files Files Community

Papaflessas commited on 6 days ago

Commit

18b3eb0

1 Parent(s): f4ede1d

Deploy Signal Generator app

Browse files

Files changed (5) hide show

.env.example +10 -0
requirements.txt +2 -1
src/db/local_database.py +37 -0
src/main.py +32 -0
src/templates/index.html +352 -0

.env.example ADDED Viewed

	@@ -0,0 +1,10 @@

+# API Security
+API_SECRET=change_this_to_a_secure_random_string
+# Database Configuration (TiDB / MySQL)
+DB_HOST=gateway01.us-west-2.prod.aws.tidbcloud.com
+DB_PORT=4000
+DB_USERNAME=your_username
+DB_PASSWORD=your_password
+DB_DATABASE=gotti
+DB_SSL_CA=src/db/isrgrootx1.pem

requirements.txt CHANGED Viewed

@@ -18,4 +18,5 @@ schedule
 fastapi
 uvicorn
 pydantic
-openai

 fastapi
 uvicorn
 pydantic
+openai
+jinja2

src/db/local_database.py CHANGED Viewed

@@ -843,6 +843,43 @@ class LocalDatabase:
                 cursor.close()
             if 'conn' in locals():
                 conn.close()
     def get(self, date_str: str, data_type: str, ticker: str) -> Optional[DatabaseEntry]:
         """

                 cursor.close()
             if 'conn' in locals():
                 conn.close()
+    def get_recent_signals(self, limit: int = 50) -> List[Dict]:
+        """Get most recent signals"""
+        try:
+            conn = self._create_connection()
+            if not conn:
+                return []
+            cursor = conn.cursor(dictionary=True)
+            cursor.execute('''
+                SELECT * FROM signals
+                ORDER BY created_at DESC
+                LIMIT %s
+            ''', (limit,))
+            results = cursor.fetchall()
+            parsed_results = []
+            for result in results:
+                # Parse JSON fields
+                for field in ['calendar_event_keys', 'news_keys', 'metadata', 'sentiment']:
+                    if result.get(field) and isinstance(result[field], str):
+                        try:
+                            result[field] = json.loads(result[field])
+                        except:
+                            pass
+                parsed_results.append(result)
+            return parsed_results
+        except Error as e:
+            print(f"❌ Error getting recent signals: {e}")
+            return []
+        finally:
+            if 'cursor' in locals():
+                cursor.close()
+            if 'conn' in locals():
+                conn.close()
     def get(self, date_str: str, data_type: str, ticker: str) -> Optional[DatabaseEntry]:
         """

src/main.py CHANGED Viewed

@@ -200,6 +200,38 @@ async def health_check():
     logger.info(f"Health Check: {vitals}")
     return vitals
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)

     logger.info(f"Health Check: {vitals}")
     return vitals
+from fastapi import FastAPI, HTTPException, Header, BackgroundTasks, Depends, Request
+from fastapi.responses import HTMLResponse
+from fastapi.templating import Jinja2Templates
+from fastapi.staticfiles import StaticFiles
+# ... imports ...
+app = FastAPI(title="Stock Alchemist Signal Generator")
+# Setup templates
+templates = Jinja2Templates(directory="src/templates")
+# ... existing code ...
+@app.get("/api/signals", dependencies=[Depends(verify_api_secret)])
+async def get_signals():
+    """Get recent signals"""
+    try:
+        db = LocalDatabase()
+        signals = db.get_recent_signals(limit=50)
+        return signals
+    except Exception as e:
+        logger.error(f"Error fetching signals: {e}")
+        raise HTTPException(status_code=500, detail="Database error")
+@app.get("/", response_class=HTMLResponse)
+async def root(request: Request):
+    """
+    Serve the Home Screen Dashboard
+    """
+    return templates.TemplateResponse("index.html", {"request": request})
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)

src/templates/index.html ADDED Viewed

	@@ -0,0 +1,352 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Stock Alchemist | Signal Generator</title>
+    <link href="https://fonts.googleapis.com/css2?family=Outfit:wght@300;400;500;600;700&display=swap" rel="stylesheet">
+    <style>
+        :root {
+            --bg-color: #0F172A;
+            --card-bg: #1E293B;
+            --text-primary: #F8FAFC;
+            --text-secondary: #94A3B8;
+            --accent-green: #10B981;
+            --accent-red: #EF4444;
+            --accent-yellow: #F59E0B;
+            --accent-blue: #3B82F6;
+        }
+        * { margin: 0; padding: 0; box-sizing: border-box; }
+        body {
+            font-family: 'Outfit', sans-serif;
+            background-color: var(--bg-color);
+            color: var(--text-primary);
+            min-height: 100vh;
+            padding: 2rem;
+        }
+        .container {
+            max-width: 1200px;
+            margin: 0 auto;
+        }
+        header {
+            display: flex;
+            justify-content: space-between;
+            align-items: center;
+            margin-bottom: 2rem;
+            padding-bottom: 1rem;
+            border-bottom: 1px solid rgba(255,255,255,0.1);
+        }
+        h1 { font-weight: 700; font-size: 1.8rem; background: linear-gradient(to right, var(--accent-blue), var(--accent-green)); -webkit-background-clip: text; -webkit-text-fill-color: transparent; }
+        .status-badge {
+            padding: 0.5rem 1rem;
+            border-radius: 20px;
+            font-size: 0.875rem;
+            font-weight: 500;
+            background: rgba(16, 185, 129, 0.2);
+            color: var(--accent-green);
+        }
+        .actions {
+            display: grid;
+            grid-template-columns: repeat(auto-fit, minmax(250px, 1fr));
+            gap: 1rem;
+            margin-bottom: 2rem;
+        }
+        .card {
+            background: var(--card-bg);
+            border-radius: 12px;
+            padding: 1.5rem;
+            box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06);
+            border: 1px solid rgba(255,255,255,0.05);
+        }
+        h2 { font-size: 1.1rem; margin-bottom: 1rem; color: var(--text-secondary); }
+        .btn {
+            display: inline-flex;
+            align-items: center;
+            justify-content: center;
+            width: 100%;
+            padding: 0.75rem 1rem;
+            background: var(--accent-blue);
+            color: white;
+            border: none;
+            border-radius: 8px;
+            font-weight: 600;
+            cursor: pointer;
+            transition: all 0.2s;
+            text-decoration: none;
+            font-size: 0.95rem;
+        }
+        .btn:hover { filter: brightness(1.1); transform: translateY(-1px); }
+        .btn:disabled { opacity: 0.7; cursor: not-allowed; }
+        .btn-green { background: var(--accent-green); }
+        .btn-purple { background: #8B5CF6; }
+        input[type="text"] {
+            width: 100%;
+            padding: 0.75rem;
+            border-radius: 8px;
+            border: 1px solid #334155;
+            background: #0F172A;
+            color: white;
+            margin-bottom: 1rem;
+            font-family: inherit;
+        }
+        input:focus { outline: 2px solid var(--accent-blue); border-color: transparent; }
+        .signals-table {
+            width: 100%;
+            border-collapse: collapse;
+            font-size: 0.95rem;
+        }
+        .signals-table th {
+            text-align: left;
+            padding: 1rem;
+            color: var(--text-secondary);
+            font-weight: 600;
+            border-bottom: 1px solid rgba(255,255,255,0.1);
+        }
+        .signals-table td {
+            padding: 1rem;
+            border-bottom: 1px solid rgba(255,255,255,0.05);
+        }
+        .tag {
+            padding: 0.25rem 0.75rem;
+            border-radius: 12px;
+            font-size: 0.75rem;
+            font-weight: 600;
+            display: inline-block;
+        }
+        .tag-buy { background: rgba(16, 185, 129, 0.2); color: var(--accent-green); }
+        .tag-sell { background: rgba(239, 68, 68, 0.2); color: var(--accent-red); }
+        .tag-hold { background: rgba(245, 158, 11, 0.2); color: var(--accent-yellow); }
+        .empty-state {
+            text-align: center;
+            padding: 3rem;
+            color: var(--text-secondary);
+        }
+        /* Modal for secrets */
+        #secretModal {
+            display: flex;
+            position: fixed; top: 0; left: 0; width: 100%; height: 100%;
+            background: rgba(0,0,0,0.8);
+            justify-content: center; align-items: center;
+            z-index: 100;
+        }
+    </style>
+</head>
+<body>
+<div id="secretModal">
+    <div class="card" style="width: 400px; text-align: center;">
+        <h2>🔐 Enter API Secret</h2>
+        <p style="margin-bottom: 1rem; color: var(--text-secondary); font-size: 0.9rem;">Required to perform actions.</p>
+        <input type="password" id="apiSecretInput" placeholder="API_SECRET" />
+        <button class="btn" onclick="saveSecret()">Access Dashboard</button>
+    </div>
+</div>
+<div class="container" id="mainContent" style="display:none; filter: blur(5px);">
+    <header>
+        <h1>🧪 Stock Alchemist</h1>
+        <div class="status-badge" id="systemStatus">● System Online</div>
+    </header>
+    <div class="actions">
+        <!-- Generate Signal Card -->
+        <div class="card">
+            <h2>⚡ Generate Signal</h2>
+            <input type="text" id="tickerInput" placeholder="Enter Ticker (e.g. AAPL)" />
+            <button class="btn btn-green" onclick="generateSignal()" id="genBtn">
+                Generate AI Signal
+            </button>
+        </div>
+        <!-- System Stats -->
+        <div class="card">
+            <h2>📊 System Status</h2>
+             <div style="display: flex; justify-content: space-between; margin-bottom: 0.5rem;">
+                <span style="color:var(--text-secondary)">Database</span>
+                <span id="dbStatus">Checking...</span>
+            </div>
+            <div style="display: flex; justify-content: space-between; margin-bottom: 1.5rem;">
+                <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>
+    </div>
+    <!-- Recent Signals -->
+    <div class="card">
+        <div style="display:flex; justify-content:space-between; align-items:center; margin-bottom:1rem;">
+            <h2>📡 Recent Signals</h2>
+            <button class="btn" style="width:auto; padding: 0.5rem 1rem; font-size: 0.8rem; background: #334155;" onclick="loadSignals()">Refresh</button>
+        </div>
+        <table class="signals-table">
+            <thead>
+                <tr>
+                    <th>Ticker</th>
+                    <th>Date</th>
+                    <th>Signal</th>
+                    <th>Sentiment</th>
+                    <th>Time (UTC)</th>
+                </tr>
+            </thead>
+            <tbody id="signalsBody">
+                <tr><td colspan="5" class="empty-state">Loading signals...</td></tr>
+            </tbody>
+        </table>
+    </div>
+</div>
+<script>
+    let API_SECRET = localStorage.getItem('api_secret');
+    function saveSecret() {
+        const input = document.getElementById('apiSecretInput').value;
+        if (input) {
+            localStorage.setItem('api_secret', input);
+            API_SECRET = input;
+            document.getElementById('secretModal').style.display = 'none';
+            document.getElementById('mainContent').style.display = 'block';
+            document.getElementById('mainContent').style.filter = 'none';
+            init();
+        }
+    }
+    if (API_SECRET) {
+        document.getElementById('secretModal').style.display = 'none';
+        document.getElementById('mainContent').style.display = 'block';
+        document.getElementById('mainContent').style.filter = 'none';
+        init();
+    }
+    async function init() {
+        checkHealth();
+        loadSignals();
+    }
+    async function checkHealth() {
+        try {
+            const res = await fetch('/health');
+            const data = await res.json();
+            const dbEl = document.getElementById('dbStatus');
+            const ollamaEl = document.getElementById('ollamaStatus');
+            dbEl.innerText = data.database === 'connected' ? 'Connected' : 'Disconnected';
+            dbEl.style.color = data.database === 'connected' ? '#10B981' : '#EF4444';
+            ollamaEl.innerText = data.ollama === 'running' ? 'Active' : 'Offline';
+            ollamaEl.style.color = data.ollama === 'running' ? '#10B981' : '#EF4444';
+        } catch (e) {
+            console.error(e);
+        }
+    }
+    async function loadSignals() {
+        const tbody = document.getElementById('signalsBody');
+        try {
+            const res = await fetch('/api/signals', {
+                headers: { 'X-API-Secret': API_SECRET }
+            });
+            if (!res.ok) throw new Error('Auth failed');
+            const signals = await res.json();
+            if (signals.length === 0) {
+                tbody.innerHTML = '<tr><td colspan="5" class="empty-state">No signals generated yet.</td></tr>';
+                return;
+            }
+            tbody.innerHTML = signals.map(s => {
+                const pos = s.signal_position || 'HOLD';
+                const cls = pos === 'BUY' ? 'tag-buy' : (pos === 'SELL' ? 'tag-sell' : 'tag-hold');
+                const sentiment = s.sentiment ? (s.sentiment.confidence ? `${(s.sentiment.confidence * 100).toFixed(0)}%` : '-') : '-';
+                // Try format date
+                const date = new Date(s.created_at).toLocaleString();
+                return `
+                    <tr>
+                        <td style="font-weight:600; color:white;">${s.ticker}</td>
+                        <td>${s.signal_date}</td>
+                        <td><span class="tag ${cls}">${pos}</span></td>
+                        <td>${sentiment}</td>
+                        <td style="color:var(--text-secondary); font-size: 0.85rem;">${date}</td>
+                    </tr>
+                `;
+            }).join('');
+        } catch (e) {
+            tbody.innerHTML = '<tr><td colspan="5" class="empty-state" style="color:#EF4444">Failed to load signals. Check API Secret.</td></tr>';
+        }
+    }
+    async function generateSignal() {
+        const ticker = document.getElementById('tickerInput').value.toUpperCase();
+        const btn = document.getElementById('genBtn');
+        if (!ticker) return alert('Please enter a ticker');
+        btn.disabled = true;
+        btn.innerText = 'Generating...';
+        try {
+            const res = await fetch('/generate-signal', {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'X-API-Secret': API_SECRET
+                },
+                body: JSON.stringify({ ticker: ticker })
+            });
+            if (res.ok) {
+                alert(`Signal generation started for ${ticker}. Check back in a few seconds.`);
+                setTimeout(loadSignals, 2000);
+            } else {
+                const err = await res.json();
+                alert('Error: ' + err.detail);
+            }
+        } catch (e) {
+            alert('Request failed');
+        } finally {
+            btn.disabled = false;
+            btn.innerText = 'Generate AI Signal';
+        }
+    }
+    async function runAnalysis() {
+        if (!confirm('Run Saturday Analysis? This is a heavy background task.')) return;
+        try {
+             const res = await fetch('/saturday-analysis', {
+                method: 'POST',
+                headers: { 'X-API-Secret': API_SECRET }
+            });
+            alert('Analysis started in background.');
+        } catch (e) {
+            alert('Failed to start analysis');
+        }
+    }
+</script>
+</body>
+</html>