Upload auto_responder.py

auto_responder.py

@@ -0,0 +1,68 @@
+from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+import torch
+from typing import Dict
+
+class AutoResponder:
+    def __init__(self):
+        """
+        Initialize the AutoResponder with the Blenderbot model
+        """
+        self.model_name = "facebook/blenderbot-400M-distill"
+        self.tokenizer = BlenderbotTokenizer.from_pretrained(self.model_name)
+        self.model = BlenderbotForConditionalGeneration.from_pretrained(self.model_name)
+        self.conversation_history: Dict[str, list] = {}
+
+    def generate_response(self, message: str, sender_id: str) -> str:
+        """
+        Generate an AI response to the incoming message using Blenderbot
+        """
+        if sender_id not in self.conversation_history:
+            self.conversation_history[sender_id] = []
+        
+        # Add the user message to conversation history
+        self.conversation_history[sender_id].append(message)
+
+        try:
+            # Encode the input text
+            inputs = self.tokenizer(message, return_tensors="pt", truncation=True, max_length=512)
+            
+            # Generate response
+            reply_ids = self.model.generate(
+                inputs["input_ids"],
+                max_length=128,
+                num_beams=4,
+                no_repeat_ngram_size=3,
+                temperature=0.7
+            )
+            
+            # Decode the response
+            response = self.tokenizer.decode(reply_ids[0], skip_special_tokens=True)
+            
+            # Add AI response to conversation history
+            self.conversation_history[sender_id].append(response)
+            
+            return response
+
+        except Exception as e:
+            return f"عذراً، حدث خطأ: {str(e)}"
+
+    def clear_history(self, sender_id: str):
+        """
+        Clear conversation history for a specific sender
+        """
+        if sender_id in self.conversation_history:
+            self.conversation_history[sender_id] = []
+
+# Example usage
+if __name__ == "__main__":
+    # Create an instance of AutoResponder
+    responder = AutoResponder()
+    
+    # Example message
+    example_message = "مرحبا، كيف حالك؟"
+    sender = "user123"
+    
+    # Generate and print response
+    response = responder.generate_response(example_message, sender)
+    print(f"الرسالة: {example_message}")
+    print(f"الرد: {response}")