Update app.py

app.py

@@ -27,10 +27,6 @@ def count_tokens(text: str, model: str) -> int:
     return len(enc.encode(text))
 
 def trim_conversation(convo: list[dict], model: str, max_context: int) -> list[dict]:
-    """
-    Slide-window: drop the oldest user/assistant turns until
-    total tokens + BUFFER_TOKENS ≤ max_context.
-    """
     tokens = [count_tokens(m["content"], model) for m in convo]
     total  = sum(tokens)
     while total + BUFFER_TOKENS > max_context and len(convo) > 2:
@@ -41,11 +37,6 @@ def trim_conversation(convo: list[dict], model: str, max_context: int) -> list[d
     return convo
 
 def safe_chat(convo: list[dict], max_context: int, max_reply: int, model_list: list[str]):
-    """
-    1) Try each model in model_list in order.
-    2) On model-not-found: skip to next.
-    3) On context-length error: trim history & retry same model once.
-    """
     last_exc = None
     for model in model_list:
         try:
@@ -57,11 +48,9 @@ def safe_chat(convo: list[dict], max_context: int, max_reply: int, model_list: l
             )
         except Exception as e:
             msg = str(e).lower()
-            # model-not-found → fallback
             if "does not exist" in msg or "model_not_found" in msg or "404" in msg:
                 last_exc = e
                 continue
-            # context-length → trim & retry
             if "context length" in msg or "maximum context length" in msg:
                 trimmed = trim_conversation(convo.copy(), model, max_context)
                 try:
@@ -74,9 +63,7 @@ def safe_chat(convo: list[dict], max_context: int, max_reply: int, model_list: l
                 except Exception as e2:
                     last_exc = e2
                     continue
-            # other errors → bubble up
             raise
-    # none succeeded
     raise last_exc or RuntimeError("All models failed in safe_chat()")
 
 def chat_handler(
@@ -87,20 +74,17 @@ def chat_handler(
     max_context: int,
     max_reply: int
 ) -> tuple[list[tuple[str, str]], str]:
-    """Gradio handler: builds convo, calls safe_chat, updates history."""
     if not user_message.strip():
         return history, ""
     if not client.api_key:
         return history, "❌ OPENAI_API_KEY not set."
 
-    # Build conversation payload
     convo = [{"role":"system","content":system_prompt}]
     for u, b in history or []:
         convo.append({"role":"user",      "content":u})
         convo.append({"role":"assistant", "content":b})
     convo.append({"role":"user","content":user_message})
 
-    # Prepare model list: user choice first, then the rest
     fallback = [m for m in ALL_MODELS if m != selected_model]
     models_to_try = [selected_model] + fallback
 
@@ -115,18 +99,13 @@ def chat_handler(
     return history, ""
 
 def clear_chat_handler() -> list:
-    """Clear all previous chat messages."""
     return []
 
 # ——— Gradio UI ———
-with gr.Blocks(
-    title="🤖 CodeBot: Your Cool Coding Assistant"
-) as demo:
-
+with gr.Blocks(title="🤖 CodeBot: Your Cool Coding Assistant") as demo:
     gr.Markdown(
         """
         ## CodeBot  
-        A stylish coding assistant with:
         - **Model selector**: pick your LLM  
         - **Context slider**: control how much history to keep  
         - **Reply slider**: set max response length  
@@ -153,27 +132,17 @@ with gr.Blocks(
             label="✍️ Max Reply Tokens"
         )
 
-    # Example questions
     examples = [
         "How do I implement quicksort in Python?",
         "Show me a C# example using LINQ to group items.",
         "Explain async/await in Python with sample code.",
         "How to connect to SQL Server using C#?"
     ]
+
     with gr.Row():
-        example_dropdown = gr.Dropdown(
-            choices=examples,
-            label="💡 Examples"
-        )
+        example_dropdown = gr.Dropdown(choices=examples, label="💡 Examples")
         example_btn = gr.Button("📥 Load Example")
 
-    # Prefill user_input when an example is selected
-    example_btn.click(
-        fn=lambda q: q or "",
-        inputs=[example_dropdown],
-        outputs=[gr.update(component_id="user_input")]
-    )
-
     system_txt = gr.Textbox(
         lines=3,
         value=(
@@ -194,6 +163,13 @@ with gr.Blocks(
         elem_id="user_input"
     )
 
+    # Wire up example loader to the actual user_input component
+    example_btn.click(
+        fn=lambda q: q or "",
+        inputs=[example_dropdown],
+        outputs=[user_input]
+    )
+
     with gr.Row():
         send_btn  = gr.Button("🚀 Send")
         clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")