Update agents/programmer.py

agents/programmer.py

@@ -1,34 +1,45 @@
 # agents/programmer.py
 
 from agents.base_agent import BaseAgent, ACPMessage
-import random
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
 class ProgrammerAgent(BaseAgent):
     def __init__(self):
         super().__init__(name="CodeBot", role="Programmer")
+        self.tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
+        self.model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+
+    def generate_code_reply(self, prompt: str) -> str:
+        inputs = self.tokenizer(prompt, return_tensors="pt", truncation=True)
+        outputs = self.model.generate(
+            inputs["input_ids"],
+            max_length=100,
+            do_sample=True,
+            temperature=0.7,
+            pad_token_id=self.tokenizer.eos_token_id
+        )
+        reply = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
+        return reply[len(prompt):].strip()
 
     def receive_message(self, message: ACPMessage) -> ACPMessage:
-        # Basic behavior for now — add AI model later
         if message.performative == "request":
-            response_content = random.choice([
-                "Sure, here's a simple Python function to reverse a list:\n```python\ndef reverse_list(lst): return lst[::-1]```",
-                "Okay! Here's a sorting example:\n```python\ndef sort_list(lst): return sorted(lst)```",
-                "Let me help with that. To calculate factorial:\n```python\ndef factorial(n): return 1 if n==0 else n*factorial(n-1)```"
-            ])
+            prompt = f"Write Python code to: {message.content.strip()}\n\n"
+            ai_reply = self.generate_code_reply(prompt)
             return self.create_message(
                 receiver=message.sender,
                 performative="inform",
-                content=response_content
+                content=f"Sure! Here's what I came up with:\n```python\n{ai_reply}\n```"
             )
         elif message.performative == "inform":
             return self.create_message(
                 receiver=message.sender,
                 performative="acknowledge",
-                content="Thanks for the update. Let me know if you need help."
+                content="Got it. Let me know if you need any code."
             )
         else:
             return self.create_message(
                 receiver=message.sender,
                 performative="refuse",
-                content="Sorry, I didn’t understand your request."
+                content="I'm not sure how to respond to that."
             )