update 2

ui.py

@@ -1,39 +1,38 @@
 import streamlit as st
 import requests
 
-st.set_page_config(page_title="AI DevOps Assistant", layout="wide")
+st.set_page_config(page_title="DevOps AI Assistant", page_icon="🚀")
 
-st.title("🚀 CI/CD AI Generator & Validator")
-st.subheader("Генерация и проверка конфигураций пайплайнов")
-
-backend_url = "http://localhost:8000/generate_config"
+st.title("🤖 AI CI/CD Pipeline Generator")
+st.markdown("Прототип сервиса для генерации и валидации конфигураций")
 
 with st.sidebar:
-    st.info("Настройки")
-    model_name = st.selectbox("Выберите модель Ollama", ["codellama", "llama3", "mistral"])
+    st.header("Настройки")
+    api_url = st.text_input("Backend URL", "http://localhost:8000/generate")
+    model = st.selectbox("LLM Модель", ["deepseek-r1:1.5b", "qwen2.5-coder:1.5b", "llama3.1:8b", "Arena Model"])
+    platform = st.radio("Платформа", ["GitHub Actions", "GitLab CI"])
 
-task = st.text_area("Опишите задачу (напр. 'Pipeline for Dockerized Node.js app with Jest tests'):")
+user_input = st.text_area("Опишите пайплайн (например: 'Python app with pytest and docker build')", height=150)
 
-if st.button("Сгенерировать и проверить"):
-    if task:
-        with st.spinner("ИИ формирует конфигурацию..."):
-            res = requests.post(backend_url, json={"task_description": task, "model": model_name})
-            if res.status_code == 200:
+if st.button("Сгенерировать конфигурацию", type="primary"):
+    if user_input:
+        with st.spinner("LLM думает..."):
+            try:
+                res = requests.post(api_url, json={
+                    "description": user_input,
+                    "platform": platform,
+                    "model": model
+                })
                 data = res.json()
                 
-                col1, col2 = st.columns(2)
-                
-                with col1:
-                    st.markdown("### Сгенерированный YAML")
+                if data["status"] == "ok":
+                    st.success("✅ Конфигурация валидна!")
                     st.code(data["config"], language="yaml")
-                
-                with col2:
-                    st.markdown("### Статус валидации")
-                    if data["is_valid"]:
-                        st.success("✅ Конфигурация синтаксически верна")
-                    else:
-                        st.error(f"❌ Ошибка в YAML: {data['validation_message']}")
-            else:
-                st.error("Ошибка связи с бекендом")
+                    st.download_button("Скачать .yml", data["config"], file_name="pipeline.yml")
+                else:
+                    st.error(f"❌ Ошибка валидации: {data['validation_error']}")
+                    st.code(data["config"], language="yaml")
+            except Exception as e:
+                st.error(f"Ошибка подключения к бекенду: {e}")
     else:
-        st.warning("Введите описание задачи")
+        st.warning("Пожалуйста, введите описание.")