Update fastapi_app.py

fastapi_app.py

@@ -1,42 +1,169 @@
-from fastapi import FastAPI, HTTPException, Depends
+from fastapi import FastAPI, HTTPException, Depends, Request
+
+from fastapi.staticfiles import StaticFiles
+
+from fastapi.templating import Jinja2Templates
+
 from pydantic import BaseModel, Field
+
 from bot import Bot, temp
+
 from database.users_chats_db import db
+
 from database.ia_filterdb import Media, get_file_details, get_search_results
+
 from database.connections_mdb import add_connection, active_connection, all_connections, if_active, make_active, make_inactive, delete_connection
+
 from database.filters_mdb import add_filter, find_filter, get_filters, delete_filter, del_all, count_filters, filter_stats
+
 from database.gfilters_mdb import add_gfilter, find_gfilter, get_gfilters, delete_gfilter, del_allg, count_gfilters, gfilter_stats
+
 from utils import get_settings, save_group_settings, humanbytes, get_size, extract_user, get_file_id, get_poster, parser, get_shortlink, extract_time, admin_check
+
 from info import API_ID, API_HASH, BOT_TOKEN, LOG_CHANNEL, UPTIME, WEB_SUPPORT, LOG_MSG, ADMINS, CHANNELS
+
 from pyrogram import Client, filters, enums, types
+
 from pyrogram.errors import ChatAdminRequired, UserIsBlocked, MessageNotModified, PeerIdInvalid, FloodWait
+
 from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, Message, CallbackQuery
+
 from telegraph import upload_file
+
 from io import BytesIO
+
 import aiohttp
+
 import asyncio
+
 import requests
+
 import os
+
 import logging
+
 import re
+
 import traceback
+
 import json
+
 from googletrans import Translator
+
 from gtts import gTTS
+
 from youtube_search import YoutubeSearch
+
 from youtubesearchpython import SearchVideos
+
 from yt_dlp import YoutubeDL
+
 import shutil
+
 import psutil
+
 import wget
+
 from datetime import datetime, timedelta
+
 from urllib.parse import quote
+
 from telegraph import upload_file
+
 from PIL import Image, ImageDraw, ImageFont
+
 import textwrap
 
+
+# Initialize FastAPI
+
 app = FastAPI()
 
+
+# Configure templates and static files
+
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+templates = Jinja2Templates(directory="templates")
+
+
+# Shared instance of the Bot class
+
+bot_instance = Bot()
+
+
+# Initialize Pyrogram client
+
+@app.on_event("startup")
+
+async def startup_event():
+
+    asyncio.create_task(bot_instance.start())
+
+
+@app.on_event("shutdown")
+
+async def shutdown_event():
+
+    await bot_instance.stop()
+
+
+# Dashboard Route
+
+@app.get("/", include_in_schema=False)
+
+async def dashboard(request: Request):
+
+    try:
+
+        # Get statistics
+
+        total_users = await db.total_users_count()
+
+        total_chats = await db.total_chat_count()
+
+        uptime = datetime.now() - datetime.fromtimestamp(temp.UPTIME)
+
+        
+
+        # System metrics
+
+        cpu = psutil.cpu_percent()
+
+        mem = psutil.virtual_memory().percent
+
+        disk = psutil.disk_usage('/').percent
+
+
+        return templates.TemplateResponse("dashboard.html", {
+
+            "request": request,
+
+            "total_users": total_users,
+
+            "total_chats": total_chats,
+
+            "uptime": str(uptime).split('.')[0],
+
+            "cpu_usage": f"{cpu}%",
+
+            "mem_usage": f"{mem}%",
+
+            "disk_usage": f"{disk}%",
+
+            "bot_username": (await bot_instance.get_me()).username,
+
+            "version": "PROFESSOR-BOT v4.5.0"
+
+        })
+
+    except Exception as e:
+
+        logging.error(f"Dashboard Error: {str(e)}")
+
+        return templates.TemplateResponse("error.html", {"request": request})
+
+
 # Shared instance of the Bot class
 bot_instance = Bot()