Update app.py

app.py

@@ -1,13 +1,14 @@
 import gradio as gr
 from transformers import pipeline
 
-# Загружаем модели для анализа тональности, суммаризации текста, генерации подписей к изображениям, ответов на вопросы, перевода текста и определения эмоций
+# Загружаем модели для анализа тональности, суммаризации текста, генерации подписей к изображениям, ответов на вопросы, перевода текста, определения эмоций и автодополнения кода
 sentiment_pipeline = pipeline("sentiment-analysis")
 summarization_pipeline = pipeline("summarization")
 image_captioning_pipeline = pipeline("image-to-text")
 qa_pipeline = pipeline("question-answering")
 translation_pipeline = pipeline("translation_en_to_ru", model="Helsinki-NLP/opus-mt-en-ru")
 emotion_pipeline = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion")
+code_completion_pipeline = pipeline("text-generation", model="Salesforce/codegen-350M-mono")
 
 # Функция для анализа тональности текста
 def analyze_sentiment(text):
@@ -39,6 +40,11 @@ def detect_emotion(text):
     result = emotion_pipeline(text)[0]
     return f"Emotion: {result['label']}, Confidence: {result['score']:.4f}"
 
+# Функция для автодополнения кода
+def complete_code(code):
+    result = code_completion_pipeline(code, max_length=50, num_return_sequences=1)
+    return result[0]['generated_text']
+
 # Примеры текстов для анализа тональности
 sentiment_examples = [
     "I love programming, it's so much fun!",
@@ -85,6 +91,13 @@ emotion_examples = [
     "I am surprised by the results."
 ]
 
+# Примеры кода для автодополнения
+code_examples = [
+    "def factorial(n):",
+    "import numpy as np",
+    "for i in range(10):"
+]
+
 # Создаем интерфейс Gradio с вкладками
 with gr.Blocks() as demo:
     with gr.Tab("Sentiment Analysis"):
@@ -150,6 +163,16 @@ with gr.Blocks() as demo:
             examples=emotion_examples,
             examples_per_page=2
         )
+    with gr.Tab("Code Completion"):
+        gr.Interface(
+            fn=complete_code,
+            inputs=gr.Textbox(lines=5, placeholder="Введите начало кода..."),
+            outputs="text",
+            title="Автодополнение кода",
+            description="Введите начало кода, чтобы получить его продолжение.",
+            examples=code_examples,
+            examples_per_page=2
+        )
 
 # Запускаем интерфейс
 demo.launch()