Update app.py

app.py

@@ -1,10 +1,41 @@
-from flask import Flask
+from flask import Flask, request, send_file
+from io import BytesIO
+import mimetypes
 
-app = Flask(__name__)
 
+#########################################
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+app = Flask(__name__)
+###########################################
+#############################################
 @app.route('/')
 def hello():
-    return "Hello from Hugging Face Docker Space!"
-
+    return "This is my python code Adew"
+########################################
+@app.route('/1', methods=['POST'])
+def upload_file():
+    if 'image' not in request.files:
+        return {"message": "No file part"}, 400
+    file = request.files['image']
+    if file.filename == '':
+        return {"message": "No selected file"}, 400
+    
+    # send file with angles on image or just angles
+    if file and allowed_file(file.filename):
+        try:
+            file.seek(0)
+            mime_type, _ = mimetypes.guess_type(file.filename)
+            
+            return send_file(
+                BytesIO(file.read()), mimetype=mime_type,
+                as_attachment=False, download_name=file.filename
+            )
+        
+        except Exception as e:
+            return {"message": f"Error processing file: {str(e)}"}, 500
+    return {"message": "File not allowed"}, 400
+#####################################
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)