Doubleupai commited on
Commit
2085fa0
·
verified ·
1 Parent(s): 5784f1c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
 
3
- def generate_code(language, task):
4
  # Простые шаблоны для генерации кода
5
  templates = {
6
  "Python": {
@@ -20,20 +20,25 @@ def generate_code(language, task):
20
  }
21
  }
22
 
23
- # Получаем код по языку и задаче
24
- code = templates.get(language, {}).get(task, "Code template not found.")
25
- return code
 
 
 
 
 
26
 
27
  # Создаем интерфейс Gradio
28
  iface = gr.Interface(
29
  fn=generate_code,
30
  inputs=[
31
  gr.Dropdown(choices=["Python", "JavaScript", "Java"], label="Выберите язык программирования"),
32
- gr.Dropdown(choices=["Hello World", "Factorial", "Fibonacci"], label="Выберите задачу")
33
  ],
34
  outputs=gr.Code(label="Сгенерированный код"),
35
  title="Генератор кода",
36
- description="Выберите язык программирования и задачу для генерации кода."
37
  )
38
 
39
  # Запускаем интерфейс
 
1
  import gradio as gr
2
 
3
+ def generate_code(language, task_description):
4
  # Простые шаблоны для генерации кода
5
  templates = {
6
  "Python": {
 
20
  }
21
  }
22
 
23
+ # Поиск подходящего шаблона на основе описания задачи
24
+ task_description = task_description.lower()
25
+ for task, code in templates.get(language, {}).items():
26
+ if task.lower() in task_description:
27
+ return code
28
+
29
+ # Если задача не найдена
30
+ return "Код для данной задачи не найден. Попробуйте описать задачу более четко."
31
 
32
  # Создаем интерфейс Gradio
33
  iface = gr.Interface(
34
  fn=generate_code,
35
  inputs=[
36
  gr.Dropdown(choices=["Python", "JavaScript", "Java"], label="Выберите язык программирования"),
37
+ gr.Textbox(label="Опишите задачу", placeholder="Например, 'напиши код для вычисления факториала'")
38
  ],
39
  outputs=gr.Code(label="Сгенерированный код"),
40
  title="Генератор кода",
41
+ description="Выберите язык программирования и опишите задачу для генерации кода."
42
  )
43
 
44
  # Запускаем интерфейс