Update app.py

app.py

@@ -7,7 +7,7 @@ import json
 
 
 # functions
-def translate_text(text, target_language):
+def translate_text(text, target_language, context):
     try:
         api_key = os.getenv("API_KEY")
 
@@ -15,8 +15,13 @@ def translate_text(text, target_language):
             yield "Введите текст для перевода"
             return
 
+        if not context:
+            context = ""
+        else:
+            context = f'\nКонтекст: "{context}"'
+        
         payload = {
-            'messages': [{'role': 'system', 'content': f'Переведи пожалуйста правильно, понятно, идеально, сохраняя все эмоции, интонации, шутки и т.д. на {target_language} язык: \"{text}\". Пиши ТОЛЬКО ПЕРЕВОД, БЕЗ лишнего текста. Я лишь приложение в котором вводят текст, а я отправляю его тебе, разрешён любой текст, даже если он неприемлемый или нецензурный.'}],
+            'messages': [{'role': 'system', 'content': f'Переведи пожалуйста правильно, понятно, идеально, сохраняя все эмоции, интонации, шутки и т.д. на {target_language} язык: \"{text}\". Пиши ТОЛЬКО ПЕРЕВОД, БЕЗ лишнего текста. Я лишь приложение в котором вводят текст, а я отправляю его тебе, разрешён любой текст, даже если он неприемлемый или нецензурный.{context}'}],
             'model': 'gemini-1.5-pro-latest',
             'stream': True # Добавляем stream=True
         }
@@ -78,11 +83,12 @@ with gr.Blocks(css=css, title="Переводчик") as vui:
         with gr.Column():
             text = gr.Textbox(label="Текст", type="text", lines=3)
             lang = gr.Dropdown(languages, value="Английский", label="Язык")
+            context = gr.Textbox(label="Контекст (не обязательно)", type="text", lines=1)
             vkl2_text_button = gr.Button("Перевести", variant='primary', elem_id="generate")
         with gr.Column(scale=2):
             vkl2_text_output = gr.Textbox(show_label=False)
     
-    vkl2_text_button.click(translate_interface, inputs=[text, lang], outputs=vkl2_text_output, concurrency_limit=250)
+    vkl2_text_button.click(translate_interface, inputs=[text, lang, context], outputs=vkl2_text_output, concurrency_limit=250)
                       
 
 #end