Deploy TextAI v2 - Clean architecture

app.py

@@ -21,30 +21,27 @@ def create_app():
     state.sync_with_filesystem()
     logger.info("App", f"TextAI v{VERSION} starting")
 
-    with gr.Blocks(
-        title=APP_TITLE,
-        theme=get_theme(),
-        css=THEME_CSS
-    ) as app:
+    # Gradio 6.0: theme/css moved to launch(), but we use Blocks() for structure
+    with gr.Blocks(title=APP_TITLE) as app:
 
         with gr.Tabs() as main_tabs:
 
             # ══════════════════════════════════════════════════════════
             # TAB 1: CHAT
             # ══════════════════════════════════════════════════════════
-            with gr.Tab("Chat", id=0):
+            with gr.Tab("Chat", id="chat"):
                 chat_components = build_chat_ui()
 
             # ══════════════════════════════════════════════════════════
             # TAB 2: MODELS
             # ══════════════════════════════════════════════════════════
-            with gr.Tab("Models", id=1):
+            with gr.Tab("Models", id="models"):
                 model_components = build_model_manager_ui()
 
             # ══════════════════════════════════════════════════════════
             # TAB 3: TOOLS
             # ══════════════════════════════════════════════════════════
-            with gr.Tab("Tools", id=2):
+            with gr.Tab("Tools", id="tools"):
                 with gr.Tabs():
 
                     # System Info
@@ -110,28 +107,28 @@ def create_app():
                     with gr.Tab("API"):
                         gr.Markdown("### API Endpoints")
                         gr.Markdown("""
-                        All endpoints return JSON. Use for integration with local apps.
-
-                        **Models:**
-                        - `api_list_models()` - List installed models
-                        - `api_load_model(model_id)` - Load a model
-                        - `api_unload_model()` - Unload current model
-                        - `api_search_models(query, max_params, limit)` - Search HF
-                        - `api_download_model(repo_id, filename)` - Download model
-
-                        **Sessions:**
-                        - `api_list_sessions()` - List all sessions
-                        - `api_create_session(title, type, system_prompt)` - New session
-                        - `api_delete_session(session_id)` - Delete session
-
-                        **Chat:**
-                        - `api_chat(session_id, message, max_tokens, temperature)` - Chat
-                        - `api_inference(prompt, messages, system_prompt, ...)` - Direct inference
-
-                        **System:**
-                        - `api_health()` - Health check
-                        - `api_get_status()` - Full status
-                        - `api_get_backends()` - Available backends
+All endpoints return JSON. Use for integration with local apps.
+
+**Models:**
+- `api_list_models()` - List installed models
+- `api_load_model(model_id)` - Load a model
+- `api_unload_model()` - Unload current model
+- `api_search_models(query, max_params, limit)` - Search HF
+- `api_download_model(repo_id, filename)` - Download model
+
+**Sessions:**
+- `api_list_sessions()` - List all sessions
+- `api_create_session(title, type, system_prompt)` - New session
+- `api_delete_session(session_id)` - Delete session
+
+**Chat:**
+- `api_chat(session_id, message, max_tokens, temperature)` - Chat
+- `api_inference(prompt, messages, system_prompt, ...)` - Direct inference
+
+**System:**
+- `api_health()` - Health check
+- `api_get_status()` - Full status
+- `api_get_backends()` - Available backends
                         """)
 
                         gr.Markdown("### Test")
@@ -146,43 +143,42 @@ def create_app():
             # ══════════════════════════════════════════════════════════
             # TAB 4: ABOUT
             # ══════════════════════════════════════════════════════════
