Update app.py

app.py

@@ -1,6 +1,6 @@
 """
-app.py – Advanced Chatbot with Model Selector, Context & Reply Token Sliders
-(OpenAI Python SDK ≥1.0.0)
+app.py – Cool Coding Assistant with Model Selector, Context/Reply Sliders, Examples & Clear Chat
+(OpenAI Python SDK ≥1.0.0 + Gradio 5.34.1 + tiktoken)
 """
 
 import os
@@ -16,11 +16,11 @@ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY", "").strip())
 env_models = os.getenv("OPENAI_MODEL_LIST", "gpt-4-32k,gpt-4,gpt-3.5-turbo")
 ALL_MODELS = [m.strip() for m in env_models.split(",") if m.strip()]
 
-# ——— Default limits ———
+# ——— Configuration ———
 DEFAULT_MAX_CONTEXT = 32768   # tokens
-DEFAULT_REPLY_MAX   = 2048    # tokens
-BUFFER_TOKENS       = 500     # reserved for the model’s answer
-TEMPERATURE         = 0.3
+BUFFER_TOKENS       = 500     # reserved for the model’s reply
+DEFAULT_REPLY_MAX   = 2048    # tokens for the answer
+TEMPERATURE         = 0.3     # creativity vs determinism
 
 def count_tokens(text: str, model: str) -> int:
     enc = tiktoken.encoding_for_model(model)
@@ -34,17 +34,17 @@ def trim_conversation(convo: list[dict], model: str, max_context: int) -> list[d
     tokens = [count_tokens(m["content"], model) for m in convo]
     total  = sum(tokens)
     while total + BUFFER_TOKENS > max_context and len(convo) > 2:
-        convo.pop(1)  # oldest user
-        convo.pop(1)  # its assistant reply
+        convo.pop(1)  # remove oldest user
+        convo.pop(1)  # remove corresponding assistant reply
         tokens = [count_tokens(m["content"], model) for m in convo]
         total  = sum(tokens)
     return convo
 
 def safe_chat(convo: list[dict], max_context: int, max_reply: int, model_list: list[str]):
     """
-    Try each model in model_list:
-      - on model-not-found: skip to next
-      - on context-length: trim history & retry same model once
+    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:
@@ -57,11 +57,11 @@ def safe_chat(convo: list[dict], max_context: int, max_reply: int, model_list: l
             )
         except Exception as e:
             msg = str(e).lower()
-            # Model unavailable → fallback
+            # 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 error → slide window & retry
+            # 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 +74,9 @@ 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
+            # other errors → bubble up
             raise
-    # All models failed
+    # none succeeded
     raise last_exc or RuntimeError("All models failed in safe_chat()")
 
 def chat_handler(
@@ -93,7 +93,7 @@ def chat_handler(
     if not client.api_key:
         return history, "❌ OPENAI_API_KEY not set."
 
-    # Build the conversation payload
+    # Build conversation payload
     convo = [{"role":"system","content":system_prompt}]
     for u, b in history or []:
         convo.append({"role":"user",      "content":u})
@@ -114,15 +114,26 @@ def chat_handler(
     history.append((user_message, reply))
     return history, ""
 
+def clear_chat_handler() -> list:
+    """Clear all previous chat messages."""
+    return []
+
 # ——— Gradio UI ———
-with gr.Blocks(title="🤖 Advanced Chatbot") as demo:
+with gr.Blocks(
+    title="🤖 CodeBot: Your Cool Coding Assistant",
+    theme=gr.themes.SoftDark()
+) as demo:
+
     gr.Markdown(
         """
-        # Advanced Chatbot  
-        - **Model selector**: pick any supported OpenAI model  
-        - **Context slider**: adjust how many tokens of history to keep  
-        - **Reply slider**: adjust maximum tokens in the answer  
-        - **Sliding-window**: oldest history drops automatically when over limit  
+        ## 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  
+        - **Examples**: load sample coding questions  
+        - **Clear Chat**: reset conversation anytime  
+        - **Sliding-window**: auto-drop oldest history when over limit
         """
     )
 
@@ -130,31 +141,61 @@ with gr.Blocks(title="🤖 Advanced Chatbot") as demo:
         model_dropdown = gr.Dropdown(
             choices=ALL_MODELS,
             value=ALL_MODELS[0],
-            label="Choose Model"
+            label="🔍 Choose Model"
         )
         context_slider = gr.Slider(
             minimum=1000, maximum=DEFAULT_MAX_CONTEXT,
             step=256, value=DEFAULT_MAX_CONTEXT,
-            label="Max Context Tokens"
+            label="🗂️ Max Context Tokens"
         )
         reply_slider = gr.Slider(
             minimum=100, maximum=8192,
             step=100, value=DEFAULT_REPLY_MAX,
-            label="Max Reply Tokens"
+            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_btn = gr.Button("📥 Load Example")
+    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=(
-            "You are an expert software engineer specializing in Python and C#. "
-            "Provide detailed, production-grade answers and include code snippets when appropriate."
+            "You are CodeBot, an expert software engineer specializing in Python and C#. "
+            "Provide detailed, production-grade answers including runnable code snippets."
         ),
-        label="System Prompt"
+        label="💻 System Prompt"
     )
 
-    chatbot    = gr.Chatbot(label="Conversation")
-    user_input = gr.Textbox(placeholder="Type your message here...", label="You")
-    send_btn   = gr.Button("Send")
+    chatbot = gr.Chatbot(
+        value=[("", "👋 Hello! I'm CodeBot. How can I help you with code today?")],
+        label="💬 Conversation",
+        height=500
+    )
+    user_input = gr.Textbox(
+        placeholder="Type your question or paste code here...",
+        label="📝 Your Message",
+        elem_id="user_input"
+    )
+
+    with gr.Row():
+        send_btn  = gr.Button("🚀 Send")
+        clear_btn = gr.Button("🗑️ Clear Chat", variant="secondary")
 
     send_btn.click(
         fn=chat_handler,
@@ -168,6 +209,10 @@ with gr.Blocks(title="🤖 Advanced Chatbot") as demo:
         ],
         outputs=[chatbot, user_input]
     )
+    clear_btn.click(
+        fn=clear_chat_handler,
+        outputs=[chatbot]
+    )
 
 if __name__ == "__main__":
     demo.launch()