Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import io
 import re
 import json
-import random
 import streamlit as st
 from docx import Document
 from docx.shared import Cm, Pt, RGBColor
@@ -102,25 +101,11 @@ class RelatorioDiagnostico(BaseModel):
 # =====================================================
 class RevisorIA:
     def __init__(self):
-        # Tenta buscar a lista de chaves (plural) no secrets
-        api_keys = st.secrets.get("GEMINI_API_KEYS")
-
-        # Fallback de segurança caso só exista uma chave configurada (singular)
-        if not api_keys:
-            single_key = st.secrets.get("GEMINI_API_KEY") or os.environ.get("GEMINI_API_KEY")
-            if single_key:
-                api_keys = [single_key]
-        elif isinstance(api_keys, str):
-            # Caso o usuário tenha salvo as chaves como uma única string separada por vírgulas
-            api_keys = [k.strip() for k in api_keys.split(",")]
-
-        if not api_keys:
-            st.error("❌ ERRO: Nenhuma chave GEMINI configurada nos secrets.")
+        api_key = st.secrets.get("GEMINI_API_KEY") or os.environ.get("GEMINI_API_KEY")
+        if not api_key:
+            st.error("❌ ERRO: Chave GEMINI_API_KEY não configurada nos secrets.")
             st.stop()
-
-        # O Load Balancer: Escolhe uma chave aleatória da lista a cada execução/clique
-        chave_escolhida = random.choice(api_keys)
-        self.client = genai.Client(api_key=chave_escolhida)
+        self.client = genai.Client(api_key=api_key)
 
     def auditar_tcc(self, texto_projeto: str, texto_tcc: str, fase_tcc: str) -> Optional[AuditoriaTCC]:
         regra_fase = (
@@ -159,7 +144,7 @@ class RevisorIA:
         )
         try:
             res = self.client.models.generate_content(
-                model="gemini-3.1-flash",
+                model="gemini-3.5-flash",
                 contents=prompt,
                 config=types.GenerateContentConfig(
                     temperature=0.1,
@@ -184,7 +169,7 @@ class RevisorIA:
         )
         try:
             res = self.client.models.generate_content(
-                model="gemini-3.1-flash",
+                model="gemini-3.5-flash",
                 contents=prompt,
                 config=types.GenerateContentConfig(
                     temperature=0.0,
@@ -1394,7 +1379,7 @@ with aba_feedback:
                 )
                 try:
                     res = revisor_ia.client.models.generate_content(
-                        model="gemini-3.1-flash",
+                        model="gemini-3.5-flash",
                         contents=prompt_fb,
                         config=types.GenerateContentConfig(temperature=0.2)
                     )