-            with gr.Tab("About", id=3):
+            with gr.Tab("About", id="about"):
                 gr.Markdown(f"""
-                # TextAI v{VERSION}
+# TextAI v{VERSION}
 
-                **Local AI Chat Assistant**
+**Local AI Chat Assistant**
 
-                A clean, professional chat interface for running local LLM models.
-                Inspired by ChatGPT & Grok.
+A clean, professional chat interface for running local LLM models.
 
-                ---
+---
 
-                ### Features
+### Features
 
-                - 🤖 **Chat** - Natural conversation with AI
-                - 📁 **Session Management** - Auto-save, rename, delete chats
-                - 🔧 **Model Manager** - Download, load, configure models
-                - 🔍 **HuggingFace Search** - Find and download GGUF models
-                - 🎛️ **Customization** - System prompts, temperature, tokens
-                - 📡 **API** - Full API for integration with other apps
+- Chat - Natural conversation with AI
+- Session Management - Auto-save, rename, delete chats
+- Model Manager - Download, load, configure models
+- HuggingFace Search - Find and download GGUF models
+- Customization - System prompts, temperature, tokens
+- API - Full API for integration with other apps
 
-                ---
+---
 
-                ### Supported Models
+### Supported Models
 
-                - **GGUF** - Via llama-cpp-python (recommended)
-                - **Transformers** - Via HuggingFace transformers
+- **GGUF** - Via llama-cpp-python (recommended)
+- **Transformers** - Via HuggingFace transformers
 
-                ---
+---
 
-                ### Keyboard Shortcuts
+### Keyboard Shortcuts
 
-                - `Enter` - Send message
-                - `Shift+Enter` - New line
+- `Enter` - Send message
+- `Shift+Enter` - New line
 
-                ---
+---
 
-                Built with ❤️ using Gradio
+Built with Gradio
                 """)
 
     return app
@@ -197,5 +193,7 @@ if __name__ == "__main__":
     app.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False
+        share=False,
+        theme=get_theme(),
+        css=THEME_CSS
     )

ui/chat.py

@@ -44,20 +44,13 @@ def build_chat_ui():
         model = service.get_loaded_model()
         return f"● {model['name']}" if model else "No model loaded"
 
-    def format_history(session_id: str) -> List[dict]:
+    def format_history(session_id: str) -> List[Tuple[str, str]]:
         """Format session messages for Gradio chatbot"""
         service = get_session_service()
         display = service.get_session_for_display(session_id)
         if not display:
             return []
-        # Convert to new message format
-        messages = []
-        for user_msg, bot_msg in display.get("history", []):
-            if user_msg:
-                messages.append({"role": "user", "content": user_msg})
-            if bot_msg:
-                messages.append({"role": "assistant", "content": bot_msg})
-        return messages
+        return display.get("history", [])
 
     # ══════════════════════════════════════════════════════════════════
     # EVENT HANDLERS
@@ -111,10 +104,7 @@ def build_chat_ui():
 
         # Check model
         if not model_service.is_model_loaded():
-            history = history + [
-                {"role": "user", "content": message},
-                {"role": "assistant", "content": "⚠️ Please load a model first. Go to Models tab."}
-            ]
+            history = history + [(message, "⚠️ Please load a model first. Go to Models tab.")]
             return history, "", session_id, get_sessions_for_sidebar()
 
         # Create session if needed
@@ -122,8 +112,7 @@ def build_chat_ui():
             session = session_service.create_session(system_prompt=system_prompt)
             session_id = session["id"]
 
-        # Add user message
-        history = history + [{"role": "user", "content": message}]
+        # Add user message to session
         session_service.add_message(session_id, "user", message)
 
         # Build messages for model
@@ -138,10 +127,10 @@ def build_chat_ui():
         try:
             response = model_service.generate(messages, max_tokens, temperature)
             session_service.add_message(session_id, "assistant", response)
-            history = history + [{"role": "assistant", "content": response}]
+            history = history + [(message, response)]
         except Exception as e:
             logger.error("Chat", f"Generation error: {e}")
-            history = history + [{"role": "assistant", "content": f"⚠️ Error: {e}"}]
+            history = history + [(message, f"⚠️ Error: {e}")]
 
         return history, "", session_id, get_sessions_for_sidebar()
 
@@ -199,7 +188,7 @@ def build_chat_ui():
                 interactive=False,
                 show_label=False,
                 row_count=10,
-                col_count=(2, "fixed"),
+                column_count=(2, "fixed"),
                 elem_id="sessions-table"
             )
 
@@ -237,7 +226,6 @@ def build_chat_ui():
                 show_label=False,
                 height=400,
                 elem_id="chatbot",
-                type="messages",
                 autoscroll=False
             )