Update app.py

app.py

@@ -1,25 +1,66 @@
 import os
+import time
 from huggingface_hub import hf_hub_download
+from flask import Flask, jsonify
+import threading
+
+app = Flask(__name__)
 
 # تنظیم مسیرهای کش
 os.environ["HF_HOME"] = "/home/user/app/hf_cache"
 
-# مسیر CelebAMask-HQ
-celeb_path = "/home/user/app/huggingface_models/CelebAMask-HQ"
-face_parsing_path = os.path.join(celeb_path, "face_parsing")
-
-print("[Info] PYTHONPATH:", os.environ.get("PYTHONPATH"))
-print("[Info] CelebAMask-HQ path exists:", os.path.exists(celeb_path))
-print("[Info] face_parsing folder exists:", os.path.exists(face_parsing_path))
-
-# دانلود مدل از Hugging Face
-try:
-    model_path = hf_hub_download(
-        repo_id="public-data/CelebAMask-HQ-Face-Parsing",
-        filename="models/model.pth",
-        cache_dir="/home/user/app/hf_cache"
-    )
-    print("[Success] Model downloaded to:", model_path)
+def initialize_application():
+    """Initialize the application and download models"""
+    print("===== Application Startup at {} =====".format(time.strftime("%Y-%m-%d %H:%M:%S")))
+    
+    # مسیر CelebAMask-HQ
+    celeb_path = "/home/user/app/huggingface_models/CelebAMask-HQ"
+    face_parsing_path = os.path.join(celeb_path, "face_parsing")
+
+    print("[Info] PYTHONPATH:", os.environ.get("PYTHONPATH"))
+    print("[Info] CelebAMask-HQ path exists:", os.path.exists(celeb_path))
+    print("[Info] face_parsing folder exists:", os.path.exists(face_parsing_path))
+
+    # دانلود مدل از Hugging Face
+    try:
+        model_path = hf_hub_download(
+            repo_id="public-data/CelebAMask-HQ-Face-Parsing",
+            filename="models/model.pth",
+            cache_dir="/home/user/app/hf_cache"
+        )
+        print("[Success] Model downloaded to:", model_path)
+        return True
+    except Exception as e:
+        print("[Error] Failed to download model:", str(e))
+        return False
+
+@app.route('/')
+def home():
+    return jsonify({
+        "status": "running",
+        "message": "CelebAMask-HQ Face Parsing Application",
+        "timestamp": time.strftime("%Y-%m-%d %H:%M:%S")
+    })
+
+@app.route('/health')
+def health():
+    return jsonify({"status": "healthy"})
+
+@app.route('/info')
+def info():
+    celeb_path = "/home/user/app/huggingface_models/CelebAMask-HQ"
+    return jsonify({
+        "celebamask_hq_exists": os.path.exists(celeb_path),
+        "face_parsing_exists": os.path.exists(os.path.join(celeb_path, "face_parsing")),
+        "cache_directory": os.environ.get("HF_HOME"),
+        "pythonpath": os.environ.get("PYTHONPATH")
+    })
+
+if __name__ == '__main__':
+    # Initialize the application in a separate thread
+    init_thread = threading.Thread(target=initialize_application)
+    init_thread.start()
     
-except Exception as e:
-    print("[Error] Failed to download model:", str(e))
+    # Run Flask app
+    print("Starting Flask server on 0.0.0.0:7860...")
+    app.run(host='0.0.0.0', port=7860, debug=False)