Update app.py

app.py

@@ -25,8 +25,8 @@ def convert_to_wav(audio_file):
  # Extract file extension
  file_extension = os.path.splitext(audio_file)[-1].lower()
 
- # Convert audio file to .wav format
- if file_extension != ".wav":
+ # Convert audio file to.wav format
+ if file_extension!= ".wav":
  audio = AudioSegment.from_file(audio_file)
  audio.export("temp.wav", format="wav")
  os.remove(audio_file)
@@ -34,21 +34,21 @@ def convert_to_wav(audio_file):
 
  return audio_file
 
-def clone(text, url, language):
- # Download zip file
- response = requests.get(url)
- with open("temp.zip", "wb") as f:
- f.write(response.content)
-
- # Extract audio file from zip archive
- with zipfile.ZipFile("temp.zip", "r") as zip_ref:
- for file in zip_ref.namelist():
- if os.path.splitext(file)[-1].lower() in AUDIO_FORMATS:
- zip_ref.extract(file, ".")
- audio_file = file
- break
+def clone(text, audio_or_url, language):
+ # Check if audio_or_url is a file object or a string
+ if isinstance(audio_or_url, str):
+ # Download audio file from URL
+ response = requests.get(audio_or_url)
+ with open("temp.wav", "wb") as f:
+ f.write(response.content)
+ audio_file = "temp.wav"
+ else:
+ # Save uploaded audio file to a temporary location
+ audio_file = "temp.wav"
+ with open(audio_file, "wb") as f:
+ f.write(audio_or_url.read())
 
- # Convert audio file to .wav format
+ # Convert audio file to.wav format
  audio_file = convert_to_wav(audio_file)
 
  # Generate audio using TTS model
@@ -56,16 +56,15 @@ def clone(text, url, language):
 
  # Clean up
  os.remove(audio_file)
- os.remove("temp.zip")
 
  return "./output.wav"
 
 # Create Gradio interface
 iface = gr.Interface(
  fn=clone,
- inputs=["text", gr.components.Text(label="URL"), gr.Dropdown(choices=LANGUAGES, label="Language")],
- outputs=gr.Audio(type='filepath'),
- title='Voice Clone',
+ inputs=["text", gr.Audio(label="Upload audio file or enter URL"), gr.Dropdown(choices=LANGUAGES, label="Language")],
+ outputs=gr.outputs.Audio(type="filepath"),
+ title="Voice Clone",
  description=""" by [Angetyde](https://youtube.com/@Angetyde?si=7nusP31nTumIkPTF) and [Tony Assi](https://www.tonyassi.com/ ) use this colab with caution <3. """,
  theme=gr.themes.Base(primary_hue="teal", secondary_hue="teal", neutral_hue="slate")
 )