AniseF commited on
Commit
d721fef
·
verified ·
1 Parent(s): 96bc042

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +33 -35
utils.py CHANGED
@@ -1,61 +1,59 @@
1
  import os
2
  import re
3
  import requests
4
-
5
- # Perguntas de backup caso o link externo falhe
6
- BACKUP_QUESTIONS = {
7
- "Syntax": ["Analise a estrutura sintática das orações.", "Identifique os casos e suas funções.", "Explique as concordâncias verbais."],
8
- "Morphology": ["Faça a análise morfológica detalhada das palavras.", "Identifique declinações e conjugações.", "Aponte formas nominais e tempos verbais."],
9
- "Semantics": ["Explique o significado contextual dos termos principais.", "Analise as nuances semânticas da passagem.", "Traduza e explique os campos semânticos."]
10
- }
11
 
12
  def detect_language(passage):
13
  greek_chars = re.findall(r'[\u0370-\u03FF\u1F00-\u1FFF]', passage)
14
  return 'greek' if len(greek_chars) > len(passage) * 0.1 else 'latin'
15
 
16
  def call_openrouter(passage, mode, category):
 
17
  api_key = os.getenv("OPENROUTER_API_KEY")
18
  lang = detect_language(passage)
19
 
20
- # Tenta carregar do GitHub, se falhar, usa o backup
 
 
 
 
 
 
21
  try:
22
- urls = {
23
- "Syntax": "https://raw.githubusercontent.com/u-cl-lab/classical-query/main/syntax.txt",
24
- "Morphology": "https://raw.githubusercontent.com/u-cl-lab/classical-query/main/morphology.txt",
25
- "Semantics": "https://raw.githubusercontent.com/u-cl-lab/classical-query/main/semantics.txt"
26
- }
27
- resp = requests.get(urls[category], timeout=10)
28
  questions = [l.strip() for l in resp.text.splitlines() if l.strip().endswith('?')]
29
- if not questions: questions = BACKUP_QUESTIONS[category]
30
- except:
31
- questions = BACKUP_QUESTIONS[category]
 
 
32
 
33
  model = "anthropic/claude-3.5-sonnet" if "Alta" in mode else "google/gemini-flash-1.5"
34
 
35
- report = [f"--- ANÁLISE FILOLÓGICA ({category.upper()}) ---", f"Língua Detectada: {lang.upper()}", f"Modelo: {model}\n"]
36
-
37
- # Para não demorar eternamente e evitar 0kb, vamos processar as perguntas em grupos
38
- # ou garantir que ao menos a primeira tentativa funcione
39
- full_prompt = f"Atue como um filólogo especialista em {lang}. Analise a passagem: {passage}\n\nResponda detalhadamente em PORTUGUÊS às seguintes questões:\n" + "\n".join(questions)
40
-
41
  try:
42
- r = requests.post(
43
- "https://openrouter.ai/api/v1/chat/completions",
44
  headers={"Authorization": f"Bearer {api_key}"},
45
  json={
46
  "model": model,
47
- "messages": [{"role": "user", "content": full_prompt}],
48
  "temperature": 0.1,
49
- "max_tokens": 4000 # Voltamos ao fôlego alto, mas com prompt único para não travar o loop
50
  },
51
- timeout=120
52
  )
53
- if r.status_code == 200:
54
- content = r.json()['choices'][0]['message']['content']
55
- report.append(content)
56
  else:
57
- report.append(f"Erro na API ({r.status_code}): {r.text}")
 
58
  except Exception as e:
59
- report.append(f"Erro de conexão: {str(e)}")
60
-
61
- return "\n".join(report), model
 
1
  import os
2
  import re
3
  import requests
4
+ # Importa as URLs do seu config.py
5
+ from config import SYNTAX_DOC_URL, MORPHOLOGY_DOC_URL, SEMANTICS_DOC_URL
 
 
 
 
 
6
 
7
  def detect_language(passage):
8
  greek_chars = re.findall(r'[\u0370-\u03FF\u1F00-\u1FFF]', passage)
9
  return 'greek' if len(greek_chars) > len(passage) * 0.1 else 'latin'
10
 
11
  def call_openrouter(passage, mode, category):
12
+ # Puxa a chave dos Secrets do Hugging Face
13
  api_key = os.getenv("OPENROUTER_API_KEY")
14
  lang = detect_language(passage)
15
 
16
+ # Mapeia os links do config.py
17
+ urls = {
18
+ "Syntax": SYNTAX_DOC_URL,
19
+ "Morphology": MORPHOLOGY_DOC_URL,
20
+ "Semantics": SEMANTICS_DOC_URL
21
+ }
22
+
23
  try:
24
+ url = urls.get(category)
25
+ resp = requests.get(url, timeout=15)
26
+ resp.raise_for_status()
27
+ # Filtra as perguntas do seu Google Docs
 
 
28
  questions = [l.strip() for l in resp.text.splitlines() if l.strip().endswith('?')]
29
+ except Exception as e:
30
+ return f"Erro ao acessar Google Docs: {str(e)}", "Nenhum"
31
+
32
+ if not questions:
33
+ return "Nenhuma pergunta detectada no documento.", "Nenhum"
34
 
35
  model = "anthropic/claude-3.5-sonnet" if "Alta" in mode else "google/gemini-flash-1.5"
36
 
37
+ # Monta o prompt com todas as perguntas para evitar cortes
38
+ prompt_completo = f"Atue como um filólogo especialista em {lang}. Analise: {passage}\n\nResponda em PORTUGUÊS:\n" + "\n".join(questions)
39
+
 
 
 
40
  try:
41
+ response = requests.post(
42
+ url="https://openrouter.ai/api/v1/chat/completions",
43
  headers={"Authorization": f"Bearer {api_key}"},
44
  json={
45
  "model": model,
46
+ "messages": [{"role": "user", "content": prompt_completo}],
47
  "temperature": 0.1,
48
+ "max_tokens": 4000
49
  },
50
+ timeout=180
51
  )
52
+
53
+ if response.status_code == 200:
54
+ return response.json()['choices'][0]['message']['content'], model
55
  else:
56
+ return f"Erro na API: {response.status_code}", "Falha"
57
+
58
  except Exception as e:
59
+ return f"Erro de conexão: {str(e)}", "Falha"