Update app.py

app.py

@@ -26,8 +26,8 @@ def chat_function(user_message, history):
     response = generator(
         prompt,
         max_new_tokens=250,      # Give it enough space to write a full function
-        temperature=0.4,         # Low temperature = Focused logic (No creative babbling)
-        repetition_penalty=1.2,  # STRICT penalty. If it repeats "Hello", it gets banned.
+        temperature=0.4,         # Low temperature = Focused logic
+        repetition_penalty=1.2,  # STRICT penalty. Bans "Hello World" loops.
         do_sample=True,
         pad_token_id=tokenizer.eos_token_id
     )
@@ -36,27 +36,24 @@ def chat_function(user_message, history):
     full_text = response[0]['generated_text']
     
     # Isolate just the new code
-    # We split by "### Response:" to get the answer
     answer_part = full_text.split("### Response:")[-1].strip()
     
     # Ensure the formatting looks nice in the chat
-    # We stripped the ```python start, so we add it back for the UI
     if not answer_part.startswith("```"):
         answer_part = "```python\n" + answer_part
         
     return answer_part
 
-# 6. Launch the Chat Interface
+# 6. Launch the Chat Interface (Fixed: No theme argument)
 demo = gr.ChatInterface(
     fn=chat_function,
-    title="🐍 AI Python Assistant",
+    title="🐍 AI Python Graduate (Step 1000)",
     description="I am trained! Ask me to 'Write a function to...' or 'Create a loop...'",
     examples=[
         "Write a python function to add two numbers",
         "Create a loop from 1 to 10",
         "Write a script to calculate the area of a circle"
-    ],
-    theme="soft" 
+    ]
 )
 
 demo.launch()