Update app.py

app.py

@@ -2,12 +2,13 @@ import gradio as gr
 import subprocess
 from huggingface_hub import hf_hub_download
 
-# 1. Install llama-cpp-python in runtime
+# --- 1. Setup & Install ---
+# Install llama-cpp-python in runtime (prevents build errors in Spaces)
 subprocess.run("pip install -q 'llama_cpp_python==0.3.15'", shell=True, check=False)
 
 from llama_cpp import Llama
 
-# 2. Load GGUF model
+# --- 2. Load Model (GGUF) ---
 MODEL_REPO = "Jeppcode/ScalableLab2"
 GGUF_FILENAME = "model-q4_k_m.gguf" 
 
@@ -27,69 +28,45 @@ llm = Llama(
     use_mlock=False,
 )
 
-# 3. System Prompts
+# --- 3. Prompts & Logic ---
 STYLE_SYSTEM_PROMPTS = {
     "Default": "You are a helpful, polite assistant.",
-    "Short answer": (
-        "You are a helpful assistant. Answer as concisely as possible, usually in 1–3 sentences."
-    ),
-    "Detailed explanation": (
-        "You are a helpful teaching assistant. Give clear, structured and detailed explanations, "
-        "often with bullet points or numbered steps when useful."
-    ),
-    "Step-by-step reasoning": (
-        "You are a careful problem solver. Think step by step and explain your reasoning clearly "
-        "before giving the final answer."
-    ),
+    "Short answer": "You are a helpful assistant. Answer as concisely as possible, usually in 1–3 sentences.",
+    "Detailed explanation": "You are a helpful teaching assistant. Give clear, structured and detailed explanations.",
+    "Step-by-step reasoning": "You are a careful problem solver. Think step by step and explain your reasoning clearly.",
 }
 
-def _extract_text_from_content(content):
+def _extract_text(content):
     if isinstance(content, list):
-        texts = []
-        for block in content:
-            if isinstance(block, dict) and block.get("type") == "text":
-                texts.append(block.get("text", ""))
-            else:
-                texts.append(str(block))
-        return "\n".join(t for t in texts if t)
-    else:
-        return str(content)
+        return "\n".join(
+            block.get("text", "") for block in content 
+            if isinstance(block, dict) and block.get("type") == "text"
+        )
+    return str(content)
 
 def build_prompt(message, history, style):
     system_prompt = STYLE_SYSTEM_PROMPTS.get(style, STYLE_SYSTEM_PROMPTS["Default"])
-    prompt_parts = []
-    prompt_parts.append(f"System: {system_prompt}\n")
-    prompt_parts.append("Conversation:\n")
+    prompt_parts = [f"System: {system_prompt}\n", "Conversation:\n"]
     
     for msg in history or []:
         role = msg.get("role")
-        content = _extract_text_from_content(msg.get("content", ""))
-        if not content:
-            continue
-        if role == "user":
-            prompt_parts.append(f"User: {content}\n")
-        elif role == "assistant":
-            prompt_parts.append(f"Assistant: {content}\n")
-        elif role == "system":
-            prompt_parts.append(f"System (previous): {content}\n")
+        content = _extract_text(msg.get("content", ""))
+        if content:
+            if role == "user": prompt_parts.append(f"User: {content}\n")
+            elif role == "assistant": prompt_parts.append(f"Assistant: {content}\n")
             
     prompt_parts.append(f"User: {message}\n")
     prompt_parts.append("Assistant:")
     return "".join(prompt_parts)
 
 def chat_fn(message, history, max_new_tokens, style):
-    # Removed sliders are now hardcoded defaults here
+    # Hardcoded values for the hidden sliders
     temperature = 0.7
     top_p = 0.9
     repetition_penalty = 1.1
 
     prompt = build_prompt(message, history, style)
     
-    # Simple logic for temperature
-    if temperature <= 0.0:
-        temperature = 0.0
-        top_p = 1.0
-
     output = llm(
         prompt,
         max_tokens=int(max_new_tokens),
@@ -98,105 +75,94 @@ def chat_fn(message, history, max_new_tokens, style):
         repeat_penalty=repetition_penalty,
         stop=["User:", "Assistant:", "System:", "Conversation:"],
     )
-    reply = output["choices"][0]["text"].strip()
-    return reply
+    return output["choices"][0]["text"].strip()
 
-# --- Christmas Theme Configuration ---
+# --- 4. Winter/Christmas Theme Configuration ---
 
