Spaces:
Paused
Paused
giorgio-caparvi
commited on
Commit
·
98777c8
1
Parent(s):
194aaac
adding debug error
Browse files- api/app.py +21 -13
api/app.py
CHANGED
@@ -19,34 +19,35 @@ def index():
|
|
19 |
@app.route('/generate-design', methods=['GET','POST'])
|
20 |
def generate_design():
|
21 |
try:
|
|
|
22 |
|
23 |
# Getting json
|
24 |
json_data_from_req = request.get_json()
|
25 |
if not json_data_from_req:
|
|
|
26 |
return "Invalid or missing JSON data", 400
|
27 |
-
print(json_data_from_req)
|
28 |
-
|
29 |
|
30 |
# Getting Image
|
31 |
-
|
32 |
if 'image' not in request.files:
|
|
|
33 |
return "No image file in request", 400
|
34 |
|
35 |
image_file = request.files['image']
|
36 |
try:
|
37 |
image = Image.open(image_file)
|
|
|
38 |
except Exception as e:
|
|
|
39 |
return f"Failed to open the image: {str(e)}", 400
|
40 |
-
|
41 |
-
#
|
42 |
img_sketch_buffer = io.BytesIO()
|
43 |
-
# Save the image to the buffer in JPEG format
|
44 |
image.save(img_sketch_buffer, format='JPEG')
|
45 |
-
# Rewind the buffer's position to the beginning
|
46 |
img_sketch_buffer.seek(0)
|
|
|
47 |
|
48 |
-
|
49 |
-
# Creiamo una lista di argomenti come quelli che passeresti via CLI
|
50 |
sys.argv = [
|
51 |
'eval.py',
|
52 |
'--dataset_path', '/api/model/assets/data/vitonhd',
|
@@ -58,31 +59,38 @@ def generate_design():
|
|
58 |
'--start_cond_rate', '0.0',
|
59 |
'--test_order', 'paired'
|
60 |
]
|
|
|
61 |
|
62 |
-
|
63 |
-
# Cattura tutti gli errori generati dalla chiamata a eval.main
|
64 |
try:
|
65 |
final_image = eval.main(img_sketch_buffer, json_data_from_req)
|
|
|
66 |
except AttributeError as e:
|
|
|
67 |
return f"AttributeError: {str(e)}", 500
|
68 |
except TypeError as e:
|
|
|
69 |
return f"TypeError: {str(e)}", 500
|
70 |
except ValueError as e:
|
|
|
71 |
return f"ValueError: {str(e)}", 500
|
72 |
except IOError as e:
|
|
|
73 |
return f"IOError: {str(e)}", 500
|
74 |
except Exception as e:
|
|
|
75 |
return f"Unexpected error: {str(e)}", 500
|
76 |
|
77 |
-
#
|
78 |
img_io = io.BytesIO()
|
79 |
final_image.save(img_io, 'JPEG')
|
80 |
img_io.seek(0)
|
|
|
81 |
|
82 |
-
# Return the image as a file download
|
83 |
return send_file(img_io, mimetype='image/jpeg', as_attachment=True, download_name='inmemory_image.jpg')
|
84 |
|
85 |
except Exception as e:
|
|
|
86 |
return str(e), 500
|
87 |
|
88 |
if __name__ == '__main__':
|
|
|
19 |
@app.route('/generate-design', methods=['GET','POST'])
|
20 |
def generate_design():
|
21 |
try:
|
22 |
+
print("### Inizio generate_design ###") # Stampa per il debug
|
23 |
|
24 |
# Getting json
|
25 |
json_data_from_req = request.get_json()
|
26 |
if not json_data_from_req:
|
27 |
+
print("### JSON non ricevuto ###") # Stampa per il debug
|
28 |
return "Invalid or missing JSON data", 400
|
29 |
+
print("### JSON ricevuto:", json_data_from_req) # Stampa per il debug
|
|
|
30 |
|
31 |
# Getting Image
|
|
|
32 |
if 'image' not in request.files:
|
33 |
+
print("### Nessuna immagine ricevuta ###") # Stampa per il debug
|
34 |
return "No image file in request", 400
|
35 |
|
36 |
image_file = request.files['image']
|
37 |
try:
|
38 |
image = Image.open(image_file)
|
39 |
+
print("### Immagine aperta con successo ###") # Stampa per il debug
|
40 |
except Exception as e:
|
41 |
+
print(f"### Errore nell'apertura dell'immagine: {str(e)} ###") # Stampa per il debug
|
42 |
return f"Failed to open the image: {str(e)}", 400
|
43 |
+
|
44 |
+
# Creazione buffer di immagine
|
45 |
img_sketch_buffer = io.BytesIO()
|
|
|
46 |
image.save(img_sketch_buffer, format='JPEG')
|
|
|
47 |
img_sketch_buffer.seek(0)
|
48 |
+
print("### Immagine salvata nel buffer ###") # Stampa per il debug
|
49 |
|
50 |
+
# Argomenti per eval.main
|
|
|
51 |
sys.argv = [
|
52 |
'eval.py',
|
53 |
'--dataset_path', '/api/model/assets/data/vitonhd',
|
|
|
59 |
'--start_cond_rate', '0.0',
|
60 |
'--test_order', 'paired'
|
61 |
]
|
62 |
+
print("### Esecuzione eval.main ###") # Stampa per il debug
|
63 |
|
64 |
+
# Esecuzione del modello
|
|
|
65 |
try:
|
66 |
final_image = eval.main(img_sketch_buffer, json_data_from_req)
|
67 |
+
print("### eval.main eseguito con successo ###") # Stampa per il debug
|
68 |
except AttributeError as e:
|
69 |
+
print(f"### AttributeError: {str(e)} ###") # Stampa per il debug
|
70 |
return f"AttributeError: {str(e)}", 500
|
71 |
except TypeError as e:
|
72 |
+
print(f"### TypeError: {str(e)} ###") # Stampa per il debug
|
73 |
return f"TypeError: {str(e)}", 500
|
74 |
except ValueError as e:
|
75 |
+
print(f"### ValueError: {str(e)} ###") # Stampa per il debug
|
76 |
return f"ValueError: {str(e)}", 500
|
77 |
except IOError as e:
|
78 |
+
print(f"### IOError: {str(e)} ###") # Stampa per il debug
|
79 |
return f"IOError: {str(e)}", 500
|
80 |
except Exception as e:
|
81 |
+
print(f"### Unexpected error: {str(e)} ###") # Stampa per il debug
|
82 |
return f"Unexpected error: {str(e)}", 500
|
83 |
|
84 |
+
# Salvataggio immagine e invio come risposta
|
85 |
img_io = io.BytesIO()
|
86 |
final_image.save(img_io, 'JPEG')
|
87 |
img_io.seek(0)
|
88 |
+
print("### Immagine finale pronta per il download ###") # Stampa per il debug
|
89 |
|
|
|
90 |
return send_file(img_io, mimetype='image/jpeg', as_attachment=True, download_name='inmemory_image.jpg')
|
91 |
|
92 |
except Exception as e:
|
93 |
+
print(f"### Errore globale: {str(e)} ###") # Stampa per il debug
|
94 |
return str(e), 500
|
95 |
|
96 |
if __name__ == '__main__':
|