Update app.py

app.py

@@ -15,16 +15,28 @@ interface = outetts.InterfaceHF(model_version="0.2", cfg=model_config)
 st.title("OuteTTS Speech Synthesis")
 st.write("Enter text below to generate speech.")
 
+# Sidebar for reference voice
+st.sidebar.title("Voice Cloning")
+reference_audio = st.sidebar.file_uploader("Upload a reference audio (wav)", type=["wav"])
+
+if reference_audio:
+    ref_audio_path = "reference.wav"
+    with open(ref_audio_path, "wb") as f:
+        f.write(reference_audio.read())
+else:
+    ref_audio_path = None
+
 text_input = st.text_area("Text to convert to speech:", "Hello, this is an AI-generated voice.")
 
 if st.button("Generate Speech"):
     with st.spinner("Generating audio..."):
-        # Generate speech
+        # Generate speech with reference audio
         output = interface.generate(
             text=text_input,
             temperature=0.1,
             repetition_penalty=1.1,
-            max_length=4096
+            max_length=4096,
+            speaker_wav=ref_audio_path if ref_audio_path else None
         )
         
         # Save the synthesized speech to a file