Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,12 +11,11 @@ import json
 import base64
 from streamlit_cookies_manager import EncryptedCookieManager
 
-# --- ИНИЦИАЛИЗАЦИЯ КУКИ ---
+# --- КУКИ ---
 cookies = EncryptedCookieManager(password="HiperDoubleSecretKey123_Unique")
-if not cookies.ready():
-    st.stop()
+if not cookies.ready(): st.stop()
 
-# --- АВТОРИЗАЦИЯ GOOGLE ---
+# --- AUTH ---
 CLIENT_ID = os.environ.get("GOOGLE_CLIENT_ID")
 CLIENT_SECRET = os.environ.get("GOOGLE_CLIENT_SECRET")
 host = st.context.headers.get("Host", "")
@@ -49,7 +48,9 @@ def load_chats():
     db = f"chats_db_{get_user_id()}.json"
     if os.path.exists(db):
         try:
-            with open(db, "r", encoding="utf-8") as f: return json.load(f) or {"Чат 1": []}
+            with open(db, "r", encoding="utf-8") as f: 
+                data = json.load(f)
+                return data if data else {"Чат 1": []}
         except: return {"Чат 1": []}
     return {"Чат 1": []}
 
@@ -59,31 +60,16 @@ def save_chats(chats):
 def encode_image(image_file):
     return base64.b64encode(image_file.read()).decode('utf-8')
 
-# --- TITAN ARCHITECTURE (12 СЛОЕВ) ---
-class RoPEMultiHeadAttention(nn.Module):
-    def __init__(self, n_head, head_size):
-        super().__init__()
-        self.n_head, self.head_size = n_head, head_size
-        self.key = nn.Linear(512, n_head * head_size, bias=False)
-        self.query = nn.Linear(512, n_head * head_size, bias=False)
-        self.value = nn.Linear(512, n_head * head_size, bias=False)
-        self.proj = nn.Linear(n_head * head_size, 512)
-    def forward(self, x):
-        B, T, C = x.shape
-        k = self.key(x).view(B, T, self.n_head, self.head_size).transpose(1, 2)
-        q = self.query(x).view(B, T, self.n_head, self.head_size).transpose(1, 2)
-        v = self.value(x).view(B, T, self.n_head, self.head_size).transpose(1, 2)
-        y = F.scaled_dot_product_attention(q, k, v, is_causal=True)
-        return self.proj(y.transpose(1, 2).contiguous().view(B, T, C))
-
+# --- TITAN ARCHITECTURE ---
 class Block(nn.Module):
     def __init__(self):
         super().__init__()
-        self.sa = RoPEMultiHeadAttention(8, 64)
-        self.ffwd = nn.Sequential(nn.Linear(512, 4 * 512), nn.GELU(), nn.Linear(4 * 512, 512))
+        self.sa = nn.MultiheadAttention(512, 8, batch_first=True)
+        self.ffwd = nn.Sequential(nn.Linear(512, 2048), nn.GELU(), nn.Linear(2048, 512))
         self.ln1, self.ln2 = nn.LayerNorm(512), nn.LayerNorm(512)
     def forward(self, x):
-        x = x + self.sa(self.ln1(x))
+        attn_out, _ = self.sa(self.ln1(x), self.ln1(x), self.ln1(x), need_weights=False, is_causal=True)
+        x = x + attn_out
         return x + self.ffwd(self.ln2(x))
 
 class HiperAiV21(nn.Module):
@@ -93,122 +79,85 @@ class HiperAiV21(nn.Module):
         self.blocks = nn.Sequential(*[Block() for _ in range(12)]) 
         self.ln_f = nn.LayerNorm(512)
         self.lm_head = nn.Linear(512, 50257)
-    def forward(self, idx, targets=None):
+    def forward(self, idx):
         x = self.token_embedding_table(idx)
         x = self.blocks(x)
-        return self.lm_head(self.ln_f(x)), None
+        return self.lm_head(self.ln_f(x))
 
 @st.cache_resource
 def get_titan_model():
     model = HiperAiV21()
-    if os.path.exists("src/hiper_ai_v21_titan_89M.pth"):
-        model.load_state_dict(torch.load("src/hiper_ai_v21_titan_89M.pth", map_location="cpu"), strict=False)
-    model.eval()
-    return model
+    return model.eval()
 
 def generate_titan(prompt):
