Update backend/app2.py

backend/app2.py

@@ -1,17 +1,30 @@
 import io
+import os
 from pathlib import Path
 
 import numpy as np
 import tensorflow as tf
-from flask import Flask, jsonify, request, send_file
+from flask import Flask, jsonify, request, send_file, send_from_directory
 from flask_cors import CORS
 from PIL import Image, ImageEnhance, ImageFilter
 
-app = Flask(__name__)
+# Prevent unnecessary GPU init on cloud
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
+# React build folder
+FRONTEND_DIST = Path(__file__).resolve().parent.parent / "frontend" / "dist"
+
+app = Flask(
+    __name__,
+    static_folder=str(FRONTEND_DIST),
+    static_url_path=""
+)
+
 CORS(app)
 
 BASE_DIR = Path(__file__).resolve().parent
-MODEL_PATH = BASE_DIR.parent / "model.h5"
+MODEL_PATH = BASE_DIR / "model.h5"
+
 TARGET_SHORT_SIDE = 2048
 MAX_LONG_SIDE = 4096
 GENERATOR_WORKING_LONG_SIDE = 768
@@ -45,7 +58,7 @@ def _load_gan_generator():
     global gan_generator, model_load_error
 
     if not MODEL_PATH.exists():
-        model_load_error = f"{MODEL_PATH.name} not found at project root"
+        model_load_error = f"{MODEL_PATH.name} not found in backend folder"
         return False
 
     try:
@@ -106,6 +119,7 @@ def improve_clarity(original_image, enhanced_image):
     pixels = np.asarray(image).astype("float32")
     brightness = float(np.mean(pixels))
     night_scene = brightness < 95
+
     if brightness < 95:
         image = ImageEnhance.Brightness(image).enhance(1.08)
     elif brightness < 135:
@@ -124,20 +138,40 @@ def improve_clarity(original_image, enhanced_image):
     width, height = image.size
     shortest_side = min(width, height)
     longest_side = max(width, height)
+
     scale = max(1.0, TARGET_SHORT_SIDE / shortest_side)
     scale = min(scale, MAX_LONG_SIDE / longest_side)
-    image = image.resize((round(width * scale), round(height * scale)), Image.Resampling.LANCZOS)
+
+    image = image.resize(
+        (round(width * scale), round(height * scale)),
+        Image.Resampling.LANCZOS
+    )
 
     image = ImageEnhance.Contrast(image).enhance(1.08)
     image = image.filter(ImageFilter.UnsharpMask(radius=0.8, percent=175, threshold=2))
     image = ImageEnhance.Sharpness(image).enhance(1.18)
+
     return image
 
 
+# ---------- FRONTEND ROUTES ----------
+
 @app.route("/")
-def home():
-    return "GAN enhancement backend is running"
+def serve_react():
+    return send_from_directory(app.static_folder, "index.html")
+
+
+@app.route("/<path:path>")
+def serve_static(path):
+    requested = FRONTEND_DIST / path
+
+    if requested.exists() and requested.is_file():
+        return send_from_directory(app.static_folder, path)
 
+    return send_from_directory(app.static_folder, "index.html")
+
+
+# ---------- BACKEND API ----------
 
 @app.route("/enhance", methods=["POST"])
 def enhance():
@@ -152,6 +186,7 @@ def enhance():
         image = Image.open(file.stream).convert("RGB")
 
         img = gan_generator.generate(image)
+
         buf = io.BytesIO()
         img.save(buf, format="PNG")
         buf.seek(0)
@@ -164,4 +199,4 @@ def enhance():
 
 
 if __name__ == "__main__":
-    app.run(debug=True, use_reloader=False)
+    app.run(host="0.0.0.0", port=7860, debug=False)