Update app.py

app.py

@@ -13,36 +13,60 @@ MODEL_REPO = "Jeppcode/ScalableLab2"
 GGUF_FILENAME = "model-q4_k_m.gguf" 
 
 print(f"Downloading GGUF model {MODEL_REPO}/{GGUF_FILENAME} ...")
-model_path = hf_hub_download(
-    repo_id=MODEL_REPO,
-    filename=GGUF_FILENAME,
-)
-
-print("Initializing llama.cpp LLM ...")
-llm = Llama(
-    model_path=model_path,
-    n_ctx=2048,          
-    n_threads=2,         
-    n_batch=64,          
-    use_mmap=True,
-    use_mlock=False,
-)
+try:
+    model_path = hf_hub_download(
+        repo_id=MODEL_REPO,
+        filename=GGUF_FILENAME,
+    )
+except Exception as e:
+    print(f"Error downloading model: {e}")
+    # Fallback for debugging if download fails
+    model_path = "" 
+
+if model_path:
+    print("Initializing llama.cpp LLM ...")
+    llm = Llama(
+        model_path=model_path,
+        n_ctx=2048,          
+        n_threads=2,         
+        n_batch=64,          
+        use_mmap=True,
+        use_mlock=False,
+    )
+else:
+    print("WARNING: Model path invalid. App will start but chat will fail.")
 
 # --- 3. Image Handling (The Fix) ---
-def encode_image(image_path):
+def get_background_image_css():
+    """
+    Tries to load local image. If missing, uses a URL fallback.
+    Returns the CSS string for the background.
     """
-    Reads the local image and converts it to a base64 string.
-    This ensures it loads in the CSS without 404 errors.
+    local_img = "Cute-Christmas-Background-edit-online-1.jpg"
+    
+    # 1. Try Local Base64
+    if os.path.exists(local_img):
+        print(f"Found local image: {local_img}. Encoding...")
+        with open(local_img, "rb") as f:
+            encoded = base64.b64encode(f.read()).decode('utf-8')
+        img_url = f"data:image/jpeg;base64,{encoded}"
+    # 2. Fallback to URL if local fails
+    else:
+        print(f"⚠️ Could not find {local_img}. Using fallback URL.")
+        img_url = "https://images.unsplash.com/photo-1544976735-a10c71a39644?q=80&w=2560&auto=format&fit=crop"
+
+    return f"""
+    body, .gradio-container, .gradio-app {{
+        background-image: url('{img_url}') !important;
+        background-size: cover !important;
+        background-position: center center !important;
+        background-attachment: fixed !important;
+        background-repeat: no-repeat !important;
+        background-color: #0b121e !important; /* Fallback color */
+    }}
     """
-    if not os.path.exists(image_path):
-        return "" # Fail silently if file missing
-    with open(image_path, "rb") as image_file:
-        encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
-    return f"data:image/jpeg;base64,{encoded_string}"
 
-# Exact filename from your screenshot
-IMG_FILENAME = "Cute-Christmas-Background-edit-online-1.jpg"
-bg_image_data = encode_image(IMG_FILENAME)
+bg_css_rule = get_background_image_css()
 
 # --- 4. Prompts & Logic ---
 STYLE_SYSTEM_PROMPTS = {
@@ -76,11 +100,16 @@ def build_prompt(message, history, style):
     return "".join(prompt_parts)
 
 def chat_fn(message, history, max_new_tokens, style):
+    # Safe defaults
     temperature = 0.7
     top_p = 0.9
     repetition_penalty = 1.1
     prompt = build_prompt(message, history, style)
     
+    # Check if LLM loaded correctly
+    if 'llm' not in globals():
+        return "Error: Model not loaded."
+
     output = llm(
         prompt,
         max_tokens=int(max_new_tokens),
@@ -92,66 +121,60 @@ def chat_fn(message, history, max_new_tokens, style):
     return output["choices"][0]["text"].strip()
 
 # --- 5. Theme & CSS ---
+
+# A robust theme base
 winter_theme = gr.themes.Soft(
     primary_hue="blue", 
     neutral_hue="slate",
 ).set(
     body_background_fill="transparent",
-    block_background_fill="rgba(10, 20, 40, 0.7)", 
+    block_background_fill="rgba(15, 23, 42, 0.7)", 
     border_color_primary="#FFFFFF", 
     button_primary_background_fill="#FFFFFF", 
     button_primary_text_color="#000000",
 )
 
-# We inject the base64 image data directly into the URL()
-custom_css = f"""
-/* Apply Background to the main app container */
-.gradio-container {{
-    background-image: url('{bg_image_data}') !important;
-    background-size: cover !important;
-    background-position: center center !important;
-    background-attachment: fixed !important;
-    background-repeat: no-repeat !important;
-}}
-
-/* Ensure the main content area is transparent so bg shows */
-.gradio-container > .main {{
+# Combined CSS
+custom_css = bg_css_rule + """
+/* Make the App Wrapper Transparent */
+.gradio-container {
     background: transparent !important;
-}}
+}
 
-/* Dark Glassy Look for Chat and Settings */
-.group, .form, .bubble-wrap {{
-    background: rgba(15, 23, 42, 0.75) !important; 
+/* Chat & Settings Blocks - Frosty Glass with White Borders */
+.group, .form, .bubble-wrap, .chatbot {
+    background-color: rgba(15, 23, 42, 0.8) !important; 
     border: 2px solid #FFFFFF !important; 
     border-radius: 12px !important;
-    backdrop-filter: blur(4px); 
+    backdrop-filter: blur(5px); 
     box-shadow: 0 4px 15px rgba(0,0,0,0.5);
-}}
+}
 
-/* Chat Bubbles */
-.user-message {{
+/* Specific Chat Bubble Coloring */
+.user-message {
     background-color: rgba(255, 255, 255, 0.2) !important; 
     border: 1px solid #FFFFFF !important;
     color: #FFFFFF !important;
-}}
-.bot-message {{
+}
+.bot-message {
     background-color: rgba(0, 0, 0, 0.6) !important; 
     border: 1px solid #A0A0A0 !important;
     color: #E0E0E0 !important;
-}}
+}
 
-/* Inputs and Text */
-textarea, input {{
+/* Text Colors */
+textarea, input {
     background-color: rgba(0, 0, 0, 0.5) !important;
     border: 1px solid #FFFFFF !important;
     color: white !important;
-}}
-label, span, p, .prose {{
+}
+label, span, p, .prose {
     color: #FFFFFF !important;
     text-shadow: 1px 1px 2px black; 
-}}
+}
 
-footer {{visibility: hidden}}
+/* Hide Footer */
+footer {visibility: hidden}
 """
 
 max_new_tokens_slider = gr.Slider(
@@ -170,12 +193,9 @@ demo = gr.ChatInterface(
     description="Stay frosty. Chat with the fine-tuned model.",
     additional_inputs=[max_new_tokens_slider, style_radio],
     additional_inputs_accordion="Settings",
+    theme=winter_theme,
+    css=custom_css
 )
 
-# Apply configuration
-demo.theme = winter_theme
-demo.css = custom_css
-
 if __name__ == "__main__":
-    # allowed_paths=["."] is a backup, but base64 should solve it.
     demo.launch(allowed_paths=["."])