Spaces:

justadri23
/

Choru25

Paused

App Files Files Community

justadri23 commited on Mar 26, 2025

Commit

54cf7c2

verified ·

1 Parent(s): 38a9b9a

Upload app.py

Browse files

Files changed (1) hide show

app.py +123 -48

app.py CHANGED Viewed

@@ -24,18 +24,33 @@ active_keys = []
 revoked_keys = []
 def verify_api_key(api_key):
-    """Verifica si una clave API de Gemini es válida."""
     try:
         # Configurar temporalmente con esta clave
         genai.configure(api_key=api_key)
-        # Intentar una solicitud simple
         model = genai.GenerativeModel("gemini-pro")
-        response = model.generate_content("Hello, this is a test.")
         # Si llegamos aquí, la clave es válida
         return True
     except Exception as e:
         logger.warning(f"Clave API inválida: {str(e)}")
         return False
@@ -52,14 +67,19 @@ def initialize_api_keys():
     logger.info(f"Verificando {len(api_keys)} claves API...")
-    # Verificar cada clave API
-    for key in api_keys:
         if verify_api_key(key):
             active_keys.append(key)
-            logger.info(f"Clave API verificada con éxito")
         else:
             revoked_keys.append(key)
-            logger.warning(f"Clave API inválida o revocada")
     if not active_keys:
         logger.warning("No hay claves API válidas disponibles. El servidor se iniciará pero los endpoints no funcionarán.")
@@ -87,7 +107,7 @@ def get_current_api_key():
     return current_key
-# Plantilla HTML para la página de estado
 STATUS_PAGE_TEMPLATE = """
 <!DOCTYPE html>
 <html>
@@ -95,62 +115,117 @@ STATUS_PAGE_TEMPLATE = """
     <title>Gemini Proxy Status</title>
     <style>
         body {
-            font-family: Arial, sans-serif;
-            max-width: 800px;
-            margin: 0 auto;
             padding: 20px;
         }
-        h1 {
-            color: #333;
         }
-        .status-card {
-            background-color: #f5f5f5;
-            border-radius: 8px;
-            padding: 20px;
-            margin-top: 20px;
-            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
         }
-        .status-info {
-            font-family: monospace;
-            font-size: 16px;
-            white-space: pre;
-            padding: 15px;
-            background-color: #fff;
-            border-radius: 4px;
-            border: 1px solid #ddd;
         }
-        .status-warning {
-            margin-top: 15px;
-            padding: 15px;
-            background-color: #fff3cd;
-            border-radius: 4px;
-            border: 1px solid #ffeeba;
-            color: #856404;
         }
-        .footer {
-            margin-top: 30px;
-            color: #666;
-            font-size: 14px;
         }
     </style>
 </head>
 <body>
-    <h1>Gemini Proxy Status</h1>
-    <div class="status-card">
-        <div class="status-info">Gemini Status: {
   activeKeys: {{ active_keys }},
   revokedKeys: {{ revoked_keys }}
 }</div>
         {% if active_keys == 0 %}
-        <div class="status-warning">
-            <strong>⚠️ Advertencia:</strong> No hay claves API activas disponibles.
-            Los endpoints no funcionarán hasta que configures al menos una clave API válida de Gemini.
-            Edita el archivo .env y añade tus claves API.
-        </div>
         {% endif %}
     </div>
-    <div class="footer">
-        <p>Proxy funcionando en {{ host }}:{{ port }}</p>
     </div>
 </body>
 </html>

 revoked_keys = []
 def verify_api_key(api_key):
+    """Verifica si una clave API de Gemini es válida de manera moderada para evitar rate limits."""
     try:
         # Configurar temporalmente con esta clave
         genai.configure(api_key=api_key)
+        # Intentar una solicitud simple y ligera
+        # Usamos una solicitud mínima para evitar consumir cuota innecesariamente
         model = genai.GenerativeModel("gemini-pro")
+        response = model.generate_content("test",
+                                         generation_config={
+                                             "max_output_tokens": 5,
+                                             "temperature": 0
+                                         })
         # Si llegamos aquí, la clave es válida
+        # Esperamos un breve momento para evitar rate limits
+        time.sleep(1)
         return True
     except Exception as e:
+        error_str = str(e).lower()
+        # Detectar específicamente errores de rate limit
+        if "rate limit" in error_str or "quota" in error_str:
+            logger.warning(f"Rate limit detectado durante verificación: {str(e)}")
+            # Asumimos que la clave es válida pero está limitada temporalmente
+            time.sleep(2)  # Esperamos más tiempo en caso de rate limit
+            return True
         logger.warning(f"Clave API inválida: {str(e)}")
         return False
     logger.info(f"Verificando {len(api_keys)} claves API...")