-# 1. THEME: Red (Santa) and Green (Tree)
-christmas_theme = gr.themes.Soft(
-    primary_hue="red",
-    secondary_hue="green",
+# We create a base theme, but most work is done in CSS
+winter_theme = gr.themes.Soft(
+    primary_hue="blue", # Ice blue accents
     neutral_hue="slate",
 ).set(
     body_background_fill="transparent",
-    block_background_fill="rgba(255, 250, 240, 0.9)", # Creamy white snow color
-    border_color_primary="#D4AF37", # Gold Border
-    button_primary_background_fill="#C62828", # Santa Red
-    button_primary_text_color="white",
+    block_background_fill="rgba(10, 20, 40, 0.7)", # Dark Blue Glass
+    border_color_primary="#FFFFFF", # Explicit White Border
+    button_primary_background_fill="#FFFFFF", # White buttons
+    button_primary_text_color="#000000",
+    text_color_subdued="#E0E0E0",
 )
 
-# 2. CSS: Background image + Festive Styling
+# Custom CSS for the "Perfect" Look
 custom_css = """
-/* Background: A cozy Christmas scene */
+/* 1. Background Image: Snowy Winter Forest */
 .gradio-container {
-    background: url('https://images.unsplash.com/photo-1544976735-a10c71a39644?q=80&w=2560&auto=format&fit=crop') no-repeat center center fixed; 
+    background: url('https://images.unsplash.com/photo-1477601263568-180e2c6d046e?q=80&w=2560&auto=format&fit=crop') no-repeat center center fixed; 
     background-size: cover;
 }
 
-/* Make main container transparent */
+/* 2. Transparency */
 .gradio-container > .main {
     background: transparent !important;
 }
 
-/* Chatbot Window - Glassy Snow Look */
-.bubble-wrap {
-    background: rgba(255, 255, 255, 0.85) !important;
-    border: 2px solid #D4AF37 !important; /* Gold Border */
-    border-radius: 15px !important;
+/* 3. The Chat & Settings Blocks - "Frosty Glass" with White Borders */
+.group, .form, .bubble-wrap {
+    background: rgba(15, 23, 42, 0.75) !important; /* Dark Blue/Grey Glass */
+    border: 2px solid #FFFFFF !important; /* THE WHITE BORDER */
+    border-radius: 12px !important;
+    backdrop-filter: blur(4px); /* Slight blur behind the glass */
+    box-shadow: 0 4px 15px rgba(0,0,0,0.5);
 }
 
-/* User Message - Christmas Red */
+/* 4. Chat Bubbles */
 .user-message {
-    background-color: #D32F2F !important; 
-    color: white !important;
-    border: 1px solid #B71C1C !important;
+    background-color: rgba(255, 255, 255, 0.2) !important; /* Icy White transparency */
+    border: 1px solid #FFFFFF !important;
+    color: #FFFFFF !important;
 }
 
-/* Bot Message - Christmas Green */
 .bot-message {
-    background-color: #2E7D32 !important; 
-    color: white !important;
-    border: 1px solid #1B5E20 !important;
+    background-color: rgba(0, 0, 0, 0.6) !important; /* Darker for contrast */
+    border: 1px solid #A0A0A0 !important;
+    color: #E0E0E0 !important;
 }
 
-/* Accordion/Settings - Snowy background with Gold border */
-.group, .form {
-    background: rgba(255, 255, 255, 0.9) !important;
-    border: 2px solid #D4AF37 !important;
-    border-radius: 10px;
-    padding: 10px;
+/* 5. Inputs and Text */
+textarea, input {
+    background-color: rgba(0, 0, 0, 0.5) !important;
+    border: 1px solid #FFFFFF !important;
+    color: white !important;
 }
-
-/* Labels */
-label, span {
-    color: #3E2723 !important; /* Dark chocolate text for readability */
-    font-weight: bold;
+label, span, p {
+    color: #FFFFFF !important;
+    text-shadow: 1px 1px 2px black; /* Makes text readable on snow */
 }
 
+/* Hide Footer */
 footer {visibility: hidden}
 """
 
-# 4. Inputs (Only Max Tokens + Style)
+# Only Max Tokens + Style
 max_new_tokens_slider = gr.Slider(
-    minimum=16, maximum=256, value=64, step=8, label="Max new tokens (Length)",
+    minimum=16, maximum=256, value=64, step=8, label="Max Response Length",
 )
-
 style_radio = gr.Radio(
-    choices=[
-        "Default",
-        "Short answer",
-        "Detailed explanation",
-        "Step-by-step reasoning",
-    ],
+    choices=["Default", "Short answer", "Detailed explanation"],
     value="Detailed explanation",
     label="Answer style",
 )
 
-# Instantiate ChatInterface
+# Init App
 demo = gr.ChatInterface(
     fn=chat_fn,
-    title="🎄 Holiday Chat Lab 2 🎅",
-    description="Chat with the fine-tuned model. Grab a hot chocolate and enjoy the holidays.",
-    additional_inputs=[
-        max_new_tokens_slider,
-        style_radio,
-    ],
-    additional_inputs_accordion="Holiday Settings",
+    title="❄️ WinterChat Lab 2 ❄️",
+    description="Stay frosty. Chat with the fine-tuned model.",
+    additional_inputs=[max_new_tokens_slider, style_radio],
+    additional_inputs_accordion="Settings",
 )
 
-# Apply Theme and CSS manually (Safe for older Gradio versions)
-demo.theme = christmas_theme
+# Apply Visuals manually
+demo.theme = winter_theme
 demo.css = custom_css
 
 if __name__ == "__main__":