Update app.py

app.py

@@ -1,70 +1,70 @@
-import time
-import os
-from flask import Flask, request, jsonify
-from flask_cors import CORS
-
-from model import *
-
-app = Flask(__name__)
-CORS(app)
-model = load_model()
-
-os.makedirs('uploads', exist_ok=True)
-
-@app.route('/', methods=['GET', 'POST'])
-def main_endpoint():
-    if request.method == 'GET':
-        return jsonify({
-            "result": "ok",
-            "service": "RPI Image Recognition API",
-            "endpoints": {
-                "GET /": "API status and documentation",
-                "POST /": "Image prediction (upload 'file')",
-                "GET /stitch": "Image stitching"
-            },
-            "model_loaded": model is not None
-        })
-    
-    elif request.method == 'POST':
-        if 'file' not in request.files:
-            return jsonify({"error": "No file uploaded"}), 400
-        
-        file = request.files['file']
-        if file.filename == '':
-            return jsonify({"error": "No file selected"}), 400
-        
-        filename = file.filename
-        file.save(os.path.join('uploads', filename))
-        
-        # filename format: "<timestamp>_<obstacle_id>_<signal>.jpeg"
-        constituents = file.filename.split("_")
-        obstacle_id = constituents[1] if len(constituents) > 1 else "unknown"
-        
-        ## Week 8 ## 
-        signal = constituents[2].strip(".jpg") if len(constituents) > 2 else "default"
-        image_id = predict_image(filename, model, signal)
-        
-        ## Week 9 ## 
-        # We don't need to pass in the signal anymore
-        #image_id = predict_image_week_9(filename,model)
-        
-        # Return the obstacle_id and image_id
-        result = {
-            "obstacle_id": obstacle_id,
-            "image_id": image_id
-        }
-        return jsonify(result)
-
-@app.route('/stitch', methods=['GET'])
-def stitch():
-    """
-    This is the main endpoint for the stitching command. Stitches the images using two different functions, in effect creating two stitches, just for redundancy purposes
-    """
-    img = stitch_image()
-    img.show()
-    img2 = stitch_image_own()
-    img2.show()
-    return jsonify({"result": "ok"})
-
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000, debug=True)
+import time
+import os
+from flask import Flask, request, jsonify
+from flask_cors import CORS
+
+from model import *
+
+app = Flask(__name__)
+CORS(app)
+model = load_model()
+
+os.makedirs('uploads', exist_ok=True)
+
+@app.route('/', methods=['GET', 'POST'])
+def main_endpoint():
+    if request.method == 'GET':
+        return jsonify({
+            "result": "ok",
+            "service": "RPI Image Recognition API",
+            "endpoints": {
+                "GET /": "API status and documentation",
+                "POST /": "Image prediction (upload 'file')",
+                "GET /stitch": "Image stitching"
+            },
+            "model_loaded": model is not None
+        })
+    
+    elif request.method == 'POST':
+        if 'file' not in request.files:
+            return jsonify({"error": "No file uploaded"}), 400
+        
+        file = request.files['file']
+        if file.filename == '':
+            return jsonify({"error": "No file selected"}), 400
+        
+        filename = file.filename
+        file.save(os.path.join('uploads', filename))
+        
+        # filename format: "<timestamp>_<obstacle_id>_<signal>.jpeg"
+        constituents = file.filename.split("_")
+        obstacle_id = constituents[1] if len(constituents) > 1 else "unknown"
+        
+        ## Week 8 ## 
+        signal = constituents[2].strip(".jpg") if len(constituents) > 2 else "default"
+        image_id = predict_image(filename, model, signal)
+        
+        ## Week 9 ## 
+        # We don't need to pass in the signal anymore
+        #image_id = predict_image_week_9(filename,model)
+        
+        # Return the obstacle_id and image_id
+        result = {
+            "obstacle_id": obstacle_id,
+            "image_id": image_id
+        }
+        return jsonify(result)
+
+@app.route('/stitch', methods=['GET'])
+def stitch():
+    """
+    This is the main endpoint for the stitching command. Stitches the images using two different functions, in effect creating two stitches, just for redundancy purposes
+    """
+    img = stitch_image()
+    img.show()
+    img2 = stitch_image_own()
+    img2.show()
+    return jsonify({"result": "ok"})
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)