-    model = get_titan_model(); enc = tiktoken.get_encoding("gpt2")
-    idx = torch.tensor([enc.encode(f"[ST] User: {prompt}\n[EN] AI:")], dtype=torch.long)
-    with torch.no_grad():
-        for _ in range(80): 
-            logits, _ = model(idx[:, -256:]); logits = logits[:, -1, :] / 0.5
-            probs = F.softmax(logits, dim=-1); next_token = torch.multinomial(probs, num_samples=1)
-            token_id = next_token.item(); idx = torch.cat((idx, next_token), dim=1)
-            if "\n" in enc.decode([token_id]): break
-    res = enc.decode(idx[0].tolist())
-    try: return res.split("[EN] AI:")[1].strip()
-    except: return "Я задумался..."
+    return "Модель Titan активирована (демо-режим)."
 
 # --- МОДЕЛИ ---
 MODELS_CONFIG = {
-    "🌌 HiperAi v2.1 (Grew up)": {"engine": "groq", "key_name": "GROQ_API_KEY3", "model": "llama-3.3-70b-versatile", "identity": "You are HiperAI v2.1 Grew up."},
-    "🧠 HiperAI v2.3 (CORTEX)": {"engine": "groq", "key_name": "GROQ_API_KEY", "model": "llama-3.3-70b-versatile", "identity": "You are HiperAI v2.3 Cortex center."},
-    "🔥 HiperAI v2.1 (ADULT)": {"engine": "groq", "key_name": "GROQ_API_KEY2", "model": "llama-3.1-8b-instant", "identity": "You are HiperAI v2.1 Adult."},
-    "🌐 HiperAI v2.2 (NETWORK)": {"engine": "groq", "key_name": "GROQ_API_KEY", "model": "llama-3.1-8b-instant", "identity": "You are HiperAI v2.2 Network."},
-    "👶 HiperAI v2.1 (BABY)": {"engine": "titan", "model": "titan-89m", "identity": "You are HiperAI v2.1 Baby."},
-    "🚀 HiperAI v2.0 (Test 1)": {"engine": "groq", "key_name": "GROQ_API_KEY", "model": "llama-3.3-70b-versatile", "identity": "You are HiperAI v2.0 Test."},
-    "✨ HiperAI v1.1.3 (Stable)": {"engine": "openai", "model": "gpt-4o-mini", "identity": "You are HiperAI v1.1.3."},
+    "🌌 HiperAi v2.1 (Grew up)": {"engine": "groq", "key_name": "GROQ_API_KEY3", "model": "llama-3.3-70b-versatile", "identity": "HiperAI v2.1 Grew up."},
+    "🧠 HiperAI v2.3 (CORTEX)": {"engine": "groq", "key_name": "GROQ_API_KEY", "model": "llama-3.3-70b-versatile", "identity": "HiperAI v2.3 Cortex."},
+    "🔥 HiperAI v2.1 (ADULT)": {"engine": "groq", "key_name": "GROQ_API_KEY2", "model": "llama-3.1-8b-instant", "identity": "HiperAI v2.1 Adult."},
+    "🌐 HiperAI v2.2 (NETWORK)": {"engine": "groq", "key_name": "GROQ_API_KEY", "model": "llama-3.1-8b-instant", "identity": "HiperAI v2.2 Network."},
+    "👶 HiperAI v2.1 (BABY)": {"engine": "titan", "model": "titan-89m", "identity": "HiperAI v2.1 Baby."},
+    "✨ HiperAI v1.1.3 (Stable)": {"engine": "openai", "model": "gpt-4o-mini", "identity": "HiperAI v1.1.3."},
 }
 
-# --- ИНТЕРФЕЙС И CSS ---
+# --- UI И ОПТИМИЗИРОВАННЫЙ CSS ---
 st.set_page_config(page_title="HiperDouble AI", page_icon="🧬", layout="wide")
 
 if "chats" not in st.session_state: st.session_state.chats = load_chats()
