Update app.py

app.py

@@ -40,7 +40,7 @@ def initialize():
             MODEL_NAME,
             cache_dir="model_cache",
             device_map="auto",
-            torch_dtype=torch.float16,  # لتوفير الذاكرة
+            torch_dtype=torch.float32,  # تغيير من float16 إلى float32 للإصلاح
             low_cpu_mem_usage=True
         )
         
@@ -48,7 +48,8 @@ def initialize():
             model.to('cuda')
             logger.info("✅ تم تحميل النموذج على GPU")
         else:
-            logger.info("✅ تم تحميل النموذج على CPU")
+            model.float()  # تأكد من تحويل النموذج إلى float32 على CPU
+            logger.info("✅ تم تحميل النموذج على CPU (باستخدام float32)")
             
         model.eval()
     except Exception as e:
@@ -210,40 +211,33 @@ def handle_query():
         data = request.get_json()
         logger.info(f"📩 بيانات الطلب: {data}")
         
-        # التحقق من وجود البيانات المطلوبة
         if not data or 'text' not in data or 'cam_mac' not in data:
-            logger.error("❌ بيانات ناقصة في الطلب")
             return jsonify({"error": "يرجى إرسال 'text' و 'cam_mac'"}), 400
 
-        # التحقق من صحة cam_mac
         if not validate_cam_mac(data['cam_mac']):
-            logger.error(f"❌ cam_mac غير صالح: {data['cam_mac']}")
             return jsonify({"error": "عنوان MAC غير صالح"}), 403
 
-        # إنشاء الـ prompt مع تعليمات صارمة
         prompt = f"""
 ### التعليمات الصارمة:
 1. قم بتحويل السؤال إلى استعلام SELECT فقط لـ PostgreSQL.
 2. يجب أن يتضمن الشرط: WHERE cam_mac = '{data['cam_mac']}'.
-3. ممنوع تمامًا استخدام أي أوامر غير SELECT (مثل INSERT, UPDATE, DELETE, CREATE, DROP).
+3. ممنوع تمامًا استخدام أي أوامر غير SELECT.
 
 ### هيكل قاعدة البيانات:
 {DB_SCHEMA}
 
-### أمثلة صحيحة:
-- السؤال: "ما عدد زياراتي للمكتب؟"
-  SQL: SELECT COUNT(*) FROM data WHERE cam_mac = '{data['cam_mac']}';
-
-- السؤال: "ما هي آخر 10 زيارات؟"
-  SQL: SELECT * FROM data WHERE cam_mac = '{data['cam_mac']}' ORDER BY created_at DESC LIMIT 10;
-
-### السؤال الحالي:
+### السؤال:
 {data['text']}
 
 ### استعلام SQL (SELECT فقط):
 """
-        # توليد الاستعلام
+
         inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
+        
+        # تحويل المدخلات إلى float32 إذا كنا على CPU
+        if not torch.cuda.is_available():
+            inputs = inputs.to(torch.float32)
+        
         if torch.cuda.is_available():
             inputs = inputs.to('cuda')
         
@@ -256,26 +250,43 @@ def handle_query():
             )
             
         sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        logger.info(f"⚡ الاستعلام المولد (قبل التنظيف): {sql}")
-        
-        # تنظيف الاستعلام
         sql = clean_sql(sql)
-        logger.info(f"🔧 الاستعلام بعد التنظيف: {sql}")
         
-        # التحقق من أن الاستعلام آمن
         if not is_safe_sql(sql):
-            logger.error(f"❌ استعلام غير آمن: {sql}")
-            return jsonify({
-                "error": "تم توليد استعلام غير آمن",
-                "details": "الاستعلام يجب أن يكون SELECT فقط ويجب أن يتضمن شرط WHERE لـ cam_mac",
-                "sql": sql
-            }), 400
-        
-        # تنفيذ الاستعلام
+            return jsonify({"error": "تم توليد استعلام غير آمن"}), 400
+
         conn = get_db_connection()
         if not conn:
             return jsonify({"error": "فشل الاتصال بقاعدة البيانات"}), 500
 
+        try:
+            cursor = conn.cursor()
+            cursor.execute(sql)
+            
+            columns = [desc[0] for desc in cursor.description] if cursor.description else []
+            rows = cursor.fetchall()
+            
+            return jsonify({
+                "data": [dict(zip(columns, row)) for row in rows],
+                "sql": sql,
+                "timestamp": datetime.now().isoformat()
+            })
+            
+        except Exception as e:
+            return jsonify({
+                "error": "خطأ في تنفيذ الاستعلام",
+                "sql": sql,
+                "details": str(e)
+            }), 500
+            
+        finally:
+            if conn:
+                conn.close()
+                
+    except Exception as e:
+        logger.error(f"❌ خطأ غير متوقع: {str(e)}", exc_info=True)
+        return jsonify({"error": "فشل في معالجة الطلب"}), 500
+
         try:
             cursor = conn.cursor()
             cursor.execute(sql)