Create backend/models.py

backend/models.py

@@ -0,0 +1,66 @@
+from datetime import datetime
+from typing import List, Dict
+import json
+
+class Message:
+    def __init__(self, role: str, content: str, language: str = 'en'):
+        self.role = role
+        self.content = content
+        self.language = language
+        self.timestamp = datetime.now().isoformat()
+    
+    def to_dict(self):
+        return {
+            'role': self.role,
+            'content': self.content,
+            'language': self.language,
+            'timestamp': self.timestamp
+        }
+
+class ConversationManager:
+    def __init__(self):
+        self.conversations: Dict[str, List[Message]] = {}
+    
+    def add_message(self, conversation_id: str, role: str, content: str, language: str = 'en'):
+        """إضافة رسالة للمحادثة"""
+        if conversation_id not in self.conversations:
+            self.conversations[conversation_id] = []
+        
+        message = Message(role, content, language)
+        self.conversations[conversation_id].append(message)
+    
+    def get_messages(self, conversation_id: str) -> List[Dict]:
+        """الحصول على رسائل المحادثة"""
+        if conversation_id not in self.conversations:
+            return []
+        
+        return [msg.to_dict() for msg in self.conversations[conversation_id]]
+    
+    def get_context(self, conversation_id: str, limit: int = 10) -> List[Dict]:
+        """الحصول على السياق (آخر n رسالة)"""
+        messages = self.get_messages(conversation_id)
+        return messages[-limit:] if len(messages) > limit else messages
+    
+    def get_all_conversations(self) -> List[Dict]:
+        """الحصول على قائمة المحادثات"""
+        conversations = []
+        for conv_id, messages in self.conversations.items():
+            if messages:
+                conversations.append({
+                    'id': conv_id,
+                    'title': messages[0].content[:50] + '...' if len(messages[0].content) > 50 else messages[0].content,
+                    'last_message': messages[-1].timestamp,
+                    'message_count': len(messages)
+                })
+        return sorted(conversations, key=lambda x: x['last_message'], reverse=True)
+    
+    def delete_conversation(self, conversation_id: str):
+        """حذف محادثة"""
+        if conversation_id in self.conversations:
+            del self.conversations[conversation_id]
+    
+    def get_last_timestamp(self, conversation_id: str) -> str:
+        """الحصول على آخر وقت رسالة"""
+        if conversation_id in self.conversations and self.conversations[conversation_id]:
+            return self.conversations[conversation_id][-1].timestamp
+        return datetime.now().isoformat()