Update app.py

app.py

@@ -244,42 +244,25 @@ def handle_query():
     try:
         cursor = conn.cursor()
         cursor.execute(sql_query)
-        
-        # إذا كان الاستعلام لا يعيد أي أعمدة (مثل SELECT COUNT(*))
-        if cursor.description is None:
-            return jsonify({"data": []}), 200
-            
         columns = [desc[0] for desc in cursor.description]
         rows = cursor.fetchall()
+        data = [dict(zip(columns, row)) for row in rows]
+
+        response_data = {
+            "data": data,
+        }
         
-        # تحويل النتائج إلى JSON قابل للتسلسل
-        result = []
-        for row in rows:
-            row_dict = {}
-            for i, col in enumerate(columns):
-                # معالجة أنواع البيانات غير القابلة للتسلسل مباشرةً
-                value = row[i]
-                if isinstance(value, (datetime.date, datetime.datetime)):
-                    value = value.isoformat()
-                elif isinstance(value, decimal.Decimal):
-                    value = float(value)
-                row_dict[col] = value
-            result.append(row_dict)
+        response_json = json.dumps(response_data, ensure_ascii=False)
         
-        # إرجاع الرد مع ضمان تنسيق JSON صحيح
-        response_data = {"data": result}
         return Response(
-            json.dumps(response_data, ensure_ascii=False),
+            response_json,
             status=200,
             mimetype='application/json; charset=utf-8'
         )
         
     except Exception as e:
         print(f"SQL execution error: {e}")
-        return jsonify({
-            "error": str(e),
-            "generated_sql": sql_query
-        }), 500
+        return jsonify({"error": str(e), "generated_sql": sql_query}), 500
     finally:
         if cursor:
             cursor.close()