+    # Verificar cada clave API con una pausa entre verificaciones
+    for i, key in enumerate(api_keys):
+        logger.info(f"Verificando clave API {i+1} de {len(api_keys)}...")
         if verify_api_key(key):
             active_keys.append(key)
+            logger.info(f"Clave API {i+1} verificada con éxito")
         else:
             revoked_keys.append(key)
+            logger.warning(f"Clave API {i+1} inválida o revocada")
+        # Pausa entre verificaciones para evitar rate limits
+        if i < len(api_keys) - 1:
+            time.sleep(2)
     if not active_keys:
         logger.warning("No hay claves API válidas disponibles. El servidor se iniciará pero los endpoints no funcionarán.")
     return current_key
+# Plantilla HTML para la página de estado con estilo de terminal
 STATUS_PAGE_TEMPLATE = """
 <!DOCTYPE html>
 <html>
     <title>Gemini Proxy Status</title>
     <style>
         body {
+            background-color: #111;
+            color: #0f0;
+            font-family: 'Courier New', monospace;
             padding: 20px;
+            max-width: 900px;
+            margin: 0 auto;
+            line-height: 1.4;
         }
+        a {
+            color: #0f0;
+            text-decoration: underline;
         }
+        .terminal-header {
+            margin-bottom: 20px;
+            border-bottom: 1px dashed #0f0;
+            padding-bottom: 10px;
+        }
+        .terminal-prompt::before {
+            content: "$ ";
+        }
+        .terminal-prompt {
+            white-space: nowrap;
+            overflow: hidden;
+            border-right: 5px solid #0f0;
+            animation: terminal-cursor 1s step-end infinite;
+        }
+        @keyframes terminal-cursor {
+            0%, 100% { border-color: #0f0; }
+            50% { border-color: transparent; }
+        }
+        .terminal-output {
+            margin: 15px 0;
+            padding-left: 15px;
+            border-left: 2px solid #0a0;
+            white-space: pre-wrap;
+        }
+        .terminal-section {
+            margin-bottom: 25px;
+        }
+        .warning {
+            color: #ff0;
         }
+        .error {
+            color: #f00;
         }
+        .url {
+            text-decoration: underline;
+            color: #0ff;
         }
+        .method {
+            font-weight: bold;
+            color: #f0f;
+        }
+        .comment {
+            color: #777;
+            font-style: italic;
         }
     </style>
 </head>
 <body>
+    <div class="terminal-header">
+        <h1 style="margin-top:0;">GEMINI PROXY</h1>
+    </div>
+    <div class="terminal-section">
+        <div class="terminal-prompt">status</div>
+        <div class="terminal-output">{
+  host: "{{ host }}:{{ port }}",
   activeKeys: {{ active_keys }},
   revokedKeys: {{ revoked_keys }}
 }</div>
         {% if active_keys == 0 %}
+        <div class="terminal-output error">[ERROR] No hay claves API activas. Los endpoints no funcionarán.</div>
         {% endif %}
     </div>
+    <div class="terminal-section">
+        <div class="terminal-prompt">endpoints --list</div>
+        <div class="terminal-output"><span class="comment"># Compatible con OpenAI</span>
+<span class="method">POST</span>   http://{{ host }}:{{ port }}/v1/chat/completions
+<span class="method">POST</span>   http://{{ host }}:{{ port }}/v1/embeddings
+<span class="method">GET</span>    http://{{ host }}:{{ port }}/v1/models
+<span class="comment"># API Nativa de Gemini</span>
+<span class="method">POST</span>   http://{{ host }}:{{ port }}/gemini/v1/models/{model_id}:generateContent
+<span class="method">POST</span>   http://{{ host }}:{{ port }}/gemini/v1/models/{model_id}:embedContent
+<span class="method">GET</span>    http://{{ host }}:{{ port }}/gemini/v1/models</div>
+    </div>
+    <div class="terminal-section">
+        <div class="terminal-prompt">models --list</div>
+        <div class="terminal-output">gemini-pro
+gemini-pro-vision
+gemini-2.5-pro-exp-03-25
+gemini-2.0-flash
+gemini-2.0-flash-lite
+gemini-1.5-flash
+gemini-1.5-flash-8b
+gemini-1.5-pro
+gemini-embedding-exp</div>
+    </div>
+    <div class="terminal-section">
+        <div class="terminal-prompt">help</div>
+        <div class="terminal-output"><span class="comment"># Este es un proxy de API para Gemini que proporciona compatibilidad con OpenAI
+# Los endpoints están listados arriba
+# Para más información, consulta el README.md</span></div>
+    </div>
+    <div style="margin-top: 40px; opacity: 0.6">
+        Gemini Proxy v1.0.0 [Running on {{ host }}:{{ port }}] (c) 2024
     </div>
 </body>
 </html>