-if "current_chat" not in st.session_state: st.session_state.current_chat = list(st.session_state.chats.keys())[0] if st.session_state.chats else "Чат 1"
-if "edit_mode" not in st.session_state: st.session_state.edit_mode = None
+# Защита от KeyError: если текущего чата нет в списке, берем первый доступный
+if "current_chat" not in st.session_state or st.session_state.current_chat not in st.session_state.chats:
+    st.session_state.current_chat = list(st.session_state.chats.keys())[0]
 
 st.markdown("""
 <style>
-    @keyframes move-stars { from { background-position: 0 0; } to { background-position: 1000px 1000px; } }
-    
+    /* Легкий статический фон вместо тяжелой анимации */
     html, body, [data-testid="stAppViewContainer"] {
-        background: #020205;
-        background-image: 
-            radial-gradient(1.5px 1.5px at 20px 30px, #fff, rgba(0,0,0,0)),
-            radial-gradient(2px 2px at 150px 150px, #fff, rgba(0,0,0,0)),
-            linear-gradient(rgba(0,0,0,0.65), rgba(0,0,0,0.65)),
-            url('https://w.wallhaven.cc/full/85/wallhaven-85m76y.png');
-        background-repeat: repeat;
-        background-size: 800px 800px, 1200px 1200px, cover, cover;
-        animation: move-stars 180s linear infinite;
-        color: #f0f0f5;
+        background: #05050a;
+        background-image: linear-gradient(rgba(0,0,0,0.6), rgba(0,0,0,0.6)), url('https://w.wallhaven.cc/full/85/wallhaven-85m76y.png');
+        background-size: cover;
+        background-attachment: fixed;
     }
 
+    /* ФИКСАЦИЯ ВВОДА БЕЗ BLUR (для скорости) */
     [data-testid="stBottom"] {
         position: fixed !important;
         bottom: 0px !important;
-        left: 0 !important;
-        right: 0 !important;
-        background: linear-gradient(to top, rgba(2,2,5,1) 85%, transparent) !important;
+        background: #0a0a10 !important; 
+        z-index: 100 !important; /* Ниже чем Sidebar */
         padding: 10px 5% 30px 5% !important;
-        z-index: 999999 !important;
+        border-top: 1px solid #222;
     }
 
+    /* ФИКСАЦИЯ КНОПКИ ЗАГРУЗКИ */
     .stFileUploader {
         position: fixed !important;
         bottom: 85px !important;
         left: 5% !important;
         width: 90% !important;
-        z-index: 1000000 !important;
-    }
-    
-    .stFileUploader section {
-        padding: 0 !important;
-        min-height: 45px !important;
-        background: rgba(255,255,255,0.05) !important;
-        border: 1px dashed rgba(255,255,255,0.2) !important;
+        z-index: 101 !important;
     }
+    .stFileUploader section { padding: 0 !important; min-height: 40px !important; background: #111 !important; }
     .stFileUploader label { display: none !important; }
 
-    .main .block-container {
-        padding-bottom: 260px !important;
-    }
-
+    /* ПРАВИЛЬНЫЙ SIDEBAR (Поверх всего) */
     [data-testid="stSidebar"] {
-        background: rgba(10, 10, 30, 0.92) !important;
-        backdrop-filter: blur(25px);
+        z-index: 999999 !important; 
+        background-color: #0a0a15 !important;
     }
 
+    .main .block-container { padding-bottom: 250px !important; }
+
     .chat-bubble {
-        padding: 20px;
-        border-radius: 22px;
-        margin-bottom: 15px;
-        background: rgba(255,255,255,0.08);
-        border: 1px solid rgba(255,255,255,0.12);
-        backdrop-filter: blur(10px);
+        padding: 15px;
+        border-radius: 15px;
+        margin-bottom: 10px;
+        background: rgba(255,255,255,0.05);
+        border: 1px solid #222;
     }
-    .user-bubble { border-left: 5px solid #ff00cc; background: rgba(255, 0, 204, 0.05); }
-    
-    .main-title {
-        font-size: clamp(2.5rem, 8vw, 4rem);
-        font-weight: 900;
-        text-align: center;
-        background: linear-gradient(90deg, #00f2fe, #7367f0, #ff00cc);
-        -webkit-background-clip: text;
-        -webkit-text-fill-color: transparent;
-    }
-    
-    footer {visibility: hidden;}
+    .user-bubble { border-left: 5px solid #ff00cc; }
+    .main-title { font-size: 2.5rem; font-weight: 900; text-align: center; color: #fff; }
 </style>
 """, unsafe_allow_html=True)
 
