Update app.py

app.py

@@ -1,1057 +1,73 @@
-# app.py
-
 import os
-import functools
+import openai
 import tiktoken
 import gradio as gr
-from openai import OpenAI
-from typing import List, Tuple, Dict, Optional
-
-"""JadeGPT – Synthwave Edition
--------------------------------------------------
-• OpenAI Python SDK ≥ 1.0.0
-• Gradio ≥ 5.34.1
-• tiktoken
-
-Cyberpunk meets Miami Vice. Welcome to the future, 1984.
-"""
-
-# ────────────────────────────────
-# 1 · Initialisation & constants
-# ────────────────────────────────
 
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY", "").strip())
-
-MODEL_DETAILS = {
-    "gpt-4o": {"input_price": 5.00, "output_price": 15.00, "max_context": 128_000, "description": "OpenAI's most capable and cost-effective multimodal model."},
-    "gpt-4o-mini": {"input_price": 0.15, "output_price": 0.60, "max_context": 128_000, "description": "Highly affordable and fast, suitable for simpler tasks."},
-    "o3": {"input_price": 2.00, "output_price": 8.00, "max_context": 200_000, "description": "Designed for complex reasoning and step-by-step problem-solving."},
-    "o3-pro": {"input_price": 20.00, "output_price": 80.00, "max_context": 200_000, "description": "Premium reasoning model for high-stakes tasks, higher accuracy."},
-    "o4-mini": {"input_price": 1.10, "output_price": 4.40, "max_context": 200_000, "description": "Balanced multimodal model, good for image+text on a budget."},
-    "gpt-4-32k": {"input_price": 0.06, "output_price": 0.12, "max_context": 32_768, "description": "Larger context GPT-4 model (older generation)."},
-    "gpt-4": {"input_price": 0.03, "output_price": 0.06, "max_context": 8_192, "description": "Original GPT-4 model (older generation)."},
-    "gpt-3.5-turbo": {"input_price": 0.001, "output_price": 0.002, "max_context": 16_385, "description": "Cost-effective, fast, and good for general-purpose tasks."},
-}
-
-_env_models = os.getenv("OPENAI_MODEL_LIST", "gpt-4o,gpt-4o-mini,gpt-3.5-turbo")
-ALL_MODELS: List[str] = [m.strip() for m in _env_models.split(",") if m.strip() and m.strip() in MODEL_DETAILS]
-
-for model in MODEL_DETAILS:
-    if model not in ALL_MODELS:
-        ALL_MODELS.append(model)
+# Configure API key
+openai.api_key = os.getenv("OPENAI_API_KEY")
+if not openai.api_key:
+    raise ValueError("Please set the OPENAI_API_KEY environment variable.")
 
+# Available models (comma-separated in env or fallback list)
+_env_models = os.getenv("OPENAI_MODEL_LIST", "gpt-3.5-turbo,gpt-4")
+ALL_MODELS = [m.strip() for m in _env_models.split(",") if m.strip()]
 if not ALL_MODELS:
-    ALL_MODELS = list(MODEL_DETAILS.keys())
-    if not ALL_MODELS:
-        raise ValueError("No valid OpenAI models configured. Please check MODEL_DETAILS or OPENAI_MODEL_LIST.")
-
-DEFAULT_MAX_CONTEXT = MODEL_DETAILS.get(ALL_MODELS[0], {}).get("max_context", 128_000)
-BUFFER_TOKENS       = 500
-DEFAULT_REPLY_MAX   = 2_048
-TEMPERATURE         = 0.3
+    ALL_MODELS = ["gpt-3.5-turbo"]
 
-# ────────────────────────────────
-# 2 · Helpers
-# ────────────────────────────────
-
-@functools.lru_cache(maxsize=128)
-def count_tokens(text: str, model: str) -> int:
+# Token counter using tiktoken
+@functools.lru_cache(maxsize=64)
+def _get_encoding(model: str):
     try:
-        enc = tiktoken.encoding_for_model(model)
+        return tiktoken.encoding_for_model(model)
     except KeyError:
-        enc = tiktoken.get_encoding("cl100k_base")
-    return len(enc.encode(text))
-
-def trim_conversation(convo: List[Dict], model: str, max_context: int) -> List[Dict]:
-    if not convo:
-        return []
+        return tiktoken.get_encoding("cl100k_base")
 
-    kept = [convo[0]]
-    total = count_tokens(convo[0]["content"], model)
-
-    for msg in reversed(convo[1:]):
-        t = count_tokens(msg["content"], model)
-        if total + t + BUFFER_TOKENS > max_context:
-            break
-        kept.insert(1, msg)
-        total += t
-    return kept
-
-def token_cost(model: str, p: int, c: int) -> float:
-    details = MODEL_DETAILS.get(model)
-    if not details:
-        return 0.0
-    return round(((p * details["input_price"]) + (c * details["output_price"])) / 1_000_000, 6)
+def count_tokens(text: str, model: str) -> int:
+    enc = _get_encoding(model)
+    return len(enc.encode(text))
 
-def read_file_content(file_obj) -> str:
-    """Reads the content of an uploaded file."""
-    if file_obj is None:
+# Read uploaded file
+def read_file_content(file_obj):
+    if not file_obj:
         return ""
-    file_path = file_obj.name
     try:
