add postgres

frontend/ui.py

@@ -16,11 +16,18 @@ def _env(name: str, default: str) -> str:
     return value if value else default
 
 
-default_backend_url = _env("BACKEND_URL", "http://backend:8000/generate")
+default_backend_base_url = _env("BACKEND_BASE_URL", "http://backend:8000")
+
+
+def _normalize_base_url(value: str) -> str:
+    v = (value or "").strip().rstrip("/")
+    if v.endswith("/generate"):
+        v = v[: -len("/generate")]
+    return v.rstrip("/")
 
 with st.sidebar:
     st.header("Настройки")
-    api_url = st.text_input("Backend URL", default_backend_url)
+    backend_base_url = _normalize_base_url(st.text_input("Backend URL", default_backend_base_url))
     model = st.selectbox(
         "LLM Модель",
         ["deepseek-r1:1.5b", "qwen2.5-coder:1.5b", "llama3.1:8b", "Arena Model"],
@@ -29,6 +36,8 @@ with st.sidebar:
 
 if "history" not in st.session_state:
     st.session_state.history = []
+if "token" not in st.session_state:
+    st.session_state.token = None
 
 
 def _push_history(*, platform: str, model: str, prompt: str, status: str, config: str, error: str | None):
@@ -47,8 +56,8 @@ def _push_history(*, platform: str, model: str, prompt: str, status: str, config
     st.session_state.history = st.session_state.history[:50]
 
 
-tab_overview, tab_generator, tab_validator, tab_history = st.tabs(
-    ["Обзор", "Pipeline generator", "Проверка YAML", "История"]
+tab_overview, tab_account, tab_generator, tab_validator, tab_history = st.tabs(
+    ["Обзор", "Аккаунт", "Pipeline generator", "Проверка YAML", "История"]
 )
 
 with tab_overview:
@@ -59,7 +68,78 @@ with tab_overview:
         "- **История**: хранит последние результаты в рамках сессии браузера."
     )
     st.divider()
-    st.caption(f"Backend URL: {api_url}")
+    st.caption(f"Backend URL: {backend_base_url}")
+    if st.session_state.token:
+        st.success("Вы вошли в систему.")
+    else:
+        st.info("Вы не авторизованы.")
+
+
+def _auth_headers():
+    if not st.session_state.token:
+        return {}
+    return {"Authorization": f"Bearer {st.session_state.token}"}
+
+
+with tab_account:
+    st.subheader("Аутентификация")
+
+    c1, c2 = st.columns(2)
+    with c1:
+        st.markdown("**Регистрация**")
+        reg_email = st.text_input("Email", key="reg_email")
+        reg_full_name = st.text_input("Имя (опционально)", key="reg_full_name")
+        reg_password = st.text_input("Пароль", type="password", key="reg_password")
+        if st.button("Создать аккаунт"):
+            try:
+                r = requests.post(
+                    f"{backend_base_url}/auth/register",
+                    json={"email": reg_email, "password": reg_password, "full_name": reg_full_name or None},
+                    timeout=30,
+                )
+                if r.status_code >= 400:
+                    st.error(f"Ошибка регистрации: {r.text}")
+                else:
+                    st.success("Аккаунт создан. Теперь можно войти.")
+            except Exception as e:
+                st.error(f"Ошибка подключения к бекенду: {e}")
+
+    with c2:
+        st.markdown("**Вход**")
+        login_email = st.text_input("Email ", key="login_email")
+        login_password = st.text_input("Пароль ", type="password", key="login_password")
+        login_col_a, login_col_b = st.columns([1, 1])
+        with login_col_a:
+            if st.button("Войти", type="primary"):
+                try:
+                    r = requests.post(
+                        f"{backend_base_url}/auth/login",
+                        json={"email": login_email, "password": login_password},
+                        timeout=30,
+                    )
+                    if r.status_code >= 400:
+                        st.error(f"Ошибка входа: {r.text}")
+                    else:
+                        st.session_state.token = r.json().get("access_token")
+                        st.success("Успешный вход.")
+                        st.rerun()
+                except Exception as e:
+                    st.error(f"Ошибка подключения к бекенду: {e}")
+        with login_col_b:
+            if st.button("Выйти", disabled=not bool(st.session_state.token)):
+                st.session_state.token = None
+                st.rerun()
+
+    st.divider()
+    if st.session_state.token:
+        try:
+            r = requests.get(f"{backend_base_url}/auth/me", headers=_auth_headers(), timeout=20)
+            if r.status_code >= 400:
+                st.error(f"Не удалось получить профиль: {r.text}")
+            else:
+                st.json(r.json())
+        except Exception as e:
+            st.error(f"Ошибка подключения к бекенду: {e}")
 
 with tab_generator:
     st.subheader("AI CI/CD Pipeline Generator")
@@ -71,17 +151,25 @@ with tab_generator:
 
     col_a, col_b = st.columns([1, 3])
     with col_a:
-        generate_clicked = st.button("Сгенерировать конфигурацию", type="primary")
+        generate_clicked = st.button(
+            "Сгенерировать конфигурацию",
+            type="primary",
+            disabled=not bool(st.session_state.token),
+        )
     with col_b:
-        st.caption("Подсказка: чем конкретнее требования, тем валиднее YAML.")
+        if st.session_state.token:
+            st.caption("Подсказка: чем конкретнее требования, тем валиднее YAML.")
+        else:
+            st.warning("Нужно войти во вкладке «Аккаунт», чтобы генерировать.")
 
     if generate_clicked:
         if user_input.strip():
             with st.spinner("LLM думает..."):
                 try:
                     res = requests.post(
-                        api_url,
+                        f"{backend_base_url}/generate",
                         json={"description": user_input, "platform": platform, "model": model},
+                        headers=_auth_headers(),
                         timeout=90,
                     )
                     res.raise_for_status()