Update app.py

app.py

@@ -1,33 +1,55 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import AutoModelForCausalLM
+import torch
 
-# Load a custom model (replace with your DeepSeek model)
-# Example: Using a Hugging Face model as a placeholder
-code_generator = pipeline("text-generation", model="deepseek-ai/deepseek-coder-1.3b-instruct")  # Replace with your DeepSeek model
+# Загрузка модели DeepSeek-Coder-1.3b-instruct
+model_name = "deepseek-ai/deepseek-coder-1.3b-instruct"
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Function to generate code
+# Проверка доступности GPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model.to(device)
+
+# Функция для генерации кода
 def generate_code(prompt, file_type):
     if file_type in ["Gradio", "Vercel", "Streamlit"]:
-        # Handle platform-specific code generation
+        # Генерация кода для платформ
         prompt_with_file_type = f"Write a configuration or setup code for {file_type} to: {prompt}"
     else:
-        # Handle programming language code generation
+        # Генерация кода для языков программирования
         prompt_with_file_type = f"Write a {file_type} code for: {prompt}"
     
-    # Generate code using the custom model
-    generated_code = code_generator(prompt_with_file_type, max_length=200, num_return_sequences=1)
-    return generated_code[0]['generated_text']
+    # Генерация кода с использованием модели
+    inputs = prompt_with_file_type  # Передаем текст напрямую
+    outputs = model.generate(
+        inputs,  # Модель должна поддерживать прямой ввод текста
+        max_length=200,  # Ограничение длины вывода
+        num_return_sequences=1,  # Один вариант ответа
+        temperature=0.7,  # Контроль случайности
+        top_p=0.9,  # Контроль разнообразия
+    )
+    
+    # Возвращаем сгенерированный код
+    return outputs[0]  # Предполагаем, что модель возвращает текст напрямую
 
-# Function to update code
+# Функция для обновления кода
 def update_code(existing_code, update_prompt):
-    # Combine the existing code and the update prompt
+    # Объединение существующего кода и нового запроса
     prompt_with_update = f"Rewrite the following code to: {update_prompt}\n\nExisting Code:\n{existing_code}"
     
-    # Generate updated code using the custom model
-    updated_code = code_generator(prompt_with_update, max_length=200, num_return_sequences=1)
-    return updated_code[0]['generated_text']
+    # Генерация обновленного кода
+    outputs = model.generate(
+        prompt_with_update,
+        max_length=250,  # Увеличен для обновления кода
+        num_return_sequences=1,
+        temperature=0.7,
+        top_p=0.9,
+    )
+    
+    # Возвращаем обновленный код
+    return outputs[0]
 
-# Examples for the interface
+# Примеры для интерфейса
 examples = [
     ["Write a function to calculate factorial", "Python"],
     ["Create a simple interface for a calculator", "Gradio"],
@@ -37,7 +59,7 @@ examples = [
     ["Create a responsive navbar", "HTML"],
 ]
 
-# Gradio interface
+# Gradio интерфейс
 with gr.Blocks(theme='Nymbo/Nymbo_Theme') as demo:
     gr.Markdown("# AI Code Generator with Update Feature")
     gr.Markdown("Enter a prompt and select the file type or platform to generate code. You can also update the generated code with a new prompt.")
@@ -50,27 +72,40 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme') as demo:
             value="Python"
         )
     
-    output_code = gr.Textbox(label="Generated Code", lines=10, interactive=False)
+    # Панель для кода с безлимитными строками
+    output_code = gr.Textbox(label="Generated Code", lines=None, interactive=False)
+    
+    # Кнопка "Скопировать"
+    copy_button = gr.Button("Скопировать код")
+    copy_button.click(
+        None,  # Не требует Python-функции
+        inputs=[output_code],  # Входные данные — текст из output_code
+        outputs=None,  # Нет выходных данных
+        js="""(text) => {
+            navigator.clipboard.writeText(text);
+            alert('Код скопирован в буфер обмена!');
+        }"""
+    )
     
     generate_button = gr.Button("Generate Code")
     generate_button.click(fn=generate_code, inputs=[input_prompt, file_type], outputs=output_code)
 
-    # Update Code Section
+    # Секция для обновления кода
     with gr.Row():
         update_prompt = gr.Textbox(label="Update Prompt", placeholder="e.g., Add error handling to the code...")
         update_button = gr.Button("Update Code")
     
     update_button.click(fn=update_code, inputs=[output_code, update_prompt], outputs=output_code)
 
-    # Add examples
+    # Добавление примеров
     gr.Examples(
         examples=examples,
         inputs=[input_prompt, file_type],
         outputs=output_code,
         fn=generate_code,
-        cache_examples=True,  # Cache results for faster loading
+        cache_examples=True,  # Кэширование для ускорения
         label="Click on an example to get started!"
     )
 
-# Launch the interface
+# Запуск интерфейса
 demo.launch()