Update app.py

app.py

@@ -6,14 +6,32 @@ generator = pipeline(
     model="deepseek-ai/deepseek-coder-1.3b-instruct"
 )
 
-def generate_code(prompt):
-    result = generator(prompt, max_length=200)
-    return result[0]["generated_text"]
-
-demo = gr.Interface(
-    fn=generate_code,
-    inputs="text",
-    outputs="text",
+SYSTEM_PROMPT = """
+You are an expert Python developer.
+
+Always:
+- write clean Python code
+- add comments when useful
+- avoid unnecessary explanations
+"""
+
+def chat(message, history):
+
+    prompt = SYSTEM_PROMPT + "\nUser: " + message + "\nAssistant:"
+
+    result = generator(
+        prompt,
+        max_new_tokens=200,
+        temperature=0.2
+    )
+
+    output = result[0]["generated_text"]
+
+    return output.split("Assistant:")[-1]
+
+
+demo = gr.ChatInterface(
+    fn=chat,
     title="Python Coding Assistant",
     description="Ask Python coding questions"
 )