improved code format

app.py

@@ -6,10 +6,13 @@ from TTS.api import TTS
 # By using XTTS you agree to CPML license https://coqui.ai/cpml
 os.environ["COQUI_TOS_AGREED"] = "1"
 
-# Initialize model
-model = "tts_models/multilingual/multi-dataset/xtts_v2"
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
-tts = TTS(model).to(device)
+def generate_audio(audio_file, text_input):
+  # Initialize model
+  model = "tts_models/multilingual/multi-dataset/xtts_v2"
+  device = 'cuda' if torch.cuda.is_available() else 'cpu'
+  tts = TTS(model).to(device)
+
+  tts.tts_to_file(text=text_input, speaker_wav=audio_file, language='en')
 
 def main():
   # Title
@@ -17,7 +20,7 @@ def main():
   st.markdown(title, unsafe_allow_html=True)
 
   # Subtitle
-  title = f"""<h2 align="center" style="font-size: 1.5rem";>Make your favorite characters say anything!</h2>"""
+  title = f"""<h2 align="center" style="font-size: 1.2rem; margin-bottom: 2rem;">Make your favorite characters say anything!</h2>"""
   st.markdown(title, unsafe_allow_html=True)
 
   # Upload audio file
@@ -27,11 +30,18 @@ def main():
   # Input text 
   text_input = st.text_input('Enter the text to synthesize')
 
-  if st.button('Synthesize') and uploaded_file is not None and text_input is not "":
-    with st.spinner('Synthesizing...'):
-      output_audio = tts.tts_to_file(text=text_input, speaker_wav=uploaded_file, language='en')
-
-    st.audio(output_audio, format='audio/wav')
+  if uploaded_file is not None and text_input is not "":
+    if st.button('Synthesize'):
+      with st.spinner('Synthesizing...'):
+        output_audio = generate_audio(uploaded_file, text_input)
+
+      col1, col2 = st.columns(2)
+      with col1:
+        st.header('Reference Audio')
+        st.audio(uploaded_file, format='audio/wav')
+      with col2:
+        st.header('Synthesized Audio')
+        st.audio(output_audio, format='audio/wav')
 
 if __name__ == '__main__':
   main()