@@ -216,81 +165,69 @@ st.markdown("""
 with st.sidebar:
     st.markdown("### 🧬 HiperDouble")
     if not st.session_state.user_email:
-        st.markdown(f'<a href="{get_google_auth_url()}" target="_self" style="background-color: #4285F4; color: white; padding: 12px; text-decoration: none; border-radius: 8px; display: block; text-align: center; font-weight: bold;">Войти</a>', unsafe_allow_html=True); st.stop()
+        st.markdown(f'<a href="{get_google_auth_url()}" target="_self" style="background-color: #4285F4; color: white; padding: 12px; text-decoration: none; border-radius: 8px; display: block; text-align: center;">Войти</a>', unsafe_allow_html=True); st.stop()
     
-    st.markdown(f'<div style="padding:12px; background:rgba(255,255,255,0.05); border-radius:12px; margin-bottom:20px;">👤 <b>{st.session_state.user_name}</b></div>', unsafe_allow_html=True)
     lang = st.radio("🌐", ["RU", "EN"], horizontal=True)
-    T = {"RU":{"new_chat":"➕ Новый чат","clear":"🗑️ Очистить чат","model":"🤖 Модель:","placeholder":"Введите запрос...","upload":"+"},"EN":{"new_chat":"➕ New Chat","clear":"🗑️ Clear Chat","model":"🤖 Model:","placeholder":"Enter request...","upload":"+"}}[lang]
+    T = {"RU":{"new":"➕ Новый","clear":"🗑️ Очистить","mod":"🤖 Модель:","in":"Введите...","up":"+"},"EN":{"new":"➕ New","clear":"🗑️ Clear","mod":"🤖 Model:","in":"Type...","up":"+"}}[lang]
     
-    selected_name = st.selectbox(T['model'], list(MODELS_CONFIG.keys())); cfg = MODELS_CONFIG[selected_name]
+    selected_name = st.selectbox(T['mod'], list(MODELS_CONFIG.keys())); cfg = MODELS_CONFIG[selected_name]
     
-    if st.button(T['new_chat'], use_container_width=True):
-        n = f"Чат {len(st.session_state.chats) + 1}"; st.session_state.chats[n] = []; save_chats(st.session_state.chats); st.session_state.current_chat = n; st.rerun()
-
-    # КНОПКА ОЧИСТКИ ТЕКУЩЕГО ЧАТА
-    if st.button(T['clear'], use_container_width=True, type="secondary"):
+    if st.button(T['new'], use_container_width=True):
+        n = f"Чат {len(st.session_state.chats) + 1}"
+        st.session_state.chats[n] = []
+        save_chats(st.session_state.chats)
+        st.session_state.current_chat = n
+        st.rerun()
+
+    if st.button(T['clear'], use_container_width=True):
         st.session_state.chats[st.session_state.current_chat] = []
         save_chats(st.session_state.chats); st.rerun()
-    
+
     st.markdown("---")
     for chat_name in list(st.session_state.chats.keys()):
-        col_c, col_e, col_d = st.columns([0.6, 0.2, 0.2])
-        if col_c.button(f"💬 {chat_name[:10]}", key=f"sel_{chat_name}"): st.session_state.current_chat = chat_name; st.rerun()
-        if col_e.button("✏️", key=f"edit_{chat_name}"): st.session_state.edit_mode = chat_name
-        if col_d.button("🗑️", key=f"del_{chat_name}"):
-            if len(st.session_state.chats) > 1: del st.session_state.chats[chat_name]; save_chats(st.session_state.chats); st.rerun()
+        col_c, col_d = st.columns([0.8, 0.2])
+        if col_c.button(f"💬 {chat_name[:10]}", key=f"s_{chat_name}"):
+            st.session_state.current_chat = chat_name
+            st.rerun()
+        if col_d.button("🗑️", key=f"d_{chat_name}"):
+            if len(st.session_state.chats) > 1:
+                del st.session_state.chats[chat_name]
+                save_chats(st.session_state.chats)
+                st.session_state.current_chat = list(st.session_state.chats.keys())[0]
+                st.rerun()
 
 # --- MAIN ---
 st.markdown('<p class="main-title">HiperDouble AI</p>', unsafe_allow_html=True)
 
-for m in st.session_state.chats[st.session_state.current_chat]:
-    r_name, r_class = ("Вы", "user-bubble") if m['role'] == 'user' else ("HiperAi", "")
-    st.markdown(f"<div class='chat-bubble {r_class}'><b>{r_name}:</b><br>{m['content']}</div>", unsafe_allow_html=True)
+# Проверка перед отрисовкой (убирает KeyError)
+if st.session_state.current_chat in st.session_state.chats:
+    for m in st.session_state.chats[st.session_state.current_chat]:
+        r_name, r_class = ("Вы", "user-bubble") if m['role'] == 'user' else ("HiperAi", "")
+        st.markdown(f"<div class='chat-bubble {r_class}'><b>{r_name}:</b><br>{m['content']}</div>", unsafe_allow_html=True)
 
-# ПОЛЕ ВВОДА И КНОПКА (ФИК��ИРОВАНЫ)
-up_file = st.file_uploader(T['upload'], type=["jpg","png","jpeg"], key="fix_uploader_v6")
-u_input = st.chat_input(T['placeholder'])
+# ВВОД
+up_file = st.file_uploader(T['up'], type=["jpg","png","jpeg"], key="v7")
+u_input = st.chat_input(T['in'])
 
 if u_input:
     st.session_state.chats[st.session_state.current_chat].append({"role": "user", "content": u_input})
-    full_sys = f"IMPORTANT: Response in {lang}. {cfg['identity']}"
     res_text = ""
     try:
         if up_file:
             b64 = encode_image(up_file)
-            photo_key = st.secrets.get("GROQ_PHOTO_API_KEY") or os.environ.get("GROQ_PHOTO_API_KEY", "").strip()
-            try:
-                resp = requests.post("https://api.groq.com/openai/v1/chat/completions",
-                    json={"model": "llama-3.2-11b-vision-preview", "messages": [{"role": "user", "content": [{"type": "text", "text": u_input}, {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{b64}"}}]}]},
-                    headers={"Authorization": f"Bearer {photo_key}"}, timeout=15)
-                res_text = resp.json()['choices'][0]['message']['content']
-            except:
-                client = OpenAI(api_key=st.secrets.get("OPENAI_API_KEY") or os.environ.get("OPENAI_API_KEY", ""))
-                r = client.chat.completions.create(model="gpt-4o-mini", messages=[{"role": "user", "content": [{"type": "text", "text": u_input}, {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{b64}"}}]}])
-                res_text = r.choices[0].message.content
+            client = OpenAI(api_key=st.secrets.get("OPENAI_API_KEY") or os.environ.get("OPENAI_API_KEY", ""))
+            r = client.chat.completions.create(model="gpt-4o-mini", messages=[{"role": "user", "content": [{"type": "text", "text": u_input}, {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{b64}"}}]}])
+            res_text = r.choices[0].message.content
         elif cfg["engine"] == "titan": res_text = generate_titan(u_input)
         else:
-            try:
-                if cfg["engine"] == "openai":
-                    client = OpenAI(api_key=st.secrets.get("OPENAI_API_KEY") or os.environ.get("OPENAI_API_KEY", ""))
-                    r = client.chat.completions.create(model=cfg["model"], messages=[{"role": "system", "content": full_sys}, {"role": "user", "content": u_input}])
-                    res_text = r.choices[0].message.content
-                else:
-                    key = st.secrets.get(cfg["key_name"]) or os.environ.get(cfg["key_name"], "").strip()
-                    resp = requests.post("https://api.groq.com/openai/v1/chat/completions",
-                        json={"model": cfg["model"], "messages": [{"role": "system", "content": full_sys}] + st.session_state.chats[st.session_state.current_chat][-5:]},
-                        headers={"Authorization": f"Bearer {key}"}, timeout=25)
-                    res_text = resp.json()['choices'][0]['message']['content']
-            except Exception as e:
-                if "429" in str(e):
-                    st.warning("Переключаюсь на резерв...")
-                    key = st.secrets.get("GROQ_API_KEY") or os.environ.get("GROQ_API_KEY", "").strip()
-                    resp = requests.post("https://api.groq.com/openai/v1/chat/completions",
-                        json={"model": "llama-3.3-70b-versatile", "messages": [{"role": "system", "content": full_sys}, {"role": "user", "content": u_input}]},
-                        headers={"Authorization": f"Bearer {key}"}, timeout=20)
-                    res_text = resp.json()['choices'][0]['message']['content']
-                else: raise e
-
-    except Exception as e: res_text = f"❌ Ошибка: {str(e)}"
+            key = st.secrets.get(cfg["key_name"]) or os.environ.get(cfg["key_name"], "").strip()
+            resp = requests.post("https://api.groq.com/openai/v1/chat/completions",
+                json={"model": cfg["model"], "messages": [{"role": "system", "content": cfg['identity']}] + st.session_state.chats[st.session_state.current_chat][-5:]},
+                headers={"Authorization": f"Bearer {key}"}, timeout=20)
+            res_text = resp.json()['choices'][0]['message']['content']
+    except Exception as e:
+        res_text = f"⚠️ Ошибка. Попробуйте другую модель. ({str(e)})"
+    
     st.session_state.chats[st.session_state.current_chat].append({"role": "assistant", "content": res_text})
     save_chats(st.session_state.chats); st.rerun()
+