Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from gtts import gTTS
 import torch
+import os
 
 # Load DialoGPT model for conversational style
 model_name = "microsoft/DialoGPT-medium"
@@ -32,21 +33,25 @@ st.title("Trump-like Chat Assistant")
 st.write("Type in a question or statement, and receive a 'Trump-style' response in both text and audio!")
 
 # Text input from user
-user_input = st.text_input("Enter a message here")
+user_input = st.text_input("Your message:", "Enter a message here")
 
 if user_input:
     # Generate Trump-like response
     trump_response = generate_response(user_input)
     
     # Display text output of the response
-    
-    st.write(trump_response)
+    st.subheader("Trump-like Assistant (Text Response):")
+    st.write(trump_response)  # Show the generated text directly
     
     # Convert response to audio
     audio_output_path = generate_audio(trump_response)
-    audio_file = open(audio_output_path, "rb")
-    audio_bytes = audio_file.read()
     
-    # Display audio output
-    st.subheader("Trump-like Assistant (Audio Response):")
-    st.audio(audio_bytes, format="audio/mp3")
+    # Ensure the file exists and can be played back
+    if os.path.exists(audio_output_path):
+        with open(audio_output_path, "rb") as audio_file:
+            audio_bytes = audio_file.read()
+            # Display audio output
+            st.subheader("Trump-like Assistant (Audio Response):")
+            st.audio(audio_bytes, format="audio/mp3")
+    else:
+        st.error("Failed to generate audio. Please try again.")