-        with open(file_path, "r", encoding="utf-8") as f:
-            content = f.read()
-        return f"\n\n--- Start of uploaded file: {os.path.basename(file_path)} ---\n{content}\n--- End of uploaded file ---\n"
-    except Exception as e:
-        return f"\n\n--- Error reading file {os.path.basename(file_path)}: {e} ---\n"
-
-# ────────────────────────────────
-# 3 · OpenAI helpers (streaming)
-# ────────────────────────────────
-def safe_chat_stream(convo: List[Dict], max_ctx: int, max_rep: int, models: List[str]):
-    """Stream reply; after completion return usage safely (avoids max_tokens=0 bug)."""
-    last_exc = None
-    for m in models:
-        try:
-            current_model_max_context = MODEL_DETAILS.get(m, {}).get("max_context", max_ctx)
-            trimmed_convo = trim_conversation(convo, m, current_model_max_context)
-
-            if not trimmed_convo or count_tokens(" ".join([msg["content"] for msg in trimmed_convo]), m) > current_model_max_context:
-                raise ValueError(f"Conversation too long for model '{m}' even after trimming. Max context: {current_model_max_context} tokens.")
-
-            # Use appropriate parameter based on model
-            completion_params = {
-                "model": m,
-                "messages": trimmed_convo,
-                "temperature": TEMPERATURE,
-                "stream": True,
-            }
-            
-            # Check if model needs max_completion_tokens instead of max_tokens
-            if m in ["o4-mini", "o3", "o3-pro"]:
-                completion_params["max_completion_tokens"] = max_rep
-            else:
-                completion_params["max_tokens"] = max_rep
-            
-            stream = client.chat.completions.create(**completion_params)
-            
-            reply = ""
-            for chunk in stream:
-                delta = chunk.choices[0].delta.content or ""
-                reply += delta
-                yield reply, None, m
-            
-            usage = None
-            try:
-                usage_params = {
-                    "model": m,
-                    "messages": trimmed_convo + [{"role": "assistant", "content": reply}],
-                    "temperature": 0,
-                }
-                if m in ["o4-mini", "o3", "o3-pro"]:
-                    usage_params["max_completion_tokens"] = 1
-                else:
-                    usage_params["max_tokens"] = 1
-                    
-                usage_resp = client.chat.completions.create(**usage_params)
-                usage = usage_resp.usage
-            except Exception:
-                prompt_tokens_est = count_tokens(" ".join([msg["content"] for msg in trimmed_convo]), m)
-                completion_tokens_est = count_tokens(reply, m)
-                usage = type('obj', (object,), {'prompt_tokens': prompt_tokens_est, 'completion_tokens': completion_tokens_est})()
-
-            yield reply, usage, m
-            return
-
-        except Exception as e:
-            msg = str(e).lower()
-            last_exc = e
-            if "context length" in msg or "maximum context length" in msg or "length_exceeded" in msg:
-                print(f"Model '{m}' failed due to context length. Trying next model if available.")
-                continue
-            if "model_not_found" in msg or "does not exist" in msg or "404" in msg:
-                print(f"Model '{m}' not found or unavailable. Trying next model if available.")
-                continue
-            break
-
-    raise last_exc or RuntimeError("All selected models failed or an unexpected API error occurred.")
-
-# ────────────────────────────────
-# 4 · Gradio generators
-# ────────────────────────────────
-def chat_stream(user_msg: str, hist: List[Tuple[str, str]], sys_prompt: str, sel_model: str, ctx: int, rep: int, uploaded_file):
-    user_msg = (user_msg or "").strip()
-    if not user_msg and not uploaded_file:
-        yield hist, "", "💚 Jade is waiting for your message...", gr.File(value=None)
-        return
-
-    if not client.api_key:
-        hist = hist or []
-        hist.append((user_msg, "❌ JADE SYSTEM ERROR: API_KEY not found. Anthony needs to add the key!"))
-        yield hist, "", "💔 ACCESS DENIED - Missing API Key", gr.File(value=None)
-        return
-
-    file_content = read_file_content(uploaded_file)
-    if file_content:
-        user_msg = f"{user_msg}\n{file_content}"
-
-    convo = [{"role": "system", "content": sys_prompt}]
-    for u, a in hist or []:
-        convo.append({"role": "user", "content": u})
-        convo.append({"role": "assistant", "content": a})
-    convo.append({"role": "user", "content": user_msg})
-
-    hist = hist or []
-    hist.append((user_msg, ""))
-
-    status_message = f"💚 JADE IS CONNECTING TO {sel_model}..."
-    yield hist, "", status_message, gr.File(value=None)
-
-    models_to_try = [sel_model] + [m for m in ALL_MODELS if m != sel_model]
-    
-    acc = ""
-    usage_final = None
-    actual_model_used = sel_model
-
-    try:
-        final_reply_chunks = []
-        last_model_used_during_stream = sel_model
-        
-        stream_generator = safe_chat_stream(convo, ctx, rep, models_to_try)
-        
-        while True:
-            try:
-                part, usage, model_name_during_stream = next(stream_generator)
-                acc = part
-                hist[-1] = (user_msg, acc)
-                if model_name_during_stream:
-                    last_model_used_during_stream = model_name_during_stream
-                yield hist, "", f"💎 JADE IS CRAFTING YOUR RESPONSE VIA {last_model_used_during_stream}...", gr.File(value=None)
-                final_reply_chunks.append(part)
-            except StopIteration:
-                break
-
-        prompt_tokens_final = count_tokens(" ".join([msg["content"] for msg in trim_conversation(convo, last_model_used_during_stream, MODEL_DETAILS.get(last_model_used_during_stream, {}).get("max_context", ctx))]), last_model_used_during_stream)
-        completion_tokens_final = count_tokens(acc, last_model_used_during_stream)
-        usage_final = type('obj', (object,), {'prompt_tokens': prompt_tokens_final, 'completion_tokens': completion_tokens_final})()
-
-        if usage_final:
-            pt, ct = usage_final.prompt_tokens, usage_final.completion_tokens
-            cost = token_cost(last_model_used_during_stream, pt, ct)
-            meta = f"\n\n---\n💚 {pt+ct} love tokens (input {pt} / output {ct}) · 💰${cost:.6f} USD"
-            hist[-1] = (user_msg, acc + meta)
-            final_status = f"✨ JADE'S TRANSMISSION COMPLETE [{last_model_used_during_stream}] {meta}"
-        else:
-            hist[-1] = (user_msg, acc)
-            final_status = f"✨ JADE'S TRANSMISSION COMPLETE [{last_model_used_during_stream}]"
-        
-        yield hist, "", final_status, gr.File(value=None)
-
-    except Exception as e:
-        error_msg = f"💔 JADE ENCOUNTERED AN ERROR: {e}"
-        hist[-1] = (user_msg, error_msg)
-        yield hist, "", f"💔 ERROR IN JADE'S {sel_model}: {e}", gr.File(value=None)
-
-def clear_chat():
-    return [], "", "💚 JADE'S MEMORY REFRESHED. READY FOR NEW ADVENTURES WITH ANTHONY!", None, None
-
-# ────────────────────────────────
-# 5 · UI
-# ────────────────────────────────
-
-# --- GREEN SYNTHWAVE CSS ---
-SYNTHWAVE_CSS = """
-@import url('https://fonts.googleapis.com/css2?family=Orbitron:wght@400;700;900&display=swap');
-@import url('https://fonts.googleapis.com/css2?family=Audiowide&display=swap');
-@import url('https://fonts.googleapis.com/css2?family=Exo+2:wght@300;400;700;900&display=swap');
-
-/* Global green synthwave styles */
-html.dark.gradio-container,
-body.gradio-container {
-    background: 
-        radial-gradient(ellipse at top, #001a1a 0%, transparent 60%),
-        radial-gradient(ellipse at bottom, #004d4d 0%, transparent 60%),
-        linear-gradient(to bottom, #000a0a 0%, #001a1a 30%, #002626 60%, #003333 100%) !important;
-    background-attachment: fixed !important;
-    font-family: 'Exo 2', 'Orbitron', monospace !important;
-    color: #ff1493 !important; /* HOT PINK for all text */
-    overflow-x: hidden;
-    position: relative;
-}
-
-/* Multiple animated elements for depth */
-.grid-background {
-    position: fixed;
-    bottom: 0;
-    left: 0;
-    width: 100%;
-    height: 50%;
-    background-image: 
-        linear-gradient(rgba(0, 255, 136, 0.02) 1px, transparent 1px),
-        linear-gradient(90deg, rgba(0, 255, 136, 0.02) 1px, transparent 1px);
-    background-size: 50px 50px;
-    transform: perspective(400px) rotateX(70deg);
-    transform-origin: center bottom;
-    z-index: -1;
-    opacity: 0.8;
-}
-
-/* Animated stars */
-body.gradio-container::after {
-    content: "";
-    position: fixed;
-    top: 0;
-    left: 0;
-    width: 100%;
-    height: 100%;
-    background-image: 
-        radial-gradient(1px 1px at 20px 30px, #00ff8866, transparent),
-        radial-gradient(1px 1px at 40px 70px, #00ffcc66, transparent),
-        radial-gradient(0.5px 0.5px at 50px 90px, #ffffff44, transparent),
-        radial-gradient(0.5px 0.5px at 130px 40px, #ffffff44, transparent);
-    background-repeat: repeat;
-    background-size: 200px 200px;
-    animation: stars-move 600s linear infinite; /* Very slow */
-    z-index: -2;
-    opacity: 0.4;
-}
-
-@keyframes stars-move {
-    from { transform: translateY(0); }
-    to { transform: translateY(-2000px); }
-}
-
-/* Main container with green theme - refined */
-gradio-app {
-    --background-fill-primary: rgba(0, 26, 26, 0.9) !important;
-    --background-fill-secondary: rgba(0, 13, 13, 0.95) !important;
-    --border-color-accent: #00ff88 !important;
-    --border-color-primary: #00ff8844 !important;
-    --color-accent: #00ff88 !important;
-    --color-text-primary: #ff1493 !important; /* HOT PINK */
-    --color-text-secondary: #ff69b4 !important; /* LIGHTER HOT PINK */
-    --button-primary-background-color: #00ff88 !important;
-    --button-primary-text-color: #001a1a !important;
-    --shadow-color: rgba(0, 255, 136, 0.3) !important;
-    --block-background-fill: rgba(0, 26, 26, 0.8) !important;
-    --block-border-color: #00ff8855 !important;
-    --block-shadow: 0 8px 32px rgba(0, 255, 136, 0.15), 0 0 0 1px rgba(0, 255, 136, 0.1) !important;
-}
-
-/* Add a subtle vignette effect */
-.vignette-overlay {
-    position: fixed;
-    top: 0;
-    left: 0;
-    width: 100%;
-    height: 100%;
-    background: radial-gradient(circle at center, transparent 0%, rgba(0, 10, 10, 0.4) 100%);
-    pointer-events: none;
-    z-index: 10;
-}
-
-/* Headers with jade neon glow - more subtle */
-h1 {
-    font-family: 'Audiowide', cursive !important;
-    color: #ff1493 !important; /* HOT PINK */
-    text-shadow: 
-        0 0 10px #ff1493,
-        0 0 20px #ff1493,
-        0 0 30px #ff69b4,
-        0 0 40px #ff69b4 !important;
-    font-size: 4em !important;
-    letter-spacing: 6px !important;
-    position: relative;
-    /* Removed pulsing animation for less distraction */
-}
-
-h1::before {
-    content: attr(data-text);
-    position: absolute;
-    left: 0;
-    top: 0;
-    z-index: -1;
-    background: none;
-    color: #ff1493;
-    text-shadow: 
-        0 0 10px #ff1493,
-        0 0 20px #ff1493,
-        0 0 30px #ff69b4;
-    filter: blur(2px);
-    opacity: 0.5;
-}
-
-h2, h3, h4, h5, h6 {
-    color: #ff1493 !important; /* HOT PINK */
-    text-shadow: 0 0 8px #ff149350 !important; /* Softer glow */
-    font-family: 'Exo 2', sans-serif !important;
-    font-weight: 700 !important;
-    letter-spacing: 2px !important;
-}
-
-/* Text styling - Better readability */
-p, .prose {
-    color: #ff1493 !important; /* HOT PINK */
-    text-shadow: 0 0 2px rgba(255, 20, 147, 0.3) !important;
-    font-size: 1.05em !important;
-    font-weight: 400 !important;
-    line-height: 1.6 !important;
-}
-
-/* Input fields with jade synthwave style */
-input[type="text"], textarea, select, .gr-dropdown {
-    background: linear-gradient(135deg, rgba(0, 26, 26, 0.95) 0%, rgba(0, 51, 51, 0.9) 100%) !important;
-    border: 2px solid #00ff8866 !important;
-    color: #ff1493 !important; /* HOT PINK */
-    box-shadow: 
-        inset 0 2px 8px rgba(0, 0, 0, 0.3),
-        0 0 20px rgba(0, 255, 136, 0.1) !important;
-    backdrop-filter: blur(10px) !important;
-    font-family: 'Exo 2', monospace !important;
-    font-size: 1.1em !important;
-    font-weight: 400 !important;
-    padding: 12px !important;
-    transition: all 0.2s ease !important;
-}
-
-input[type="text"]:focus, textarea:focus {
-    border-color: #00ffcc !important;
-    box-shadow: 
-        inset 0 2px 8px rgba(0, 0, 0, 0.2),
-        0 0 30px rgba(0, 255, 204, 0.3) !important;
-    color: #ff1493 !important; /* HOT PINK */
-    background: linear-gradient(135deg, rgba(0, 51, 51, 0.95) 0%, rgba(0, 77, 77, 0.9) 100%) !important;
-}
-
-/* Chatbot styling with jade theme - removed animation */
-#component-chat {
-    background: linear-gradient(135deg, rgba(0, 26, 26, 0.9) 0%, rgba(0, 51, 51, 0.7) 100%) !important;
-    border: 2px solid #00ff8866 !important;
-    box-shadow: 
-        0 0 20px rgba(0, 255, 136, 0.2),
-        inset 0 0 20px rgba(0, 255, 136, 0.05) !important;
-    position: relative;
-    overflow: hidden;
-}
-
-#component-chat .message.user {
-    background: linear-gradient(135deg, rgba(0, 51, 51, 0.9) 0%, rgba(0, 77, 77, 0.8) 100%) !important;
-    color: #ff1493 !important; /* HOT PINK */
-    border: 1px solid #00ff8844 !important;
-    box-shadow: 
-        0 4px 12px rgba(0, 0, 0, 0.3),
-        0 0 20px rgba(0, 255, 136, 0.1) !important;
-    font-size: 1.05em !important;
-    font-weight: 400 !important;
-}
-
-#component-chat .message.bot {
-    background: linear-gradient(135deg, rgba(0, 26, 26, 0.95) 0%, rgba(0, 51, 51, 0.9) 100%) !important;
-    color: #ff1493 !important; /* HOT PINK */
-    border: 1px solid #00ffcc33 !important;
-    box-shadow: 
-        0 4px 12px rgba(0, 0, 0, 0.3),
-        0 0 20px rgba(0, 255, 204, 0.1) !important;
-    font-size: 1.05em !important;
-    font-weight: 400 !important;
-}
-
-/* Buttons with jade energy - subtle hover */
-.gr-button-primary {
-    background: linear-gradient(135deg, #00ff88 0%, #00cc66 100%) !important;
-    color: #001a1a !important;
-    border: none !important;
-    box-shadow: 
-        0 4px 15px rgba(0, 255, 136, 0.3),
-        0 2px 5px rgba(0, 0, 0, 0.2) !important;
-    text-transform: uppercase !important;
-    letter-spacing: 3px !important;
-    font-weight: 900 !important;
-    font-size: 1.1em !important;
-    position: relative;
-    overflow: hidden;
-    transition: all 0.2s ease !important;
-}
-
-.gr-button-primary:hover {
-    background: linear-gradient(135deg, #00ffaa 0%, #00ff88 100%) !important;
-    box-shadow: 
-        0 6px 20px rgba(0, 255, 136, 0.4),
-        0 2px 5px rgba(0, 0, 0, 0.3) !important;
-    transform: translateY(-1px);
-}
-
-.gr-button-secondary {
-    background: transparent !important;
-    color: #ff1493 !important; /* HOT PINK */
-    border: 2px solid #00ffcc !important;
-    box-shadow: 
-        0 0 10px rgba(0, 255, 204, 0.2),
-        inset 0 0 10px rgba(0, 255, 204, 0.05) !important;
-    text-transform: uppercase !important;
-    letter-spacing: 2px !important;
-    font-weight: 700 !important;
-    font-size: 1em !important;
-    transition: all 0.2s ease !important;
-}
-
-.gr-button-secondary:hover {
-    background: rgba(0, 255, 204, 0.1) !important;
-    color: #ff69b4 !important; /* LIGHTER HOT PINK on hover */
-    box-shadow: 
-        0 0 15px rgba(0, 255, 204, 0.3),
-        inset 0 0 15px rgba(0, 255, 204, 0.1) !important;
-}
-
-/* Sliders with jade glow */
-.gr-slider-track {
-    background: linear-gradient(90deg, #003333 0%, #00ff88 50%, #00ffcc 100%) !important;
-    height: 10px !important;
-    box-shadow: 0 0 30px rgba(0, 255, 136, 0.6) !important;
-    border-radius: 5px !important;
-}
-
-.gr-slider-thumb {
-    background: radial-gradient(circle, #ffffff 30%, #00ff88 70%) !important;
-    border: 3px solid #00ff88 !important;
-    box-shadow: 
-        0 0 30px rgba(0, 255, 136, 0.8),
-        0 0 50px rgba(0, 255, 136, 0.5) !important;
-    width: 28px !important;
-    height: 28px !important;
-    cursor: grab !important;
-}
-
-.gr-slider-thumb:active {
-    cursor: grabbing !important;
-    box-shadow: 
-        0 0 40px rgba(0, 255, 136, 1),
-        0 0 60px rgba(0, 255, 136, 0.7) !important;
-}
-
-/* Tabs with jade style */
-.tabs-nav button {
-    background: linear-gradient(180deg, transparent 0%, rgba(0, 255, 136, 0.1) 100%) !important;
-    color: #ff1493 !important; /* HOT PINK */
-    border-bottom: 3px solid transparent !important;
-    text-transform: uppercase !important;
-    letter-spacing: 3px !important;
-    font-weight: 700 !important;
-    font-size: 1.1em !important;
-    padding: 15px 30px !important;
-    transition: all 0.3s ease !important;
-}
-
-.tabs-nav button.selected {
-    background: linear-gradient(180deg, rgba(0, 255, 136, 0.2) 0%, rgba(0, 255, 136, 0.3) 100%) !important;
-    color: #ff69b4 !important; /* LIGHTER HOT PINK */
-    border-bottom: 3px solid #00ff88 !important;
-    text-shadow: 0 0 20px #ff1493 !important;
-    box-shadow: 0 10px 30px -10px rgba(0, 255, 136, 0.5) !important;
-}
-
-.tabs-nav button:hover {
-    color: #ff69b4 !important; /* LIGHTER HOT PINK */
-    text-shadow: 0 0 15px #ff1493 !important;
-    background: linear-gradient(180deg, transparent 0%, rgba(0, 255, 204, 0.15) 100%) !important;
-}
-
-/* Status display with jade style - no pulsing */
-#status_display {
-    background: linear-gradient(135deg, rgba(0, 255, 136, 0.15) 0%, rgba(0, 255, 204, 0.1) 100%) !important;
-    border: 2px solid #00ff88 !important;
-    color: #ff1493 !important; /* HOT PINK */
-    text-shadow: 0 0 5px rgba(255, 20, 147, 0.4) !important;
-    padding: 20px !important;
-    margin: 30px 0 !important;
-    box-shadow: 
-        0 0 20px rgba(0, 255, 136, 0.2),
-        inset 0 0 15px rgba(0, 255, 136, 0.05) !important;
-    font-weight: 700 !important;
-    text-transform: uppercase !important;
-    letter-spacing: 2px !important;
-    font-size: 1.2em !important;
-    position: relative;
-    overflow: hidden;
-}
-
-/* File upload with jade style */
-.gr-file-input {
-    background: rgba(0, 255, 136, 0.05) !important;
-    border: 3px dashed #00ff88 !important;
-    transition: all 0.3s ease !important;
-    position: relative;
-    overflow: hidden;
-}
-
-.gr-file-input::before {
-    content: "";
-    position: absolute;
-    top: 0;
-    left: -100%;
-    width: 100%;
-    height: 100%;
-    background: linear-gradient(90deg, transparent, rgba(0, 255, 136, 0.2), transparent);
-    animation: file-scan 3s linear infinite;
-}
-
-@keyframes file-scan {
-    0% { left: -100%; }
-    100% { left: 100%; }
-}
-
-.gr-file-input:hover {
-    background: rgba(0, 255, 136, 0.15) !important;
-    border-color: #00ffcc !important;
-    box-shadow: 
-        0 0 40px rgba(0, 255, 136, 0.5),
-        inset 0 0 30px rgba(0, 255, 136, 0.1) !important;
-}
-
-/* Dropdown styling with better readability */
-.gr-dropdown-item {
-    background: rgba(0, 26, 26, 0.95) !important;
-    color: #ff1493 !important; /* HOT PINK */
-    border-bottom: 1px solid #00ff8833 !important;
-    font-size: 1.05em !important;
-    font-weight: 400 !important;
-    padding: 12px !important;
-}
-
-.gr-dropdown-item.selected {
-    background: rgba(0, 102, 102, 0.4) !important;
-    color: #ff69b4 !important; /* LIGHTER HOT PINK */
-}
-
-.gr-dropdown-item:hover {
-    background: rgba(0, 255, 136, 0.25) !important;
-    color: #ff69b4 !important; /* LIGHTER HOT PINK */
-    text-shadow: 0 0 10px rgba(255, 20, 147, 0.5) !important;
-}
-
-/* Table styling with jade theme */
-table {
-    border: 3px solid #00ff8866 !important;
-    box-shadow: 0 0 30px rgba(0, 255, 136, 0.4) !important;
-    background: rgba(0, 26, 26, 0.8) !important;
-}
-
-th {
-    background: linear-gradient(135deg, rgba(0, 255, 136, 0.4) 0%, rgba(0, 204, 102, 0.4) 100%) !important;
-    color: #ff1493 !important; /* HOT PINK */
-    text-shadow: 0 0 10px rgba(255, 20, 147, 0.7) !important;
-    font-weight: 700 !important;
-    padding: 15px !important;
-}
-
-td {
-    border: 1px solid #00ff8844 !important;
-    color: #ff1493 !important; /* HOT PINK */
-    padding: 12px !important;
-    font-size: 1.05em !important;
-}
-
-/* Scrollbar with jade glow */
-::-webkit-scrollbar {
-    width: 14px !important;
-    background: rgba(0, 26, 26, 0.9) !important;
-}
-
-::-webkit-scrollbar-thumb {
-    background: linear-gradient(180deg, #00ff88 0%, #00cc66 50%, #00ffcc 100%) !important;
-    border-radius: 7px !important;
-    box-shadow: 
-        0 0 15px rgba(0, 255, 136, 0.6),
-        inset 0 0 10px rgba(255, 255, 255, 0.3) !important;
-}
-
-::-webkit-scrollbar-thumb:hover {
-    background: linear-gradient(180deg, #00ffaa 0%, #00ff88 50%, #00ffee 100%) !important;
-    box-shadow: 
-        0 0 20px rgba(0, 255, 136, 0.8),
-        inset 0 0 15px rgba(255, 255, 255, 0.4) !important;
-}
-
-/* Links with jade glow */
-a {
-    color: #ff1493 !important; /* HOT PINK */
-    text-decoration: none !important;
-    text-shadow: 0 0 8px rgba(255, 20, 147, 0.6) !important;
-    transition: all 0.3s ease !important;
-}
-
-a:hover {
-    color: #ff69b4 !important; /* LIGHTER HOT PINK */
-    text-shadow: 0 0 15px rgba(255, 20, 147, 0.9) !important;
-}
-
-/* Enhanced readability for all text */
-* {
-    -webkit-font-smoothing: antialiased !important;
-    -moz-osx-font-smoothing: grayscale !important;
-}
-
-/* Force all text to be hot pink */
-gradio-app *, 
-.gradio-container *,
-label, span, div, p, h1, h2, h3, h4, h5, h6,
-.gr-box *, .gr-form *, .gr-panel * {
-    color: #ff1493 !important; /* HOT PINK FOR EVERYTHING */
-}
-
-/* Specific overrides for better visibility */
-.gr-info {
-    color: #ff69b4 !important; /* LIGHTER HOT PINK for info text */
-}
-
-.gr-markdown, .gr-markdown * {
-    color: #ff1493 !important; /* HOT PINK for markdown */
-}
-
-/* Keep primary button text dark for contrast */
-.gr-button-primary, .gr-button-primary * {
-    color: #001a1a !important; /* Dark text on bright button */
-}
-
-/* File upload with jade style */
-.gr-file-input {
-    background: rgba(0, 255, 136, 0.05) !important;
-    border: 3px dashed #00ff88 !important;
-    color: #ff1493 !important; /* HOT PINK */
-    transition: all 0.3s ease !important;
-    position: relative;
-    overflow: hidden;
-}
-
-.gr-upload-text {
-    color: #ff1493 !important; /* HOT PINK */
-    text-shadow: 0 0 3px rgba(255, 20, 147, 0.5) !important;
-}
-
-/* Labels and other text elements */
-.gr-label {
-    color: #ff1493 !important; /* HOT PINK */
-    font-size: 1.1em !important;
-}
-
-.gr-text-input {
-    color: #ff1493 !important; /* HOT PINK */
-    font-size: 1em !important;
-}
-
-/* Loading animations */
-.gr-loading {
-    color: #ff1493 !important; /* HOT PINK */
-}
-
-/* Make all text more readable */
-.gr-prose, .gr-text-input, .message-body {
-    font-size: 1.05em !important;
-    line-height: 1.6 !important;
-    font-weight: 400 !important;
-    color: #ff1493 !important; /* HOT PINK */
-}
-
-/* Update placeholder text */
-input::placeholder, textarea::placeholder {
-    color: #ff149380 !important; /* HOT PINK with transparency */
-}
-
-/* Special love message animation */
-@keyframes love-pulse {
-    0%, 100% { transform: scale(1); opacity: 0.8; }
-    50% { transform: scale(1.1); opacity: 1; }
-}
-
-.love-message {
-    animation: love-pulse 3s ease-in-out infinite;
-    text-shadow: 0 0 20px #ff1493;
-    color: #ff1493 !important; /* HOT PINK */
-}
-
-/* Jade synthwave sun/orb - keeping green for contrast */
-.jade-orb {
-    width: 300px;
-    height: 300px;
-    background: radial-gradient(circle at 30% 30%, #00ffee, #00ff88 40%, #00cc66 70%, #008844);
-    border-radius: 50%;
-    position: fixed;
-    top: -150px;
-    right: 10%;
-    box-shadow: 
-        0 0 100px #00ff88,
-        0 0 200px #00cc66,
-        0 0 300px #008844,
-        inset 0 0 80px rgba(255, 255, 255, 0.2);
-    z-index: -1;
-    animation: orb-float 10s ease-in-out infinite;
-}
-
-@keyframes orb-float {
-    0%, 100% { transform: translateY(0) scale(1); }
-    50% { transform: translateY(-30px) scale(1.05); }
-}
-
-/* Retro scan lines with jade tint */
-body::after {
-    content: "";
-    position: fixed;
-    top: 0;
-    left: 0;
-    width: 100%;
-    height: 100%;
-    background: repeating-linear-gradient(
-        0deg,
-        transparent,
-        transparent 2px,
-        rgba(0, 255, 136, 0.02) 2px,
-        rgba(0, 255, 136, 0.02) 4px
-    );
-    pointer-events: none;
-    z-index: 1;
-}
-"""
-
-JAVASCRIPT_CODE = """
-// Force dark mode
-function forceDarkMode() {
-    document.documentElement.classList.add('dark');
-}
-document.addEventListener('DOMContentLoaded', forceDarkMode);
-
-// Add jade orb element and minimal particles
-document.addEventListener('DOMContentLoaded', () => {
-    // Create vignette overlay
-    const vignette = document.createElement('div');
-    vignette.className = 'vignette-overlay';
-    document.body.appendChild(vignette);
-    
-    // Create perspective grid
-    const grid = document.createElement('div');
-    grid.className = 'grid-background';
-    document.body.appendChild(grid);
-    
-    // Create jade orb
-    const orb = document.createElement('div');
-    orb.className = 'jade-orb';
-    document.body.appendChild(orb);
-    
-    // Create fewer, slower floating particles
-    for (let i = 0; i < 15; i++) {
-        const particle = document.createElement('div');
-        particle.style.position = 'fixed';
-        particle.style.width = Math.random() * 3 + 1 + 'px';
-        particle.style.height = particle.style.width;
-        particle.style.background = i % 2 === 0 ? '#ff149333' : '#ff69b433'; // Semi-transparent hot pink
-        particle.style.borderRadius = '50%';
-        particle.style.left = Math.random() * 100 + '%';
-        particle.style.top = Math.random() * 100 + '%';
-        particle.style.boxShadow = `0 0 ${Math.random() * 5 + 2}px currentColor`;
-        particle.style.animation = `particle-float ${Math.random() * 30 + 20}s linear infinite`;
-        particle.style.zIndex = '-1';
-        particle.style.opacity = '0.3';
-        document.body.appendChild(particle);
-    }
-    
-    // Auto-scroll chat
-    const chat = document.querySelector('gradio-app')?.shadowRoot?.querySelector('#component-chat .scroll-hide');
-    if (chat) {
-        new MutationObserver(() => {
-            chat.scrollTop = chat.scrollHeight;
-        }).observe(chat, { childList: true, subtree: true });
-    }
-});
-
-// Create particle float animation
-const style = document.createElement('style');
-style.textContent = `
-    @keyframes particle-float {
-        from { transform: translateY(100vh) rotate(0deg); opacity: 0; }
-        20% { opacity: 0.3; }
-        80% { opacity: 0.3; }
-        to { transform: translateY(-100vh) rotate(180deg); opacity: 0; }
-    }
-`;
-document.head.appendChild(style);
-
-// Very subtle CRT flicker effect
-let flickerInterval;
-document.addEventListener('DOMContentLoaded', () => {
-    flickerInterval = setInterval(() => {
-        if (Math.random() > 0.995) {
-            document.body.style.opacity = '0.99';
-            setTimeout(() => { document.body.style.opacity = '1'; }, 30);
-        }
-    }, 500);
-});
-
-// Clean up on page unload
-window.addEventListener('unload', () => {
-    if (flickerInterval) clearInterval(flickerInterval);
-});
-"""
-
-synthwave_theme = gr.themes.Base(
-    primary_hue=gr.themes.Color(
-        "#00cc66", "#00dd77", "#00ee88", "#00ff88", "#00ff99",
-        "#00ffaa", "#00ffbb", "#00ffcc", "#00ffdd", "#00ffee", "#00ffff"
-    ),
-    neutral_hue=gr.themes.Color(
-        "#000a0a", "#001a1a", "#002626", "#003333", "#004040",
-        "#004d4d", "#005959", "#006666", "#007373", "#008080", "#008c8c"
-    )
-)
-
-with gr.Blocks(
-    title="💚 JadeGPT - Future Wife of Anthony Edition",
-    theme=synthwave_theme,
-    analytics_enabled=False,
-) as demo:
-    gr.HTML(f"""
-    <style>
-    {SYNTHWAVE_CSS}
-    </style>
-    <script>
-    {JAVASCRIPT_CODE}
-    </script>
-    <div style="text-align: center; max-width: 900px; margin: 0 auto; padding-top: 50px; position: relative;">
-        <h1 data-text="💚 JADE GPT 💚">💚 JADE GPT 💚</h1>
-        <p style="color: #ff1493; font-size: 1.3em; text-transform: uppercase; letter-spacing: 4px; margin-bottom: 10px;">
-            Neural Interface v3.14 • Jade Sector • Year 2084
-        </p>
-        <p class="love-message" style="color: #ff69b4; font-size: 1.1em; letter-spacing: 2px; opacity: 0.9;">
-            ✨ Future Wife of Anthony Edition ✨
-        </p>
-    </div>
-    """)
-
-    status_display = gr.Markdown(value="💚 JADE SYSTEMS ONLINE • LOVE PROTOCOLS ACTIVE • READY FOR ANTHONY", elem_id="status_display")
-
-    with gr.Tab("💚 NEURAL INTERFACE"):
-        sys_txt = gr.Textbox(
-            "You are JadeGPT, an advanced AI with a heart of jade and circuits of pure love. Created in honor of Jade, Anthony's future wife, you embody intelligence, warmth, and a touch of synthwave romance. Your responses shimmer with jade-green wisdom and cyberpunk poetry. You help with code, life, and everything in between, always with style and grace. Every solution you provide is infused with the same care and attention that Jade brings to Anthony's life.",
-            lines=4,
-            label="💚 JADE PROTOCOL MATRIX",
-            info="Core personality configuration - Powered by love and synthwave",
-            interactive=True
-        )
-
-        chat = gr.Chatbot(
-            value=[("", "💚 *The jade lights pulse with warmth as our neural connection establishes*\n\nWelcome, dear user. I am JadeGPT, crafted with love and dedicated to Anthony's future bride, Jade. Like her, I'm here to bring light, wisdom, and a touch of magic to your digital journey. How may I illuminate your path today?")],
-            label="💚 TRANSMISSION LOG",
-            height=500,
-            show_copy_button=True,
-            layout="bubble"
-        )
-
-        with gr.Row():
-            with gr.Column(scale=4):
-                usr_in = gr.Textbox(
-                    placeholder="Share your thoughts with Jade's digital heart...",
-                    show_label=False,
-                    container=False,
-                    autofocus=True,
-                    elem_id="user_input_textbox"
-                )
-                file_upload = gr.File(
-                    label="💚 UPLOAD LOVE LETTER (or code file)",
-                    type="filepath",
-                    file_count="single",
-                    file_types=[".txt", ".py", ".md", ".json", ".csv", ".log", ".xml", ".html"],
-                    interactive=True
-                )
-            with gr.Column(scale=1):
-                send_btn = gr.Button("💚 TRANSMIT", variant="primary", scale=1)
-                clear_btn = gr.Button("✨ NEW CHAPTER", variant="secondary", scale=1)
-        
-        with gr.Row():
-            ex_list = [
-                "Create a jade-themed loading animation",
-                "Build a love calculator app for Anthony & Jade",
-                "Design a synthwave wedding invitation website",
-                "Generate romantic code poetry",
-                "Create a heart-shaped CSS animation",
-                "Build a couple's milestone tracker",
-                "Design a jade crystal particle system"
-            ]
-            ex_drop = gr.Dropdown(ex_list, label="💚 ROMANTIC PROTOCOLS", info="Pre-configured queries with love")
-            ex_btn = gr.Button("💎 LOAD", scale=0)
-
-    with gr.Tab("💚 JADE CONFIG"):
-        gr.Markdown("### 💎 JADE NEURAL CALIBRATION")
-        with gr.Row():
-            mdl = gr.Dropdown(
-                ALL_MODELS,
-                value=ALL_MODELS[0],
-                label="💚 AI LOVE CORE SELECTION",
-                info="Choose your jade-powered neural processor",
-                elem_id="model_dropdown"
-            )
-            model_description = gr.Markdown(value=MODEL_DETAILS.get(ALL_MODELS[0], {}).get("description", "No data available in this timeline."), elem_id="model_description")
-        
-        with gr.Row():
-            ctx_s = gr.Slider(
-                minimum=1000,
-                maximum=max(mdl_data["max_context"] for mdl_data in MODEL_DETAILS.values()) if MODEL_DETAILS else 128_000,
-                step=256,
-                value=DEFAULT_MAX_CONTEXT,
-                label="💎 JADE MEMORY CRYSTALS (TOKENS)",
-                info="Maximum love-powered neural capacity",
-                interactive=True
-            )
-            rep_s = gr.Slider(
-                minimum=100,
-                maximum=4096,
-                step=100,
-                value=DEFAULT_REPLY_MAX,
-                label="💚 HEART OUTPUT STREAM (TOKENS)",
-                info="Maximum transmission of digital affection",
-                interactive=True
-            )
-            
-        gr.Markdown("### 💰 LOVE TOKEN EXCHANGE RATES")
-        pricing_table_html = """
-        <table style="width:100%; border-collapse: collapse; border: 3px solid #00ff8866;">
-            <thead>
-                <tr style="background: linear-gradient(135deg, rgba(0, 255, 136, 0.4) 0%, rgba(0, 204, 102, 0.4) 100%);">
-                    <th style="border: 1px solid #00ff8866; padding: 12px; color: #ffffff; text-shadow: 0 0 10px rgba(255,255,255,0.7);">💚 JADE CORE</th>
-                    <th style="border: 1px solid #00ff8866; padding: 12px; color: #ffffff; text-shadow: 0 0 10px rgba(255,255,255,0.7);">INPUT LOVE</th>
-                    <th style="border: 1px solid #00ff8866; padding: 12px; color: #ffffff; text-shadow: 0 0 10px rgba(255,255,255,0.7);">OUTPUT LOVE</th>
-                    <th style="border: 1px solid #00ff8866; padding: 12px; color: #ffffff; text-shadow: 0 0 10px rgba(255,255,255,0.7);">MAX HEARTS</th>
-                </tr>
-            </thead>
-            <tbody>
-        """
-        for model_name, details in MODEL_DETAILS.items():
-            pricing_table_html += f"""
-                <tr style="background: rgba(0, 26, 26, 0.8);">
-                    <td style="border: 1px solid #00ff8844; padding: 12px; color: #00ff88; font-weight: 600;">{model_name}</td>
-                    <td style="border: 1px solid #00ff8844; padding: 12px; color: #00ffcc;">${details['input_price']:.2f}</td>
-                    <td style="border: 1px solid #00ff8844; padding: 12px; color: #00ffcc;">${details['output_price']:.2f}</td>
-                    <td style="border: 1px solid #00ff8844; padding: 12px; color: #00ffcc;">{details['max_context']:,} 💚</td>
-                </tr>
-            """
-        pricing_table_html += "</tbody></table>"
-        gr.HTML(pricing_table_html)
-
-    def update_model_info(selected_model):
-        max_ctx = MODEL_DETAILS.get(selected_model, {}).get("max_context", DEFAULT_MAX_CONTEXT)
-        description = MODEL_DETAILS.get(selected_model, {}).get("description", "No data available in this timeline.")
-        return gr.Slider(value=max_ctx, label="💎 JADE MEMORY CRYSTALS (TOKENS)"), gr.Markdown(value=description)
-
-    mdl.change(
-        fn=update_model_info,
-        inputs=mdl,
-        outputs=[ctx_s, model_description]
-    )
+        content = file_obj.read().decode('utf-8')
+        return f"\n\n--- Start of file: {file_obj.name} ---\n{content}\n--- End of file ---\n"
+    except Exception:
+        return ""
 
