app.py

# app.py
from flask import Flask, send_file, request, jsonify
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
from functools import lru_cache
from flask_cors import CORS  # إضافة دعم CORS

app = Flask(__name__)
CORS(app)  # تفعيل CORS للسماح بالاتصال

# تحميل النموذج - نقوم بتحميله مرة واحدة عند بدء التطبيق
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):
    """Generate response from the model"""
    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():
    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
        if not data:
            return jsonify({"response": "لم يتم استلام أي بيانات"}), 400
        
        user_message = data.get('message', '')
        if not user_message:
            return jsonify({"response": "الرسالة فارغة"}), 400
        
        print(f"رسالة مستلمة: {user_message}")
        response = generate_response(user_message)
        print(f"الرد: {response}")
        
        return jsonify({"response": response})
    
    except Exception as e:
        print(f"خطأ في معالجة الرسالة: {str(e)}")
        return jsonify({"response": "عذراً، حدث خطأ في معالجة رسالتك"}), 500

if __name__ == '__main__':
    app.run(debug=True, port=5000)