Update app.py

app.py

@@ -33,7 +33,6 @@ if model_path:
     )
 
 # --- 3. Style / System Prompts ---
-# These are the "Buttons" logic to change how the AI behaves
 STYLE_SYSTEM_PROMPTS = {
     "Default": "You are a helpful, polite assistant.",
     "Short answer": "Answer as concisely as possible, usually in 1–3 sentences.",
@@ -43,13 +42,17 @@ STYLE_SYSTEM_PROMPTS = {
 
 def _extract_text(content):
     if isinstance(content, list):
-        return "\n".join(b.get("text", "") for b in content if isinstance(b, dict) and b.get("type") == "text")
+        return "\n".join(
+            b.get("text", "")
+            for b in content
+            if isinstance(b, dict) and b.get("type") == "text"
+        )
     return str(content)
 
 def chat_fn(message, history, max_new_tokens, style):
-    if not llm: return "Error: Model not loaded."
+    if not llm:
+        return "Error: Model not loaded."
     
-    # Select the specific system prompt based on the button chosen
     system_prompt = STYLE_SYSTEM_PROMPTS.get(style, STYLE_SYSTEM_PROMPTS["Default"])
     
     prompt = f"System: {system_prompt}\nConversation:\n"
@@ -57,12 +60,13 @@ def chat_fn(message, history, max_new_tokens, style):
         role = msg.get("role")
         txt = _extract_text(msg.get("content", ""))
         if txt:
-            if role == "user": prompt += f"User: {txt}\n"
-            elif role == "assistant": prompt += f"Assistant: {txt}\n"
+            if role == "user":
+                prompt += f"User: {txt}\n"
+            elif role == "assistant":
+                prompt += f"Assistant: {txt}\n"
             
     prompt += f"User: {message}\nAssistant:"
 
-    # Default internal values for randomness
     output = llm(
         prompt,
         max_tokens=int(max_new_tokens),
@@ -74,7 +78,6 @@ def chat_fn(message, history, max_new_tokens, style):
 
 # --- 4. UI Controls ---
 
-# Slider for length
 max_new_tokens_slider = gr.Slider(
     minimum=16, 
     maximum=256, 
@@ -83,21 +86,34 @@ max_new_tokens_slider = gr.Slider(
     label="Max Response Length"
 )
 
-# The "Buttons" at the bottom for Style
 style_radio = gr.Radio(
     choices=["Default", "Short answer", "Detailed explanation", "Step-by-step reasoning"],
     value="Detailed explanation",
     label="Answer Style"
 )
 
-# --- 5. Launch App (Clean / No Theme) ---
+# --- 4.5. Background CSS ---
+
+BACKGROUND_URL = "https://images.template.net/269645/Cute-Christmas-Background-edit-online-1.jpg"
+
+custom_css = f"""
+.gradio-container {{
+  background-image: url('{BACKGROUND_URL}');
+  background-repeat: no-repeat;
+  background-size: cover;
+  background-position: center;
+}}
+"""
+
+# --- 5. Launch App (with background) ---
 demo = gr.ChatInterface(
     fn=chat_fn,
     title="Lab 2 – Fine-tuned GGUF model",
     description="Chat with the fine-tuned Llama model. Use the controls below to change the response style.",
     additional_inputs=[max_new_tokens_slider, style_radio],
     additional_inputs_accordion="Controls",
+    css=custom_css,  # <- here
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()