Update app.py

app.py

@@ -1,152 +1,51 @@
+
 import gradio as gr
 import requests
 import json
+import gradio
 
-# Системное описание
-system_info = {
-    "system": {
-        "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": [
-                    "Определение типа ошибки.",
-                    "Исправление ключевых ошибок для обеспечения работоспособности кода.",
-                    "Генерация улучшенных фрагментов в зависимости от уровня оценки."
-                ],
-                "examples": {
-                    "Python": [
-                        {
-                            "issue": "Пропущено двоеточие в условии",
-                            "original_code": "if x > 10\n  print(\"x больше 10\")",
-                            "fixed_code": "if x > 10:\n  print(\"x больше 10\")"
-                        },
-                        {
-                            "issue": "Пропущено двоеточие в цикле",
-                            "original_code": "for i in range(10)\n  print(i)",
-                            "fixed_code": "for i in range(10):\n  print(i)"
-                        }
-                    ],
-                    "PowerShell": [
-                        {
-                            "issue": "Пропущена фигурная скобка в условии",
-                            "original_code": "if ($x -gt 10)\n  Write-Host \"x больше 10\"",
-                            "fixed_code": "if ($x -gt 10) {\n  Write-Host \"x больше 10\"\n}"
-                        },
-                        {
-                            "issue": "Пропущена фигурная скобка в цикле",
-                            "original_code": "for ($i = 0; $i -lt 10; $i++)\n  Write-Host $i",
-                            "fixed_code": "for ($i = 0; $i -lt 10; $i++) {\n  Write-Host $i\n}"
-                        }
-                    ]
-                }
-            },
-            "code_validation": {
-                "description": "Проверка работоспособности исправленного кода.",
-                "output_format": "Вывод только корректного и проверенного кода без промежуточных комментариев."
-            },
-            "predictive_analysis": {
-                "description": "Прогнозирование результата выполнения кода и итеративное улучшение.",
-                "process": [
-                    "Оценка текущего состояния по шкале.",
-                    "Прогнозирование результата выполнения.",
-                    "Циклическое улучшение контента с динамической проверкой."
-                ],
-                "example": {
-                    "model": "RandomForestClassifier",
-                    "parameters": {
-                        "n_estimators": 100,
-                        "random_state": 42
-                    },
-                    "inputs": {
-                        "operation": {
-                            "feature1": 10,
-                            "feature2": 20
-                        }
-                    },
-                    "predicted_result": "Успех выполнения задачи"
-                }
-            }
-        },
-        "rules": {
-            "error_handling": "Добавлять обработку ошибок при необходимости.",
-            "code_style": "Следить за стилем кода и улучшать читаемость.",
-            "final_check": "Обязательно проверять работоспособность и соответствие кода перед выводом.",
-            "response_format": "Выводить только проверенный, корректный код."
-        }
-    }
-}
-
-# Функция анализа и итеративного улучшения кода
-def iterative_code_execution(input_code):
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": "Bearer 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"
-    }
+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": "Вы — API-интерфейс аналитики данных, способный анализировать настроения и отвечать в формате JSON. Схема JSON должна быть следующей:\n\n"
+            },
+            {"role": "user", "content": message}
         ]
     }
+    # Set the headers including your API key
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": "Bearer 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"  # Replace with your API key
+    }
 
-    iteration_count = 0
-    max_iterations = 5
-    final_code = input_code
-
-    while iteration_count < max_iterations:
-        try:
-            response = requests.post("https://api.groq.com/openai/v1/chat/completions", headers=headers, data=json.dumps(payload))
-            if response.status_code == 200:
-                result = response.json().get("choices", [{}])[0].get("message", {}).get("content", "")
-                if "A" in result:  # Если код получил оценку "A", он идеален
-                    return f"✅ Итоговый код после {iteration_count + 1} итераций:\n{result.strip()}"
-                final_code = result
-                payload["messages"].append({"role": "assistant", "content": result})
-                iteration_count += 1
-            else:
-                return f"❌ Ошибка API: {response.status_code} - {response.text}"
-        except requests.RequestException as e:
-            return f"❌ Ошибка запроса: {e}"
-
-    return f"⚠️ Цикл итераций завершён (лимит: {max_iterations}). Финальный код:\n{final_code.strip()}"
-
-# Gradio интерфейс
-with gr.Blocks(css="styles.css") as app:
-    gr.Markdown("### WhiteRabbitNeo: Автоматическое выполнение анализа и улучшения кода", elem_id="header")
-
-    with gr.Row():
-        with gr.Column():
-            input_code = gr.Textbox(
-                placeholder="Введите ваш код здесь... или оставьте пустым для автозаполнения примера.",
-                lines=10,
-                label="Исходный код",
-                value="""def add_numbers(a, b):\nreturn a + b"""  # Пример автозаполнения
-            )
-        with gr.Column():
-            output_code = gr.Textbox(label="Результат анализа и улучшений", lines=15, interactive=False)
-
-    with gr.Row():
-        analyze_button = gr.Button("Выполнить")
-
-    analyze_button.click(iterative_code_execution, inputs=[input_code], outputs=[output_code])
-
-app.launch()
+    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 something..."),  # 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()