AniseF commited on
Commit
0bb403d
·
verified ·
1 Parent(s): 68318a7

Upload 3 files

Browse files
Files changed (2) hide show
  1. requirements.txt +4 -0
  2. utils.py +34 -0
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ gradio
2
+ openai
3
+ requests
4
+ python-dotenv
utils.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import requests
2
+ import os
3
+ import openai
4
+ from config import MODEL_PRIORITY_A, MODEL_PRIORITY_B
5
+
6
+ def get_doc_content(url):
7
+ try:
8
+ response = requests.get(url)
9
+ return response.text if response.status_code == 200 else "Erro ao ler documento."
10
+ except:
11
+ return "Falha na conexão com o Google Docs."
12
+
13
+ def call_openrouter(prompt, mode):
14
+ api_key = os.getenv("OPENROUTER_API_KEY")
15
+ client = openai.OpenAI(
16
+ base_url="https://openrouter.ai/api/v1",
17
+ api_key=api_key
18
+ )
19
+
20
+ chain = MODEL_PRIORITY_A if mode == "Alta Precisão (Filológico)" else MODEL_PRIORITY_B
21
+
22
+ for model in chain:
23
+ try:
24
+ completion = client.chat.completions.create(
25
+ model=model,
26
+ messages=[{"role": "user", "content": prompt}],
27
+ timeout=40
28
+ )
29
+ return completion.choices[0].message.content, model
30
+ except Exception as e:
31
+ print(f"Falha no modelo {model}: {e}")
32
+ continue
33
+
34
+ return "Todos os modelos falharam. Verifique seu saldo no OpenRouter.", "Nenhum"