Update app.py

app.py

@@ -3,8 +3,8 @@ import subprocess
 from huggingface_hub import hf_hub_download
 
 # 1. Install llama-cpp-python in runtime (not via requirements.txt)
-#    Important: remove `llama-cpp-python` from requirements.txt
-#    or Spaces will try to build from source and get stuck.
+#    Important: remove `llama-cpp-python` from requirements.txt,
+#    otherwise Spaces may try to build from source and get stuck.
 subprocess.run("pip install -q 'llama_cpp_python==0.3.15'", shell=True, check=False)
 from llama_cpp import Llama
 
@@ -52,7 +52,7 @@ def _extract_text_from_content(content):
     In Gradio 6 ChatInterface, history uses the messages format.
     content can be:
       - a string
-      - a list of blocks: [{"type": "text", "text": "..."} , ...]
+      - a list of blocks: [{'type': 'text', 'text': '...'} , ...]
     We convert it into a simple string.
     """
     if isinstance(content, list):
@@ -89,7 +89,7 @@ def build_prompt(message, history, style):
     prompt_parts.append(f"System: {system_prompt}\n")
     prompt_parts.append("Conversation:\n")
 
-    # history is a list of dicts: {"role": "...", "content": ...}
+    # history is a list of dicts: {'role': 'user'/'assistant'/'system', 'content': ...}
     for msg in history or []:
         role = msg.get("role")
         content = _extract_text_from_content(msg.get("content", ""))
@@ -187,10 +187,9 @@ style_radio = gr.Radio(
     label="Answer style",
 )
 
-# 5. Christmas theme CSS
-# Assumes the image "Cute-Christmas-Background-edit-online-1.jpg"
-# is in the Space repository root, so we can reference it as file=...
-christmas_css = """
+# 5. Christmas theme: inject CSS + hero directly into description
+christmas_style_and_hero = """
+<style>
 body {
     background: radial-gradient(circle at top, #1b1c2b 0, #050611 55%, #000000 100%);
     color: #fdf6e3;
@@ -368,53 +367,43 @@ input[type="range"] {
     background: rgba(255, 255, 255, 0.5);
     border-radius: 999px;
 }
+</style>
+
+<div class="hero">
+  <div class="hero-inner">
+    <div class="hero-badge"></div>
+    <div class="hero-text">
+      <h1>Scalable Lab 2 Christmas Chat</h1>
+      <p>
+        Talk to our fine tuned Llama based model, wrapped as a compact GGUF
+        and running on CPU. Use the controls in the accordion below to tune
+        response length, randomness and style like a Christmas DJ for language models.
+      </p>
+    </div>
+  </div>
+  <div class="hero-keyline"></div>
+</div>
+
+<p>
+🎄 <strong>Tip:</strong> Try switching between short answers and step by step reasoning,
+and play with temperature and top p to see how the model behaves.
+</p>
 """
 
 
-# 6. Build the UI with Blocks so we can add a hero section
-with gr.Blocks(css=christmas_css) as demo:
-    gr.HTML(
-        """
-        <div class="hero">
-          <div class="hero-inner">
-            <div class="hero-badge"></div>
-            <div class="hero-text">
-              <h1>Scalable Lab 2 Christmas Chat</h1>
-              <p>
-                Talk to our fine tuned Llama based model, wrapped as a compact GGUF
-                and running on CPU. Use the controls below to tune response length,
-                randomness and style like a Christmas DJ for language models.
-              </p>
-            </div>
-          </div>
-          <div class="hero-keyline"></div>
-        </div>
-        """
-    )
-
-    gr.Markdown(
-        "🎄 **Tip:** Try switching between short answers and step by step reasoning, "
-        "and play with temperature and top p to see how the model behaves."
-    )
-
-    gr.ChatInterface(
-        fn=chat_fn,
-        title="Lab 2 – Fine tuned GGUF model",
-        description=(
-            "Chat with our fine tuned Llama based model, converted to GGUF and "
-            "loaded via llama.cpp from `Jeppcode/ScalableLab2`.\n\n"
-            "Use the controls in the accordion below like a DJ board to tweak "
-            "response length, randomness and style."
-        ),
-        additional_inputs=[
-            max_new_tokens_slider,
-            temperature_slider,
-            top_p_slider,
-            repetition_penalty_slider,
-            style_radio,
-        ],
-        additional_inputs_accordion="Generation controls",
-    )
+demo = gr.ChatInterface(
+    fn=chat_fn,
+    title="Lab 2 – Fine-tuned GGUF model",
+    description=christmas_style_and_hero,
+    additional_inputs=[
+        max_new_tokens_slider,
+        temperature_slider,
+        top_p_slider,
+        repetition_penalty_slider,
+        style_radio,
+    ],
+    additional_inputs_accordion="Generation controls",
+)
 
 if __name__ == "__main__":
     demo.launch()