Update utils.py

utils.py

@@ -1,9 +1,7 @@
 import os
 import re
 import requests
-import time
 
-# URLs dos documentos de perguntas
 URLS = {
     "Syntax": "https://raw.githubusercontent.com/u-cl-lab/classical-query/main/syntax.txt",
     "Morphology": "https://raw.githubusercontent.com/u-cl-lab/classical-query/main/morphology.txt",
@@ -11,57 +9,43 @@ URLS = {
 }
 
 def detect_language(passage):
-    """Detecta se o texto é Grego ou Latim (Baseado na lógica do Thomas)"""
     greek_chars = re.findall(r'[\u0370-\u03FF\u1F00-\u1FFF]', passage)
-    if greek_chars and len(greek_chars) > len(passage) * 0.2:
-        return 'greek'
-    return 'latin'
-
-def get_questions(category):
-    """Busca as perguntas no documento remoto"""
-    try:
-        response = requests.get(URLS.get(category))
-        return [l.strip() for l in response.text.splitlines() if l.strip().endswith('?')]
-    except:
-        return []
+    return 'greek' if len(greek_chars) > len(passage) * 0.1 else 'latin'
 
 def call_openrouter(passage, mode, category):
-    """Processa cada pergunta individualmente para evitar cortes"""
     api_key = os.getenv("OPENROUTER_API_KEY")
     lang = detect_language(passage)
-    questions = get_questions(category)
     
-    if not questions:
+    # Protocolo de Tradução e Especialidade (A essência do Thomas)
+    lang_instr = {
+        'latin': "Você é um latinista experiente. Use apenas regras de gramática latina. Não confunda com grego.",
+        'greek': "Você é um helenista experiente. Use apenas regras de grego antigo (incluindo o Dual e o Optativo)."
+    }
+
+    try:
+        resp_q = requests.get(URLS[category])
+        questions = [l.strip() for l in resp_q.text.splitlines() if l.strip().endswith('?')]
+    except:
         return "Erro ao carregar perguntas.", "Nenhum"
 
-    model_chain = ["anthropic/claude-3.5-sonnet", "openai/gpt-4o"] if mode == "Alta Precisão (Filológica)" else ["google/gemini-flash-1.5", "mistralai/mistral-7b-instruct"]
+    model_chain = ["anthropic/claude-3.5-sonnet", "openai/gpt-4o"] if "Alta" in mode else ["google/gemini-flash-1.5"]
     
-    full_report = []
-    model_used_final = "Nenhum"
-
+    report = []
     for q in questions:
+        # Aqui garantimos a tradução:
+        prompt = f"{lang_instr[lang]}\nAnalise o texto: {passage}\nResponda em PORTUGUÊS: {q}"
+        
         for model in model_chain:
             try:
-                # Prompt instruindo idioma e contexto específico
-                prompt = f"Analise este texto em {lang.upper()}: '{passage}'. Responda em PORTUGUÊS à pergunta: {q}"
-                
-                response = requests.post(
-                    url="https://openrouter.ai/api/v1/chat/completions",
+                r = requests.post(
+                    "https://openrouter.ai/api/v1/chat/completions",
                     headers={"Authorization": f"Bearer {api_key}"},
-                    json={
-                        "model": model,
-                        "messages": [{"role": "user", "content": prompt}],
-                        "temperature": 0.1,
-                        "max_tokens": 800
-                    },
-                    timeout=40
+                    json={"model": model, "messages": [{"role": "user", "content": prompt}], "temperature": 0.1},
+                    timeout=30
                 )
-                if response.status_code == 200:
-                    answer = response.json()["choices"][0]["message"]["content"]
-                    full_report.append(f"Q: {q}\nA: {answer}\n")
-                    model_used_final = model
-                    break # Passa para a próxima pergunta
-            except:
-                continue
-                
-    return "\n".join(full_report), model_used_final
+                if r.status_code == 200:
+                    report.append(f"PERGUNTA: {q}\nRESPOSTA: {r.json()['choices'][0]['message']['content']}\n")
+                    break
+            except: continue
+
+    return "\n".join(report), model_chain[0]