Update app.py

app.py

@@ -3,6 +3,9 @@ import requests
 import json
 import os
 import markdown2
+from pygments import highlight
+from pygments.lexers import get_lexer_by_name
+from pygments.formatters import HtmlFormatter
 
 def generate_minecraft_command(description):
     headers = {
@@ -12,7 +15,7 @@ def generate_minecraft_command(description):
 
     payload = {
         'messages': [{'role': 'system', 'content': f'{description}'}],
-        'max_tokens': 100000,
+        'max_tokens': 10000,
         'model': os.getenv("MODEL")
     }
 
@@ -24,7 +27,13 @@ def generate_minecraft_command(description):
         
         # Преобразование Markdown в HTML
         command_html = markdown2.markdown(command)
-        return command_html
+
+        # Подсветка синтаксиса кода
+        lexer = get_lexer_by_name("html", stripall=True)
+        formatter = HtmlFormatter(style="colorful", full=True, noclasses=True)
+        highlighted_code = highlight(command, lexer, formatter)
+
+        return highlighted_code
     elif 'error' in data:
         error_message = data['error']['message']
         return f'Ошибка: {error_message}'
@@ -33,5 +42,5 @@ def generate_minecraft_command(description):
 
 iface = gr.Interface(fn=generate_minecraft_command, inputs=[
     gr.Textbox(label="Запрос")
-], outputs=gr.HTML(label="Ответ"), title="GPT")  # Используем gr.HTML вместо gr.Textbox
+], outputs=gr.HTML(label="Ответ", html=True), title="GPT")
 iface.launch()