Rooni commited on
Commit
97e9490
1 Parent(s): 3c501d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -29
app.py CHANGED
@@ -1,42 +1,50 @@
1
  import gradio as gr
2
- import os
3
 
4
- # Функция для отправки запроса к OpenAI GPT-3.5-turbo
5
  def translate_text(input_text, target_language):
6
- # Здесь должен быть ваш ключ API OpenAI GPT-3.5-turbo
7
  api_key = os.getenv("API_KEY")
8
 
9
- # Формируем текст запроса
10
- prompt = f"Переведи пожалуйста правильно, понятно, сохраняя мысль, идеально на {target_language}: \"{input_text}\""
11
-
12
- # Отправляем запрос к OpenAI GPT-3.5-turbo
13
- # Замените URL на актуальный URL для вашего аккаунта OpenAI
14
- response = gr.Interface.api.api_post(
15
- os.getenv("BASE_URL"),
16
- data={
17
- "prompt": prompt,
18
- "max_tokens": 10000, # Можете настроить максимальное количество токенов
19
- },
20
- headers={"Authorization": f"Bearer {api_key}"},
21
- )
22
-
23
- # Получаем ответ от OpenAI и извлекаем переведенный текст
24
- translated_text = response.json()["choices"][0]["text"].strip()
25
- return translated_text
26
 
27
- # Список поддерживаемых языков
28
- languages = ["English", "Spanish", "French", "German", "Italian", "Russian"]
 
 
 
 
 
 
 
 
 
 
 
29
 
30
- # Создаем интерфейс Gradio
31
  iface = gr.Interface(
32
- fn=translate_text,
33
- inputs=[
34
- gr.Textbox("Введите текст для перевода:"),
35
- gr.Dropdown(languages, label="Выберите язык перевода (на):"),
36
- ],
37
  outputs="text",
38
  live=True,
 
 
39
  )
40
 
41
- # Запускаем приложение Gradio
42
  iface.launch()
 
1
  import gradio as gr
2
+ import requests
3
 
4
+ # Функция для отправки fetch запроса к OpenAI GPT-3.5-turbo
5
  def translate_text(input_text, target_language):
6
+ # Ваш ключ API GPT-3.5-turbo
7
  api_key = os.getenv("API_KEY")
8
 
9
+ # URL API GPT-3.5-turbo
10
+ api_url = os.getenv("BASE_URL")
11
+
12
+ # Формирование заголовков запроса
13
+ headers = {
14
+ "Content-Type": "application/json",
15
+ "Authorization": f"Bearer {api_key}"
16
+ }
17
+
18
+ # Формирование параметров запроса
19
+ data = {
20
+ "prompt": f"Переведи пожалуйста правильно, понятно, сохраняя мысль, идеально на {target_language}: \"{input_text}\"",
21
+ "max_tokens": 15000
22
+ "model": "gpt-3.5-turbo"
23
+ }
 
 
24
 
25
+ # Отправка запроса к OpenAI GPT-3.5-turbo
26
+ response = requests.post(api_url, headers=headers, json=data)
27
+
28
+ # Обработка ответа и возврат переведенного текста
29
+ if response.status_code == 200:
30
+ return response.json()["choices"][0]["text"]
31
+ else:
32
+ return f"Error {response.status_code}: {response.text}"
33
+
34
+ # Функция для создания интерфейса Gradio
35
+ def translate_interface(input_text, target_language):
36
+ translated_text = translate_text(input_text, target_language)
37
+ return translated_text
38
 
39
+ # Создание интерфейса Gradio
40
  iface = gr.Interface(
41
+ fn=translate_interface,
42
+ inputs=["text", gr.DropDown(["English", "Spanish", "French", "German", "Italian", "Russian", "Chinese"])],
 
 
 
43
  outputs="text",
44
  live=True,
45
+ title="OpenAI GPT-3.5-turbo Translator",
46
+ description="Translate text to the selected language using GPT-3.5-turbo."
47
  )
48
 
49
+ # Запуск интерфейса Gradio
50
  iface.launch()