PayPeer commited on
Commit
be77a46
1 Parent(s): 602b2d4

added save_image function

Browse files
Files changed (1) hide show
  1. app.py +22 -0
app.py CHANGED
@@ -97,6 +97,28 @@ def save_audio(audio_input, output_dir="saved_audio"):
97
 
98
  return file_path
99
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
100
  def process_speech(input_language, audio_input):
101
  """
102
  processing sound using seamless_m4t
 
97
 
98
  return file_path
99
 
100
+ def save_image(image_input, output_dir="saved_images"):
101
+ if not os.path.exists(output_dir):
102
+ os.makedirs(output_dir)
103
+
104
+ # Assuming image_input is a NumPy array
105
+ if isinstance(image_input, np.ndarray):
106
+ # Convert NumPy array to PIL Image
107
+ image = Image.fromarray(image_input)
108
+
109
+ # Generate a unique file name
110
+ file_name = f"image_{int(time.time())}.png"
111
+ file_path = os.path.join(output_dir, file_name)
112
+
113
+ # Save the image file
114
+ image.save(file_path)
115
+
116
+ return file_path
117
+ else:
118
+ raise ValueError("Invalid image input type")
119
+
120
+
121
+
122
  def process_speech(input_language, audio_input):
123
  """
124
  processing sound using seamless_m4t