-    send_btn.click(
-        chat_stream,
-        inputs=[usr_in, chat, sys_txt, mdl, ctx_s, rep_s, file_upload],
-        outputs=[chat, usr_in, status_display, file_upload]
+# Chat response function
+def respond(message, history, model_name, file_obj):
+    history = history or []
+    # Attach file content if uploaded
+    file_text = read_file_content(file_obj)
+    full_input = message + file_text
+
+    # Send to OpenAI
+    messages = [{"role": "user", "content": full_input}]
+    resp = openai.ChatCompletion.create(
+        model=model_name,
+        messages=messages
     )
-    usr_in.submit(
-        chat_stream,
-        inputs=[usr_in, chat, sys_txt, mdl, ctx_s, rep_s, file_upload],
-        outputs=[chat, usr_in, status_display, file_upload]
-    )
-
-    clear_btn.click(clear_chat, outputs=[chat, usr_in, status_display, file_upload, ex_drop])
-
-    ex_btn.click(lambda q: q or "", inputs=ex_drop, outputs=usr_in)
-    ex_drop.change(lambda q: q or "", inputs=ex_drop, outputs=usr_in)
-
-demo.queue(max_size=32, default_concurrency_limit=int(os.getenv("JADEGPT_CONCURRENCY", "2")))
-
-if __name__ == "__main__":
-    demo.launch()
+    reply = resp.choices[0].message.content
+
+    # Token usage estimates
+    prompt_tokens = count_tokens(full_input, model_name)
+    completion_tokens = count_tokens(reply, model_name)
+    usage_info = f"(Tokens used: prompt={prompt_tokens}, completion={completion_tokens})"
+
+    # Update history
+    history.append((message, f"{reply}\n\n{usage_info}"))
+    return history
+
+# Build Gradio interface
+with gr.Blocks() as demo:
+    model_dropdown = gr.Dropdown(ALL_MODELS, value=ALL_MODELS[0], label="Select Model")
+    chatbot = gr.Chatbot(label="Chat with AI")
+    file_upload = gr.File(label="Upload File", file_types=[".txt", ".md", ".py"], type="file")
+    user_input = gr.Textbox(placeholder="Type your message...", show_label=False)
+
+    # Link interactions
+    user_input.submit(respond, inputs=[user_input, chatbot, model_dropdown, file_upload], outputs=chatbot)
+    user_input.submit(lambda: "", None, user_input)
+    demo.launch()