Deployment from Replit

app.py

@@ -0,0 +1,146 @@
+"""
+CyberForge Dashboard - Hugging Face Spaces Version
+"""
+import os
+import sys
+import streamlit as st
+from streamlit_extras.add_vertical_space import add_vertical_space
+from streamlit_extras.colored_header import colored_header
+
+# Check if we're running on Hugging Face Spaces
+is_huggingface = os.environ.get('SPACE_ID') is not None
+
+# Set the page config
+st.set_page_config(
+    page_title="CyberForge Dashboard",
+    page_icon="🕵️‍♂️",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# Add custom CSS
+st.markdown("""
+<style>
+    .stApp {
+        background-color: #0e1117;
+    }
+    .sidebar .sidebar-content {
+        background-color: #262730;
+    }
+    h1, h2, h3 {
+        color: #f8f9fa;
+    }
+    .cybertext {
+        color: #00ff8d;
+        font-family: monospace;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+# Choose between HF demo mode or regular mode
+if is_huggingface:
+    # Initialize in-memory database for Hugging Face
+    import hf_database
+    st.session_state.is_demo = True
+    
+    # Show demo mode banner
+    st.warning("⚠️ Running in Hugging Face Spaces DEMO MODE. Data is stored in-memory and will be reset when the space restarts.")
+else:
+    # Regular database initialization
+    import src.database_init
+    
+# Import components
+from components.dashboard import render_dashboard
+from components.threats import render_threats
+from components.monitoring import render_monitoring
+from components.alerts import render_alerts
+from components.reports import render_reports
+from components.live_feed import render_live_feed, render_content_analysis
+from components.web_scraper import render_web_scraper_ui
+
+# Custom notification function
+def add_notification(title, message, severity="info", icon="🔔"):
+    """Add a notification to the session state"""
+    if "notifications" not in st.session_state:
+        st.session_state.notifications = []
+    
+    # Add notification with timestamp
+    import time
+    notification = {
+        "id": int(time.time() * 1000),
+        "title": title,
+        "message": message,
+        "severity": severity,
+        "icon": icon,
+        "read": False,
+        "timestamp": time.time()
+    }
+    st.session_state.notifications.insert(0, notification)
+
+# Initialize notifications if needed
+if "notifications" not in st.session_state:
+    st.session_state.notifications = []
+
+# Sidebar navigation
+with st.sidebar:
+    st.image("assets/cyberforge_logo.svg", width=200)
+    st.title("CyberForge")
+    
+    # Demo badge
+    if st.session_state.get("is_demo", False):
+        st.markdown("#### 🔍 Demo Mode")
+    
+    st.markdown("---")
+    
+    # Navigation
+    nav_selection = st.radio(
+        "Navigation",
+        ["Dashboard", "Threats", "Monitoring", "Alerts", "Reports", "Live Feed", "Content Analysis", "Web Scraper"]
+    )
+    
+    # User information
+    st.markdown("---")
+    st.markdown("### User Info")
+    if st.session_state.get("is_demo", False):
+        st.markdown("👤 **Admin User** (Demo)")
+        st.markdown("🔑 Role: Administrator")
+    else:
+        st.markdown("👤 **Analyst**")
+        st.markdown("🔑 Role: Security Analyst")
+    
+    # Notification count
+    unread_count = sum(1 for n in st.session_state.notifications if not n["read"])
+    if unread_count > 0:
+        st.markdown(f"🔔 **{unread_count}** unread notifications")
+    
+    # Credits
+    st.markdown("---")
+    st.markdown("### CyberForge v1.0")
+    st.markdown("© 2025 Chemically Motivated Solutions")
+    
+    # HF badge if on Hugging Face
+    if is_huggingface:
+        st.markdown("---")
+        st.markdown("""
+        <a href="https://huggingface.co/spaces" target="_blank">
+        <img src="https://img.shields.io/badge/Hosted%20on-HF%20Spaces-blue" alt="HuggingFace Spaces"/>
+        </a>
+        """, unsafe_allow_html=True)
+
+# Main content area
+if nav_selection == "Dashboard":
+    render_dashboard()
+elif nav_selection == "Threats":
+    render_threats()
+elif nav_selection == "Monitoring":
+    render_monitoring()
+elif nav_selection == "Alerts":
+    render_alerts()
+elif nav_selection == "Reports":
+    render_reports()
+elif nav_selection == "Live Feed":
+    render_live_feed()
+elif nav_selection == "Content Analysis":
+    render_content_analysis()
+elif nav_selection == "Web Scraper":
+    render_web_scraper_ui()