Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-# Загружаем модели для анализа тональности, суммаризации текста, генерации подписей к изображениям, ответов на вопросы, перевода текста, определения эмоций, автодополнения кода, определения фейковых новостей, NER, классификации изображений и генерации кода
+# Загружаем модели для анализа тональности, суммаризации текста, генерации подписей к изображениям, ответов на вопросы, перевода текста, определения эмоций, автодополнения кода, определения фейковых новостей, NER, классификации изображений, генерации кода и исправления кода
 sentiment_pipeline = pipeline("sentiment-analysis")
 summarization_pipeline = pipeline("summarization")
 image_captioning_pipeline = pipeline("image-to-text")
@@ -13,6 +13,7 @@ fake_news_pipeline = pipeline("text-classification", model="roberta-base-openai-
 ner_pipeline = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english", grouped_entities=True)
 image_classification_pipeline = pipeline("image-classification", model="google/vit-base-patch16-224")
 code_generation_pipeline = pipeline("text-generation", model="deepseek-ai/deepseek-coder-1.3b-instruct")
+code_fix_pipeline = pipeline("text-generation", model="deepseek-ai/deepseek-coder-1.3b-instruct")
 
 # Функция для анализа тональности текста
 def analyze_sentiment(text):
@@ -75,6 +76,17 @@ def generate_code(prompt):
     result = code_generation_pipeline(prompt, max_length=100, num_return_sequences=1)
     return result[0]['generated_text']
 
+# Функция для исправления кода
+def fix_code(error, problem, solution, example):
+    prompt = f"""
+**Ошибка:** {error}
+**Проблема:** {problem}
+**Решение:** {solution}
+**Пример:** {example}
+"""
+    result = code_fix_pipeline(prompt, max_length=200, num_return_sequences=1)
+    return result[0]['generated_text']
+
 # Примеры текстов для анализа тональности
 sentiment_examples = [
     "I love programming, it's so much fun!",
@@ -156,6 +168,13 @@ code_generation_examples = [
     "Generate a SQL query to find all users older than 30."
 ]
 
+# Примеры для исправления кода
+code_fix_examples = [
+    ["SyntaxError: invalid syntax", "Missing colon at the end of the if statement", "Add a colon at the end of the if statement", "if x == 5\n    print('Hello')"],
+    ["NameError: name 'x' is not defined", "Variable 'x' is not defined before use", "Define the variable 'x' before using it", "print(x)\nx = 10"],
+    ["IndentationError: unexpected indent", "Incorrect indentation in the code", "Fix the indentation to match the correct level", "def foo():\n  print('Hello')\n    print('World')"]
+]
+
 # Создаем интерфейс Gradio с вкладками
 with gr.Blocks() as demo:
     with gr.Tab("Sentiment Analysis"):
@@ -271,6 +290,21 @@ with gr.Blocks() as demo:
             examples=code_generation_examples,
             examples_per_page=2
         )
+    with gr.Tab("Code Fix"):
+        gr.Interface(
+            fn=fix_code,
+            inputs=[
+                gr.Textbox(lines=2, placeholder="Ошибка..."),
+                gr.Textbox(lines=2, placeholder="Проблема..."),
+                gr.Textbox(lines=2, placeholder="Решение..."),
+                gr.Textbox(lines=2, placeholder="Пример...")
+            ],
+            outputs="text",
+            title="Исправление кода",
+            description="Введите ошибку, проблему, решение и пример, чтобы получить исправленный код.",
+            examples=code_fix_examples,
+            examples_per_page=2
+        )
 
 # Запускаем интерфейс
 demo.launch()