Spaces:
Runtime error
Runtime error
import gradio as gr | |
import requests | |
import json | |
import os | |
def chat_function(message): | |
# Define the request payload | |
payload = { | |
"model": "mixtral-8x7b-32768", | |
"messages": [ | |
{ | |
"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": "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: | |
data = response.json() | |
return data["choices"][0]["message"]["content"] # Return the model's response | |
else: | |
return f"Error: {response.status_code}, {response.text}" | |
except requests.exceptions.RequestException as e: | |
return f"Error: {e}" | |
# 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() | |