Update app.py

app.py

@@ -3,57 +3,69 @@ from flask import Flask, send_file, request, jsonify
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 from functools import lru_cache
-import os
+from flask_cors import CORS  # إضافة دعم CORS
 
 app = Flask(__name__)
+CORS(app)  # تفعيل CORS للسماح بالاتصال
 
-@lru_cache(maxsize=1)
-def load_model():
-    """Load model and tokenizer with caching"""
-    tokenizer = AutoTokenizer.from_pretrained("amd/AMD-OLMo-1B")
-    model = AutoModelForCausalLM.from_pretrained(
-        "amd/AMD-OLMo-1B",
-        torch_dtype=torch.float16,
-        device_map="auto"
-    )
-    return model, tokenizer
+# تحميل النموذج - نقوم بتحميله مرة واحدة عند بدء التطبيق
+print("جاري تحميل النموذج...")
+tokenizer = AutoTokenizer.from_pretrained("amd/AMD-OLMo-1B")
+model = AutoModelForCausalLM.from_pretrained(
+    "amd/AMD-OLMo-1B",
+    torch_dtype=torch.float16,
+    device_map="auto"
+)
+print("تم تحميل النموذج بنجاح!")
 
-def generate_response(prompt, model, tokenizer):
+def generate_response(prompt):
     """Generate response from the model"""
-    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
-    with torch.no_grad():
-        outputs = model.generate(
-            **inputs,
-            max_length=200,
-            num_return_sequences=1,
-            temperature=0.7,
-            top_p=0.9,
-            repetition_penalty=1.2,
-            pad_token_id=tokenizer.eos_token_id
-        )
-        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        return response.replace(prompt, "").strip()
+    try:
+        inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
+        with torch.no_grad():
+            outputs = model.generate(
+                **inputs,
+                max_length=200,
+                num_return_sequences=1,
+                temperature=0.7,
+                top_p=0.9,
+                repetition_penalty=1.2,
+                pad_token_id=tokenizer.eos_token_id
+            )
+            response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            return response.replace(prompt, "").strip()
+    except Exception as e:
+        print(f"خطأ في توليد الرد: {str(e)}")
+        return "عذراً، حدث خطأ في معالجة رسالتك."
 
 @app.route('/')
 def home():
-    return send_file('index.html')
+    try:
+        return send_file('index.html')
+    except Exception as e:
+        print(f"خطأ في تحميل الصفحة: {str(e)}")
+        return "خطأ في تحميل الصفحة"
 
 @app.route('/message', methods=['POST'])
 def message():
     try:
         data = request.json
-        user_message = data.get('message', '')
+        if not data:
+            return jsonify({"response": "لم يتم استلام أي بيانات"}), 400
         
+        user_message = data.get('message', '')
         if not user_message:
-            return jsonify({"response": "عذراً، لم أفهم رسالتك"})
+            return jsonify({"response": "الرسالة فارغة"}), 400
         
-        model, tokenizer = load_model()
-        response = generate_response(user_message, model, tokenizer)
+        print(f"رسالة مستلمة: {user_message}")
+        response = generate_response(user_message)
+        print(f"الرد: {response}")
         
         return jsonify({"response": response})
     
     except Exception as e:
-        return jsonify({"response": f"عذراً، حدث خطأ: {str(e)}"})
+        print(f"خطأ في معالجة الرسالة: {str(e)}")
+        return jsonify({"response": "عذراً، حدث خطأ في معالجة رسالتك"}), 500
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(debug=True, port=5000)