Update app.py

app.py

@@ -198,8 +198,6 @@ def calculate_and_display_angles(image_path, return_angles=False):
 
     # Get pose landmarks
     results = pose.process(image_rgb)
-
-    return 'hi'
     
     if results.pose_landmarks:
         ang , locs = calculate_angles_landmarks(results.pose_landmarks)
@@ -216,8 +214,8 @@ def calculate_and_display_angles(image_path, return_angles=False):
         # Display the angle values at corresponding locations
         def overlay_angle(angle, location, color=(0, 255, 0)):
             # Coordinates are normalized, scale them to the image size
-            x = int(location[0] )#* image.shape[1])
-            y = int(location[1] )#* image.shape[0])
+            x = int(location[0] * image.shape[1])
+            y = int(location[1] * image.shape[0])
             
             cv2.putText(image, f'{angle:.2f}', (x, y), cv2.FONT_HERSHEY_SIMPLEX, 0.3, color, 1)
 
@@ -263,6 +261,7 @@ app = Flask(__name__)
 def hello():
     return "This is my python code Adew"
     
+from flask import Response
 ########################################
 @app.route('/1', methods=['POST'])
 def upload_file():
@@ -275,15 +274,21 @@ def upload_file():
     # 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)
+            # file.seek(0)
+            # mime_type, _ = mimetypes.guess_type(file.filename)
             
             f1 = calculate_and_display_angles(file)
-            
-            return send_file(
-                BytesIO(file.read()), mimetype=mime_type,
-                as_attachment=False, download_name=file.filename
-            )
+
+            success, encoded_image = cv2.imencode('.jpg', f1)
+            if not success:
+                return "Image encoding failed", 500
+
+            return Response(encoded_image.tobytes(), mimetype='image/jpeg')
+
+            # 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