Upload app.py

app.py

@@ -169,7 +169,9 @@ def respond(message, chat_history, model, system_prompt, temperature, max_tokens
             stream=True,
         )
 
-        partial = ""
+        header = f"_Модель: `{model}`_\n\n"
+        partial = header
+        chat_history[-1]["content"] = partial
         for chunk in stream:
             if not chunk.choices:
                 # some providers send trailing chunks with no `choices`
@@ -223,11 +225,73 @@ def clear_all():
     return [], None, make_preview_html(None)
 
 
+DEFAULT_SETTINGS = {
+    "model": MODEL_CHOICES[0],
+    "system_prompt": DEFAULT_SYSTEM_PROMPT,
+    "temperature": 0.7,
+    "max_tokens": 2048,
+    "top_p": 0.9,
+}
+
+
+def load_settings(saved):
+    s = {**DEFAULT_SETTINGS, **(saved or {})}
+    return (
+        gr.update(value=s["model"]),
+        gr.update(value=s["system_prompt"]),
+        gr.update(value=s["temperature"]),
+        gr.update(value=s["max_tokens"]),
+        gr.update(value=s["top_p"]),
+    )
+
+
+def save_settings(model, system_prompt, temperature, max_tokens, top_p):
+    return {
+        "model": model,
+        "system_prompt": system_prompt,
+        "temperature": temperature,
+        "max_tokens": max_tokens,
+        "top_p": top_p,
+    }
+
+
+# Thin, rounded, theme-colored scrollbars instead of the bulky default OS
+# ones, plus a layout that fits the viewport instead of forcing page scroll.
+CUSTOM_CSS = """
+:root {
+  --sb-thumb: #8a8fb8;
+  --sb-track: transparent;
+}
+@media (prefers-color-scheme: dark) {
+  :root { --sb-thumb: #55597a; }
+}
+* {
+  scrollbar-width: thin;
+  scrollbar-color: var(--sb-thumb) var(--sb-track);
+}
+*::-webkit-scrollbar { width: 8px; height: 8px; }
+*::-webkit-scrollbar-track { background: transparent; }
+*::-webkit-scrollbar-thumb {
+  background-color: var(--sb-thumb);
+  border-radius: 8px;
+  border: 2px solid transparent;
+  background-clip: content-box;
+}
+*::-webkit-scrollbar-thumb:hover { background-color: #9fa3c7; }
+
+.gradio-container { height: 100vh !important; max-height: 100vh !important; }
+#main-row { height: calc(100vh - 130px); }
+#chat-col { height: 100%; }
+#side-col { height: 100%; overflow-y: auto; }
+"""
+
 # ---------------------------------------------------------------------------
 # UI
 # ---------------------------------------------------------------------------
 
-with gr.Blocks(title="Chatterbox", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Chatterbox", theme=gr.themes.Soft(), css=CUSTOM_CSS, fill_height=True) as demo:
+    settings_state = gr.BrowserState(DEFAULT_SETTINGS)
+
     gr.Markdown("## 🤖 Chatterbox — чат + генерация файлов кода")
 
     if not HF_TOKEN:
@@ -239,13 +303,13 @@ with gr.Blocks(title="Chatterbox", theme=gr.themes.Soft()) as demo:
             "Providers*), затем перезапустите Space."
         )
 
-    with gr.Row():
-        with gr.Column(scale=3):
+    with gr.Row(elem_id="main-row"):
+        with gr.Column(scale=3, elem_id="chat-col"):
             chatbot = gr.Chatbot(
                 type="messages",
                 show_label=False,
                 show_copy_button=True,
-                height=520,
+                height="calc(100% - 110px)",
             )
             msg = gr.Textbox(
                 placeholder="Опишите, что построить, например: «сделай лендинг с формой подписки»",
@@ -256,7 +320,7 @@ with gr.Blocks(title="Chatterbox", theme=gr.themes.Soft()) as demo:
                 send_btn = gr.Button("Отправить", variant="primary")
                 clear_btn = gr.Button("Очистить")
 
-        with gr.Column(scale=2):
+        with gr.Column(scale=2, elem_id="side-col"):
             with gr.Tab("📄 Файлы"):
                 files_out = gr.File(
                     label="Сгенерированные файлы (несколько — упаковываются в .zip)",
@@ -277,6 +341,12 @@ with gr.Blocks(title="Chatterbox", theme=gr.themes.Soft()) as demo:
                 max_tokens = gr.Slider(128, 8192, value=2048, step=64, label="Max new tokens")
                 top_p = gr.Slider(0.0, 1.0, value=0.9, step=0.05, label="Top-p")
 
+    settings_widgets = [model_dd, system_box, temperature, max_tokens, top_p]
+
+    demo.load(load_settings, inputs=[settings_state], outputs=settings_widgets)
+    for widget in settings_widgets:
+        widget.change(save_settings, inputs=settings_widgets, outputs=[settings_state])
+
     inputs = [msg, chatbot, model_dd, system_box, temperature, max_tokens, top_p]
     outputs = [chatbot, msg, files_out, preview]