Update app.py

app.py

@@ -1,38 +1,38 @@
+import gradio as gr
 import openai
-from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Установите API-ключ от OpenAI
-API_KEY = api_key
+# Установите ваш API-ключ OpenAI
+openai.api_key = api_key
 
-def translate_text(text, source_lang, target_lang):
-    # Задайте модель и токенизатор
-    model_name = "gpt-3.5-turbo"
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
+# Функция для перевода текста с помощью GPT-3
+def translate_text(text, source_language, target_language):
+    # Проверяем, нужно ли перевести с русского на английский язык или наоборот
+    if source_language == "ru" and target_language == "en":
+        prompt = "Translate the following Russian text to English: " + text
+    elif source_language == "en" and target_language == "ru":
+        prompt = "Translate the following English text to Russian: " + text
+    else:
+        return "Unsupported language pair"
 
-    # Приведите текст к формату ввода для GPT-3
-    input_text = f"Translate the following text from {source_lang} to {target_lang}: {text}"
-
-    # Сделайте запрос к OpenAI GPT-3
+    # Вызываем GPT-3 API для получения перевода
     response = openai.Completion.create(
         engine="text-davinci-003",
-        prompt=tokenizer.encode(input_text, return_tensors="pt"),
+        prompt=prompt,
         max_tokens=100,
-        temperature=0.7,
         n=1,
         stop=None,
     )
 
-    # Получите перевод и раскодируйте его
-    translated_text = tokenizer.decode(response.choices[0].text.strip())
+    # Извлекаем перевод из ответа API
+    translations = response.choices[0].text.strip().split("\n")
+    translation = translations[0]
 
-    return translated_text
+    return translation
 
-# Установите API-ключ
-openai.api_key = API_KEY
+# Создание интерфейса Gradio для взаимодействия с пользователем
+input_text = gr.inputs.Textbox(label="Введите текст")
+source_lang = gr.inputs.Dropdown(["ru", "en"], label="Исходный язык")
+target_lang = gr.inputs.Dropdown(["ru", "en"], label="Целевой язык")
+output_text = gr.outputs.Textbox(label="Перевод")
 
-# Пример использования
-text = "Привет, как дела?"
-translated_text = translate_text(text, source_lang="ru", target_lang="en")
-print(f"Оригинальный текст: {text}")
-print(f"Переведенный текст: {translated_text}")
+gr.Interface(fn=translate_text, inputs=[input_text, source_lang, target_lang], outputs=output_text).launch()