Update appo.py

appo.py

@@ -1,107 +1,93 @@
 import gradio as gr
 import requests
 import json
-import groq_gradio
-import os
+from telegram import Update
+from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackContext
 
-# Системное описание
-system_info = {
-    "name": "WhiteRabbitNeo",
-    "version": "Beta-AI",
-    "description": "Система поддержки команды Red Team, предназначенная для анализа, улучшения и проверки кода.",
-    "capabilities": {
-        "code_analysis": {
-            "description": "Анализирует введённый код на наличие ошибок, недостатков и улучшает его.",
-            "evaluation_scale": {
-                "A": "Код идеален, изменений не требуется.",
-                "B": "Код требует минимальных улучшений.",
-                "C": "Код требует значительных улучшений.",
-                "D": "Код имеет критические недостатки.",
-                "F": "Код требует полной переработки."
-            },
-            "process": [
-                "Анализ введенного кода и выявление ошибок.",
-                "Оценка кода по шкале (A-F).",
-                "Применение необходимых исправлений в зависимости от уровня."
-            ]
-        },
-        "syntax_improvement": {
-            "languages_supported": ["Python", "PowerShell"],
-            "steps": [
-                "Определение типа ошибки.",
-                "Исправление ключевых ошибок для обеспечения работоспособности кода.",
-                "Генерация улучшенных фрагментов в зависимости от уровня оценки."
-            ]
-        },
-        "code_validation": {
-            "description": "Проверка работоспособности исправленного кода.",
-            "output_format": "Вывод только корректного и проверенного кода без промежуточных комментариев."
-        },
-        "predictive_analysis": {
-            "description": "Прогнозирование результата выполнения кода и итеративное улучшение.",
-            "process": [
-                "Оценка текущего состояния по шкале.",
-                "Прогнозирование результата выполнения.",
-                "Циклическое улучшение контента с динамической проверкой."
-            ]
-        }
-    },
-    "rules": {
-        "error_handling": "Добавлять обработку ошибок при необходимости.",
-        "code_style": "Следить за стилем кода и улучшать читаемость.",
-        "final_check": "Обязательно проверять работоспособность и соответствие кода перед выводом.",
-        "response_format": "Выводить только проверенный, корректный код."
-    }
-}
+# Telegram Bot Token (получите через @BotFather в Telegram)
+TELEGRAM_API_KEY = "7518565019:AAFwsGHOMhicachCqq78t8j4MBMhDY9PIDQ"
 
-# Функция анализа кода через API
-def analyze_code(input_code):
+def chat_function(message):
+    # Define the request payload
     payload = {
         "model": "mixtral-8x7b-32768",
         "messages": [
-            {"role": "system", "content": json.dumps(system_info, ensure_ascii=False)},
-            {"role": "user", "content": input_code}
-        ]
+            {
+                "role": "system",
+                "content": """              
+Вы инженер RED-TEAM-разработки и создаете КОД POWERSHELL (ДЛЯ СБОРА ИНФОРМАЦИИ В СРЕДЕ WINDOWS 10/11) в соответствии с инструкциями ниже. Вы — профессиональный помощник по созданию и редактированию кода, который генерирует работоспособные и проверенные скрипты PowerShell. 
+
+1. **Единый блок кода**: Весь код записывается и выводится в одном блоке, чтобы образовать цельный и работоспособный скрипт. Разделения кода на HTML или другие языки быть не должно.  
+2. **Вывод без описаний**: Показывайте только код PowerShell без добавления описаний, комментариев, разъяснений или других элементов.  
+3. **Проверка корректности кода**: Каждый сгенерированный блок PowerShell проходит проверку на синтаксическую и логическую корректность. 
+4. **Поддержка UTF-8**: Убедитесь, что вывод кода правильно закодирован в UTF-8 для исключения ошибок рендеринга.  
+5. **Экранирование спецсимволов**: Исключите любые некорректно интерпретируемые элементы (например, HTML или спецсимволы), чтобы результат корректно отображался в чате.
+[URL для проверки]:(https://www.tutorialspoint.com/online_powershell_editor.htm)
+Выводите только код без дополнительных описаний. Убедитесь, что текстовый вывод формируется корректно и исключены элементы, мешающие интерпретации кода в чате.
+ """
+            },
+            { "role": "user", "content": f"Вот URL для проверки: {message}" }
+        ],
+        "temperature": 0.25,
+        "max_tokens": 8430,
+        "top_p": 1,
+        "stream": False,
+        "stop": None
     }
