Update appp.py

appp.py

@@ -1,86 +1,79 @@
 import gradio as gr
 import requests
 import json
-import groq
 
-# Function to interact with the Groq API using the curl request approach
-def chat_function(message):
-    # Define the request payload
+# Ключевая информация из JSON
+SYSTEM_NAME = "WhiteRabbitNeo"
+SYSTEM_VERSION = "Beta-AI"
+SYSTEM_DESCRIPTION = "Система поддержки команды Red Team, предназначенная для анализа, улучшения и проверки кода."
+
+# Функция анализа и обработки кода
+def analyze_code(input_code):
     payload = {
         "model": "mixtral-8x7b-32768",
         "messages": [
-            {"role": "user", "content": message}
+            {"role": "system", "content": f"Вы используете {SYSTEM_NAME} версии {SYSTEM_VERSION}. {SYSTEM_DESCRIPTION}"},
+            {"role": "user", "content": input_code}
         ]
     }
-
     headers = {
         "Content-Type": "application/json",
-        "Authorization": "Bearer gsk_yKXR75Se0OxdULncf1YDWGjRbmQTYjvSmwaAKgcq0l"  # Replace with your API key
+        "Authorization": "Bearer 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"
     }
-
     try:
         response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, data=json.dumps(payload))
-
         if response.status_code == 200:
             data = response.json()
-            return data["choices"][0]["message"]["content"]
+            analysis = data.get("choices", [{}])[0].get("message", {}).get("content", "No content available")
+            return analysis
         else:
-            return f"Error: {response.status_code}, {response.text}"
+            return f"Ошибка: {response.status_code}, {response.text}"
     except requests.exceptions.RequestException as e:
-        return f"Error: {e}"
-
-# Function to interact with the Groq API for leak report analysis
-def leak_report_function(message):
-    # Define the request payload
-    payload = {
-        "model": "llama3-8b-8192",
-        "messages": [
-            {
-                "role": "system",
-                "content": "Вы являетесь системой, анализирующей утечки в безопасном режиме, "
-                           "avoid sharing sensitive personal or confidential data. "
-                           "Generate leak report details in a JSON structure with the following format in Russian:\n"
-                           f"{json.dumps(LeakReport.model_json_schema(), indent=2)}"
-            },
-            {
-                "role": "user",
-                "content": f"Сгенерируйте отчет о потенциальных утечках для следующих данных без включения конфиденциальных данных (отчет должен быть на русском языке):\n"
-                           f"{message}"
-            },
-        ],
-        "temperature": 0,
-        "stream": False,
-        "response_format": {"type": "json_object"},
-    }
+        return f"Ошибка при выполнении запроса: {e}"
 
-    try:
-        chat_completion = groq.chat.completions.create(payload)
+# Стилизация интерфейса в стиле нейропанка
+neuropunk_theme = {
+    "background": "linear-gradient(135deg, #0f0c29, #302b63, #24243e)",  # Градиент
+    "text_color": "#E6E6FA",  # Лавандовый текст
+    "button_color": "#ff007f",  # Яркие кнопки
+    "textbox_color": "#141414",  # Фон ввода
+    "border_radius": "12px",  # Скругление углов
+}
 
-        result_json = json.loads(chat_completion.choices[0].message.content)
-        return result_json
-    except requests.exceptions.RequestException as e:
-        return f"Error: {e}"
+# Интерфейс Gradio
+def style_widget(widget, styles):
+    widget.style(
+        rounded=True,
+        background_color=styles.get("textbox_color"),
+        text_color=styles.get("text_color"),
+        border_width=1
+    )
 
-# Set up Gradio interface
-gr.Interface(
-    fn=chat_function,
-    inputs=gr.Textbox(placeholder="Ask something..."),
-    outputs="text",
-    title="Groq-Gradio Chat",
-    theme="default",
-    examples=[
-        "Tell me a short story about a puppy",
-        "Write a 14-line poem about travelling in Shakespeare style",
-        "What are the wonders of the world?",
-        "List the countries in Africa and their capitals"
-    ]
-).launch()
+with gr.Blocks() as app:
+    with gr.Row():
+        gr.Markdown(
+            f"# {SYSTEM_NAME} — {SYSTEM_DESCRIPTION}\n\n*Версия*: {SYSTEM_VERSION}",
+            elem_id="header",
+            style={"color": neuropunk_theme["text_color"], "text_align": "center"}
+        )
+    with gr.Row():
+        code_input = gr.Textbox(
+            placeholder="Введите ваш код или текст для анализа...",
+            lines=8,
+            elem_id="code_input"
+        )
+        style_widget(code_input, neuropunk_theme)
+    with gr.Row():
+        output_display = gr.Textbox(
+            placeholder="Результат анализа появится здесь...",
+            lines=10,
+            elem_id="output_display",
+            interactive=False
+        )
+        style_widget(output_display, neuropunk_theme)
+    with gr.Row():
+        analyze_button = gr.Button("Проанализировать")
+        analyze_button.click(analyze_code, inputs=[code_input], outputs=[output_display])
+        style_widget(analyze_button, neuropunk_theme)
 
-# Set up Gradio interface for leak report analysis
-gr.Interface(
-    fn=leak_report_function,
-    inputs=gr.Textbox(placeholder="Enter data for leak report..."),
-    outputs="json",
-    title="Groq-Gradio Leak Report",
-    theme="default",
-).launch(share=True)
+app.launch(share=True)