Update app.py

app.py

@@ -60,7 +60,7 @@ initialize()
 
 # سكيما قاعدة البيانات (نفسها كما في الكود الأصلي)
 DB_SCHEMA = """
-CREATE TABLE public.profiles (
+profiles (
   id uuid NOT NULL,
   updated_at timestamp with time zone,
   username text UNIQUE CHECK (char_length(username) >= 3),
@@ -74,14 +74,14 @@ CREATE TABLE public.profiles (
   CONSTRAINT profiles_id_fkey FOREIGN KEY (id) REFERENCES auth.users(id)
 );
 
-CREATE TABLE public.place (
+place (
   id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
   created_at timestamp with time zone DEFAULT (now() AT TIME ZONE 'utc'::text),
   name text,
   CONSTRAINT place_pkey PRIMARY KEY (id)
 );
 
-CREATE TABLE public.user_place (
+user_place (
   id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
   created_at timestamp with time zone NOT NULL DEFAULT now(),
   place_id bigint,
@@ -91,7 +91,7 @@ CREATE TABLE public.user_place (
   CONSTRAINT user_place_user_cam_mac_fkey FOREIGN KEY (user_cam_mac) REFERENCES public.profiles(cam_mac)
 );
 
-CREATE TABLE public.data (
+data (
   id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
   created_at timestamp without time zone,
   caption text,
@@ -104,7 +104,7 @@ CREATE TABLE public.data (
   CONSTRAINT data_user_place_id_fkey FOREIGN KEY (user_place_id) REFERENCES public.user_place(id)
 );
 
-CREATE TABLE public.biodata (
+biodata (
   id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
   created_at timestamp with time zone NOT NULL DEFAULT now(),
   mac_address text,
@@ -119,7 +119,7 @@ CREATE TABLE public.biodata (
   CONSTRAINT biodata_mac_address_fkey FOREIGN KEY (mac_address) REFERENCES public.profiles(cam_mac)
 );
 
-CREATE TABLE public.notification (
+notification (
   id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
   created_at timestamp without time zone NOT NULL DEFAULT now(),
   user_cam_mac text,
@@ -136,7 +136,7 @@ CREATE TABLE public.notification (
   CONSTRAINT notification_user_cam_mac_fkey FOREIGN KEY (user_cam_mac) REFERENCES public.profiles(cam_mac)
 );
 
-CREATE TABLE public.flag (
+flag (
   id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
   flag smallint,
   user_mac_address text,
@@ -192,17 +192,27 @@ def is_safe_sql(sql):
     
     return True
 
-def clean_sql(sql):
+def clean_sql(sql, cam_mac):
     """تنظيف استعلام SQL لضمان أنه SELECT فقط"""
+    # إزالة أي أوامر غير SELECT
+    sql = re.sub(r'CREATE\s+TABLE', '', sql, flags=re.IGNORECASE)
+    sql = re.sub(r'INSERT\s+INTO', '', sql, flags=re.IGNORECASE)
+    sql = re.sub(r'UPDATE\s+', '', sql, flags=re.IGNORECASE)
+    sql = re.sub(r'DELETE\s+FROM', '', sql, flags=re.IGNORECASE)
+    
     # إزالة أي شيء قبل SELECT
     sql = re.sub(r'^[^S]*(SELECT)', 'SELECT', sql, flags=re.IGNORECASE)
     
-    # أخذ أول عبارة SQL فقط (تجاهل أي شيء بعد ;)
+    # أخذ أول عبارة SQL فقط
     sql = sql.split(';')[0] + ';'
     
     # إزالة المسافات الزائدة
     sql = ' '.join(sql.split()).strip()
     
+    # إذا لم يعد يحتوي على SELECT بعد التنظيف
+    if not sql.upper().startswith('SELECT'):
+        return f"SELECT id, created_at FROM data WHERE cam_mac = '{cam_mac}' ORDER BY created_at DESC LIMIT 1;"
+    
     return sql
 
 @app.route('/api/query', methods=['POST'])
@@ -222,39 +232,45 @@ def handle_query():
 1. قم بتحويل السؤال إلى استعلام SELECT فقط لـ PostgreSQL.
 2. يجب أن يتضمن الشرط: WHERE cam_mac = '{data['cam_mac']}'.
 3. ممنوع تمامًا استخدام أي أوامر غير SELECT.
+4. استخدم أسماء الجداول الموجودة فقط: profiles, place, user_place, data, biodata, notification, flag.
 
 ### هيكل قاعدة البيانات:
 {DB_SCHEMA}
 
+### أمثلة صحيحة:
+- السؤال: "ما عدد زياراتي للمكتب؟"
+  SQL: SELECT COUNT(*) FROM data WHERE cam_mac = '{data['cam_mac']}';
+
+- السؤال: "ما هي آخر زيارة؟"
+  SQL: SELECT id, created_at FROM data WHERE cam_mac = '{data['cam_mac']}' ORDER BY created_at DESC LIMIT 1;
+
 ### السؤال:
 {data['text']}
 
 ### استعلام SQL (SELECT فقط):
 """
 
-        # تحضير المدخلات بدون تحويل نوع البيانات مباشرة
-        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
-        
-        # نقل المدخلات إلى الجهاز المناسب (CPU/GPU)
-        device = 'cuda' if torch.cuda.is_available() else 'cpu'
-        inputs = {k: v.to(device) for k, v in inputs.items()}
+        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512).to('cpu')
         
         with torch.no_grad():
             outputs = model.generate(
                 **inputs,
                 max_length=256,
                 num_beams=4,
-                early_stopping=True
+                early_stopping=True,
+                temperature=0.7
             )
             
         sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        logger.info(f"⚡ الاستعلام المولد: {sql}")
+        logger.info(f"⚡ الاستعلام المولد (قبل التنظيف): {sql}")
         
-        sql = clean_sql(sql)
+        sql = clean_sql(sql, data['cam_mac'])
+        logger.info(f"🔧 الاستعلام بعد التنظيف: {sql}")
         
         if not is_safe_sql(sql):
-            return jsonify({"error": "تم توليد استعلام غير آمن"}), 400
-
+            logger.warning(f"استعلام غير آمن بعد التنظيف، استخدام استعلام افتراضي")
+            sql = f"SELECT id, created_at FROM data WHERE cam_mac = '{data['cam_mac']}' ORDER BY created_at DESC LIMIT 10;"
+        
         conn = get_db_connection()
         if not conn:
             return jsonify({"error": "فشل الاتصال بقاعدة البيانات"}), 500
@@ -287,45 +303,6 @@ def handle_query():
         logger.error(f"❌ خطأ غير متوقع: {str(e)}", exc_info=True)
         return jsonify({"error": "فشل في معالجة الطلب"}), 500
 
-        try:
-            cursor = conn.cursor()
-            cursor.execute(sql)
-            
-            if cursor.description:
-                columns = [desc[0] for desc in cursor.description]
-                rows = cursor.fetchall()
-                data = [dict(zip(columns, row)) for row in rows]
-                
-                response = {
-                    "data": data,
-                    "sql": sql,
-                    "timestamp": datetime.now().isoformat()
-                }
-            else:
-                conn.commit()
-                response = {
-                    "message": "تم تنفيذ الاستعلام بنجاح (لا توجد بيانات للإرجاع)",
-                    "sql": sql
-                }
-                
-            return jsonify(response)
-            
-        except Exception as e:
-            logger.error(f"❌ خطأ في تنفيذ SQL: {e}\nالاستعلام: {sql}")
-            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
-
 @app.route('/health')
 def health_check():
     return jsonify({