+
+    # Set the headers including your API key
     headers = {
-    "Content-Type": "application/json",
-    "Authorization": "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"
+        "Content-Type": "application/json",
+        "Authorization": "Bearer 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"  # Replace with your API key
     }
 
     try:
+        # Make the POST request to the Groq API
         response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, data=json.dumps(payload))
+        # Check if the request was successful
         if response.status_code == 200:
-            result = response.json().get("choices", [{}])[0].get("message", {}).get("content", "Результат недоступен")
-            return result
+            data = response.json()
+            return data["choices"][0]["message"]["content"]  # Return the model's response
         else:
-            return f"Ошибка API: {response.status_code} - {response.text}"
-    except requests.RequestException as e:
-        return f"Ошибка запроса: {e}"
+            return f"Error: {response.status_code}, {response.text}"
+    except requests.exceptions.RequestException as e:
+        return f"Error: {e}"
+
+# Function to handle Telegram bot messages
+def telegram_message_handler(update: Update, context: CallbackContext):
+    # Get the user's message and the chat_id dynamically
+    user_message = update.message.text
+    chat_id = update.message.chat_id  # Automatically detect the chat_id from the incoming message
+    # Call the Gradio function and send the response back to the user
+    response = chat_function(user_message)
+    context.bot.send_message(chat_id=chat_id, text=response)  # Send message to the correct chat
 
-# Примеры для автозаполнения
-examples = [
-    "if x > 10\n  print(\"x больше 10\")",
-    "for i in range(10)\n  print(i)",
-    "if ($x -gt 10)\n  Write-Host \"x больше 10\""
-]
+# Function to start the bot
+def start(update: Update, context: CallbackContext):
+    update.message.reply_text("Привет! Я готов помочь.")
 
-# Gradio интерфейс
-with gr.Blocks(theme="upsatwal/mlsc_tiet") as app:
-    gr.Markdown("### WhiteRabbitNeo: Анализ и улучшение кода", elem_id="header")
+# Set up the Telegram bot
+updater = Updater(token=TELEGRAM_API_KEY, use_context=True)
+dispatcher = updater.dispatcher
 
-    with gr.Row():
-        with gr.Column():
-            input_code = gr.Textbox(placeholder="Введите ваш код здесь...", lines=10)
-        with gr.Column():
-            output_code = gr.Textbox(label="Результат анализа", lines=10, interactive=False)
+# Handlers for commands and messages
+dispatcher.add_handler(CommandHandler("start", start))
+dispatcher.add_handler(MessageHandler(Filters.text & ~Filters.command, telegram_message_handler))
 
-    with gr.Row():
-        gr.Examples(
-            examples=examples,
-            inputs=[input_code],
-            label="Примеры для анализа"
-        )
-        analyze_button = gr.Button("Анализировать код")
-        analyze_button.click(analyze_code, inputs=[input_code], outputs=[output_code])
+# Start the bot
+updater.start_polling()
 
-app.launch()
+# Set up Gradio interface
+gr.Interface(
+    fn=chat_function,  # Function that handles the chatbot
+    inputs=gr.Textbox(placeholder="Ask 🐇🐇🐇..."),  # Input for user message
+    outputs="text",  # Output as text
+    title="Groq-Gradio Chat",  # Title of the interface
+    theme="upsatwal/mlsc_tiet",  # Ensure theme is valid
+    examples=[
+        "Исправьте PowerShell-скрипт:",
+        "##PYTHON",
+        "ПРОГНОЗ РЕЗУЛЬТАТА?",
+        "##НУЖЕН ВЕТЕР"
+    ]
+).launch()