Upload 4 files

Dockerfile

@@ -2,18 +2,23 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
+# تثبيت الاعتمادات النظامية
 RUN apt-get update && apt-get install -y \
     build-essential \
     libpq-dev \
+    python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
 COPY . .
 
+# إنشاء مجلد الذاكرة المؤقتة
 RUN mkdir -p /app/model_cache && chmod -R 777 /app
 
+# تثبيت المتطلبات
 RUN pip install --upgrade pip setuptools wheel
 RUN pip install --no-cache-dir -r requirements.txt
 
+# متغيرات البيئة
 ENV FLASK_APP=app.py
 ENV FLASK_ENV=production
 ENV TRANSFORMERS_CACHE=/app/model_cache
@@ -23,4 +28,5 @@ ENV TOKENIZERS_PARALLELISM=false
 
 EXPOSE 7860
 
+# تشغيل التطبيق
 CMD ["gunicorn", "--workers", "1", "--threads", "2", "--bind", "0.0.0.0:7860", "--timeout", "120", "--preload", "app:app"]

app.py

@@ -1,12 +1,12 @@
 import os
 import re
 import logging
-from flask import Flask, request, jsonify, Response
+from flask import Flask, request, jsonify
 from flask_cors import CORS
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 import psycopg2
-import json
+from datetime import datetime
 
 app = Flask(__name__)
 CORS(app)
@@ -33,18 +33,26 @@ def initialize():
         logger.info("🚀 جاري تحميل النموذج...")
         tokenizer = AutoTokenizer.from_pretrained(
             MODEL_NAME,
-            cache_dir="model_cache",
-            local_files_only=False
+            cache_dir="model_cache"
         )
+        
+        # تحميل النموذج مع إعدادات الذاكرة المنخفضة
         model = AutoModelForSeq2SeqLM.from_pretrained(
             MODEL_NAME,
             cache_dir="model_cache",
             device_map="auto",
-            torch_dtype=torch.float16,
+            torch_dtype=torch.float32,  # استخدام float32 بدلاً من float16
             low_cpu_mem_usage=True
         )
+        
+        # التحقق من توفر GPU
+        if torch.cuda.is_available():
+            model.to('cuda')
+            logger.info("✅ تم تحميل النموذج على GPU")
+        else:
+            logger.info("✅ تم تحميل النموذج على CPU")
+            
         model.eval()
-        logger.info("✅ تم تحميل النموذج بنجاح")
     except Exception as e:
         logger.error(f"❌ فشل في تحميل النموذج: {str(e)}")
         raise
@@ -55,13 +63,86 @@ initialize()
 DB_SCHEMA = """
 CREATE TABLE public.profiles (
   id uuid NOT NULL,
+  updated_at timestamp with time zone,
+  username text UNIQUE CHECK (char_length(username) >= 3),
+  full_name text,
+  avatar_url text,
+  website text,
   cam_mac text UNIQUE,
-  ...
+  fcm_token text,
+  notification_enabled boolean DEFAULT true,
+  CONSTRAINT profiles_pkey PRIMARY KEY (id),
+  CONSTRAINT profiles_id_fkey FOREIGN KEY (id) REFERENCES auth.users(id)
+);
+
+CREATE TABLE public.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 (
+  id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
+  created_at timestamp with time zone NOT NULL DEFAULT now(),
+  place_id bigint,
+  user_cam_mac text,
+  CONSTRAINT user_place_pkey PRIMARY KEY (id),
+  CONSTRAINT user_place_place_id_fkey FOREIGN KEY (place_id) REFERENCES public.place(id),
+  CONSTRAINT user_place_user_cam_mac_fkey FOREIGN KEY (user_cam_mac) REFERENCES public.profiles(cam_mac)
 );
+
 CREATE TABLE public.data (
-  id bigint GENERATED ALWAYS AS IDENTITY,
+  id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
+  created_at timestamp without time zone,
+  caption text,
+  image_url text,
+  latitude double precision DEFAULT '36.1833854'::double precision,
+  longitude double precision DEFAULT '37.1309255'::double precision,
+  user_place_id bigint,
   cam_mac text,
-  ...
+  CONSTRAINT data_pkey PRIMARY KEY (id),
+  CONSTRAINT data_user_place_id_fkey FOREIGN KEY (user_place_id) REFERENCES public.user_place(id)
+);
+
+CREATE TABLE public.biodata (
+  id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
+  created_at timestamp with time zone NOT NULL DEFAULT now(),
+  mac_address text,
+  acceleration_x double precision,
+  acceleration_y double precision,
+  acceleration_z double precision,
+  gyro_x double precision,
+  gyro_y double precision,
+  gyro_z double precision,
+  temperature double precision,
+  CONSTRAINT biodata_pkey PRIMARY KEY (id),
+  CONSTRAINT biodata_mac_address_fkey FOREIGN KEY (mac_address) REFERENCES public.profiles(cam_mac)
+);
+
+CREATE TABLE public.notification (
+  id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
+  created_at timestamp without time zone NOT NULL DEFAULT now(),
+  user_cam_mac text,
+  title text,
+  message text,
+  is_read boolean,
+  acceleration_x double precision,
+  acceleration_y double precision,
+  acceleration_z double precision,
+  gyro_x double precision,
+  gyro_y double precision,
+  gyro_z double precision,
+  CONSTRAINT notification_pkey PRIMARY KEY (id),
+  CONSTRAINT notification_user_cam_mac_fkey FOREIGN KEY (user_cam_mac) REFERENCES public.profiles(cam_mac)
+);
+
+CREATE TABLE public.flag (
+  id bigint GENERATED ALWAYS AS IDENTITY NOT NULL,
+  flag smallint,
+  user_mac_address text,
+  CONSTRAINT flag_pkey PRIMARY KEY (id),
+  CONSTRAINT flag_user_mac_address_fkey FOREIGN KEY (user_mac_address) REFERENCES public.profiles(cam_mac)
 );
 """.strip()
 
@@ -119,7 +200,10 @@ def handle_query():
         SQL:
         """
 
-        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512).to(model.device)
+        # تحضير المدخلات مع التحقق من الجهاز
+        inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
+        if torch.cuda.is_available():
+            inputs = inputs.to('cuda')
         
         with torch.no_grad():
             outputs = model.generate(
@@ -189,6 +273,7 @@ def health_check():
     return jsonify({
         "status": "healthy",
         "model_loaded": model is not None,
+        "device": str(model.device) if model else "none",
         "db_connection": get_db_connection() is not None,
         "timestamp": datetime.now().isoformat()
     })