Update main.py

main.py

@@ -84,12 +84,41 @@ claude_3_models = {
 }
 
 gemini_models = {
-    "gemini-1.5-pro", "gemini-1.5-flash", "gemini-2.0-flash-lite-preview",
-    "gemini-2.0-flash", "gemini-2.0-flash-thinking",
-    "gemini-2.0-flash-preview-image-generation", "gemini-2.5-flash",
-    "gemini-2.5-pro-exp", "gemini-exp-1206"
+    # Gemini 1.5 Series
+    "gemini-1.5-pro",            # Alias for latest stable 1.5 Pro[4][5]
+    "gemini-1.5-pro-002",        # Latest 1.5 Pro stable[4][5]
+    "gemini-1.5-flash",          # Alias for latest stable 1.5 Flash[4][5]
+    "gemini-1.5-flash-002",      # Latest 1.5 Flash stable[4][5]
+    "gemini-1.5-flash-8b",       # 1.5 Flash 8B variant[1]
+
+    # Gemini 2.0 Series
+    "gemini-2.0-flash-lite",     # Cost-efficient model[5]
+    "gemini-2.0-flash-lite-preview", # Preview version[1]
+    "gemini-2.0-flash",          # Default model as of Jan 2025[5]
+    "gemini-2.0-flash-exp",      # Experimental version[1]
+    "gemini-2.0-flash-thinking", # Exposes model reasoning[5]
+    "gemini-2.0-flash-thinking-exp-01-21", # Experimental thinking[1]
+    "gemini-2.0-flash-preview-image-generation", # Image generation[1]
+    "gemini-2.0-pro-exp-02-05", # 2.0 Pro Experimental[1]
+
+    # Gemini 2.5 Series
+    "gemini-2.5-flash",          # Default model as of May 2025[5][7]
+    "gemini-2.5-flash-preview-05-20", # 2.5 Flash preview[3]
+    "gemini-2.5-flash-preview-native-audio-dialog", # Native audio output[3]
+    "gemini-2.5-flash-exp-native-audio-thinking-dialog", # Audio thinking[3]
+    "gemini-2.5-pro",            # 2.5 Pro (active, most advanced)[6][7]
+    "gemini-2.5-pro-preview-06-05", # Latest 2.5 Pro preview[3]
+    "gemini-2.5-pro-preview-03-25", # 2.5 Pro preview[1][3]
+    "gemini-2.5-pro-exp-03-25",  # 2.5 Pro experimental[3]
+    "gemini-2.5-pro-preview-tts", # Speech generation[3]
+    "gemini-2.5-flash-preview-tts", # Speech generation[3]
+
+    # Experimental and Special Models
+    "gemini-exp-1206",           # Experimental 2.0 Pro[1]
+    "gemini-embedding-exp-03-07",# Experimental embeddings[3]
 }
 
+
 supported_image_models = {
     "Flux Pro Ultra", "grok-2-aurora", "Flux Pro", "Flux Pro Ultra Raw",
     "Flux Dev", "Flux Schnell", "stable-diffusion-3-large-turbo",
@@ -666,6 +695,12 @@ async def create_image(payload: ImageGenerationPayload, request: Request, authen
 async def get_usage_json(days: int = 7):
     return usage_tracker.get_usage_summary(days)
 
+@app.get("/usage/page", response_class=HTMLResponse)
+async def get_usage_page(days: int = Query(7, description="Number of days to include in the usage summary")):
+    usage_data = usage_tracker.get_usage_summary(days)
+    html_content = generate_usage_html(usage_data, days)
+    return HTMLResponse(content=html_content)
+
 def generate_usage_html(usage_data: dict, days: int = 7):
     model_labels = list(usage_data['model_usage_period'].keys())
     model_counts = list(usage_data['model_usage_period'].values())
@@ -1244,4 +1279,3 @@ async def health_check():
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)
-