Update app.py

app.py

@@ -22,25 +22,16 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 model.eval()
 
-# Hardcoded system prompt since we removed the style selector
+# Hardcoded system prompt
 SYSTEM_PROMPT = "You are a helpful, polite assistant. Give clear and structured explanations."
 
 def build_prompt(message, history):
-    """ 
-    Builds the prompt for the model. 
-    Parses history and adds the fixed system prompt.
-    """
     messages = []
-    
-    # Add the fixed system prompt
     messages.append({"role": "system", "content": SYSTEM_PROMPT})
     
-    # Process history
     for msg in history:
         role = msg.get("role")
         content = msg.get("content", "")
-        
-        # Handle if content is a list of blocks (Gradio 6 specific) or string
         if isinstance(content, list):
             texts = []
             for block in content:
@@ -55,7 +46,6 @@ def build_prompt(message, history):
         if text and role in ("user", "assistant", "system"):
             messages.append({"role": role, "content": text})
             
-    # Add current user message
     messages.append({"role": "user", "content": message})
     
     prompt = tokenizer.apply_chat_template(
@@ -64,13 +54,10 @@ def build_prompt(message, history):
     return prompt
 
 def chat_fn(message, history, max_new_tokens):
-    # build_prompt no longer needs 'style'
     prompt = build_prompt(message, history)
-    
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     
-    # Default hardcoded values for the removed sliders
-    # You can adjust these manually here if you want different behavior
+    # Default values for removed sliders
     temperature = 0.7 
     top_p = 0.9
     repetition_penalty = 1.1 
@@ -95,7 +82,6 @@ def chat_fn(message, history, max_new_tokens):
     return generated
 
 # --- Nature Theme Configuration ---
-# customized Soft theme with Earth tones
 nature_theme = gr.themes.Soft(
     primary_hue="green",
     secondary_hue="emerald",
@@ -103,11 +89,10 @@ nature_theme = gr.themes.Soft(
 ).set(
     body_background_fill="#f5f7f5",
     block_background_fill="rgba(255, 255, 255, 0.9)",
-    button_primary_background_fill="#2E7D32", # Forest Green
+    button_primary_background_fill="#2E7D32", 
     border_color_primary="#4CAF50",
 )
 
-# Custom CSS for a background image (Nature/Forest)
 custom_css = """
 .gradio-container {
     background: url('https://images.unsplash.com/photo-1441974231531-c6227db76b6e?q=80&w=2560&auto=format&fit=crop') no-repeat center center fixed; 
@@ -116,20 +101,22 @@ custom_css = """
 footer {visibility: hidden}
 """
 
-# Only the Token Slider remains
 max_new_tokens_slider = gr.Slider(
     minimum=16, maximum=512, value=128, step=8, label="Response Length (Tokens)",
 )
 
+# Instantiate ChatInterface WITHOUT theme/css arguments initially to avoid TypeError
 demo = gr.ChatInterface(
     fn=chat_fn,
     title="🌿 NatureChat Lab 2",
     description="Chat with the fine-tuned Llama model. Relax and enjoy the view.",
     additional_inputs=[max_new_tokens_slider],
-    additional_inputs_accordion="Settings",
-    theme=nature_theme,
-    css=custom_css
+    additional_inputs_accordion="Settings"
 )
 
+# Apply theme and CSS manually after instantiation
+demo.theme = nature_theme
+demo.css = custom_css
+
 if __name__ == "__main__":
     demo.launch()