Create AdminDashboard.py

AdminDashboard.py

@@ -0,0 +1,65 @@
+import streamlit as st
+import boto3
+import os
+from datetime import datetime
+
+# AWS DynamoDB config
+dynamodb = boto3.client(
+    "dynamodb",
+    region_name=os.getenv("DYNAMODB_REGION", "ap-south-1"),
+    aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),
+    aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),
+)
+
+TABLE_NAME = os.getenv("SESSION_TABLE", "SessionTracking")
+
+# Page Config
+st.set_page_config(
+    page_title="📊 Admin Dashboard - Base Videos",
+    layout="wide",
+    initial_sidebar_state="expanded",
+)
+
+st.markdown("<h1 style='color:#4F97FF;text-align:center;'>📊 Base Video Generation Dashboard</h1>", unsafe_allow_html=True)
+
+# Fetch all sessions
+def fetch_sessions():
+    try:
+        response = dynamodb.scan(TableName=TABLE_NAME)
+        items = response.get("Items", [])
+        sessions = []
+        for item in items:
+            sessions.append({
+                "session_id": item.get("session_id", {}).get("S"),
+                "status": item.get("status", {}).get("S", "unknown"),
+                "node": item.get("node", {}).get("S", "-"),
+                "updated_at": item.get("updated_at", {}).get("S", "-"),
+                "video_url": item.get("video_url", {}).get("S", None),  # optional if final node adds it
+            })
+        return sessions
+    except Exception as e:
+        st.error(f"⚠️ Error fetching sessions: {e}")
+        return []
+
+# Auto-refresh every 15 seconds
+st_autorefresh = st.empty()
+sessions = fetch_sessions()
+
+if not sessions:
+    st.warning("No sessions found yet.")
+else:
+    for s in sessions:
+        with st.container():
+            st.markdown(f"""
+            <div style="background:#2D2D2D;padding:1rem;border-radius:10px;margin-bottom:1rem;border:1px solid #444;">
+                <b>🆔 Session:</b> <code>{s['session_id']}</code><br>
+                <b>🔄 Current Node:</b> {s['node']}<br>
+                <b>📌 Status:</b> <span style="color:{'lime' if s['status']=='completed' else 'orange'};">{s['status']}</span><br>
+                <b>⏱️ Last Update:</b> {s['updated_at']}<br>
+                {"<b>🎥 Final Video:</b> <a href='"+s['video_url']+"' target='_blank'>Watch here</a>" if s['video_url'] else ""}
+            </div>
+            """, unsafe_allow_html=True)
+
+# Manual refresh button
+if st.button("🔄 Refresh Now"):
+    st.rerun()