Update app.py

app.py

@@ -1,97 +1,51 @@
 import streamlit as st
-from transformers import pipeline
+import openai
 from gtts import gTTS
-import requests
 import os
 
-# Initialize the text generation pipeline with GPT-2
-generator = pipeline('text-generation', model='gpt2')
+# Set your OpenAI API key
+openai.api_key = 'your_openai_api_key'  # Replace with your actual OpenAI API key
 
-# Initialize Whisper model for transcription
-transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base")
-
-# Function to generate a Trump-like response
+# Function to generate Trump-like response using OpenAI's GPT-3.5 or GPT-4
 def generate_response(prompt):
-    # This is your original `generate_response` function
-    response = generator(
-        prompt,
-        max_length=100,        # Adjust length as needed
-        num_return_sequences=1, # Generate only one response
-        temperature=0.7,        # Adjust for more randomness (higher) or more coherence (lower)
-    )
-    return response[0]['generated_text']
-
-# Uberduck API keys - Replace with your actual Uberduck API credentials
-UBERDUCK_API_KEY = 'pub_wxfzmwowyumqxfnwcc'  # Replace with your actual Uberduck API Key
-UBERDUCK_API_SECRET = 'pk_8e4b3ad9-d26a-49ca-ac83-4ddfe4dace85'  # Replace with your actual Uberduck API Secret
-
-# Function to generate audio using Uberduck API
-def generate_audio_uberduck(text):
-    url = "https://api.uberduck.ai/speak"
-    headers = {
-        "Accept": "application/json",
-        "uberduck-api-key": UBERDUCK_API_KEY,
-        "uberduck-api-secret": UBERDUCK_API_SECRET
-    }
-    json_data = {
-        "speech": text,
-        "voice": "donald-trump"
-    }
+    # Craft the prompt to instruct the model to respond in a Trump-like manner
+    trump_prompt = f"As if you were Donald Trump, respond to the following: {prompt}"
     
-    # Send a request to Uberduck's API to generate audio
-    response = requests.post(url, headers=headers, json=json_data)
-    if response.status_code == 200:
-        audio_url = response.json().get("path")
-        # Download the audio file from the URL
-        audio_data = requests.get(audio_url).content
-        with open("output.mp3", "wb") as file:
-            file.write(audio_data)
-        return "output.mp3"
-    else:
-        st.write("Error generating audio:", response.text)
-        return None
+    # Call OpenAI API to generate response
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",  # Or "gpt-4" if available in your account
+        messages=[
+            {"role": "system", "content": "You are to respond as Donald Trump would."},
+            {"role": "user", "content": trump_prompt}
+        ],
+        max_tokens=100,
+        temperature=0.8
+    )
+    return response.choices[0].message['content']
 
-# Alternative function to generate audio using gTTS
-def generate_audio_gtts(text):
+# Function to convert text to audio using gTTS
+def generate_audio(text):
     tts = gTTS(text=text, lang="en")
     audio_path = "output.mp3"
     tts.save(audio_path)
     return audio_path
 
-# Function to transcribe audio to text using Whisper
-def transcribe_audio(file_path):
-    # Transcribe audio using Whisper model
-    transcription = transcriber(file_path)
-    return transcription['text']
-
 # Streamlit app UI
-st.title("Trump-like Voice Assistant")
-st.write("Record or upload an audio clip and receive a 'Trump-style' response.")
+st.title("Trump-like Text-to-Speech Assistant")
+st.write("Type a question or statement, and receive a 'Trump-style' audio response!")
 
-# Upload or record audio
-uploaded_audio = st.file_uploader("Upload an audio file (wav format preferred)", type=["wav", "mp3"])
+# Text input from user
+user_input = st.text_input("Your message:", "Enter a message here")
 
-if uploaded_audio is not None:
-    # Save the uploaded audio file temporarily
-    audio_path = "input_audio.wav"
-    with open(audio_path, "wb") as f:
-        f.write(uploaded_audio.getbuffer())
+if user_input:
+    # Generate Trump-like response
+    trump_response = generate_response(user_input)
+    st.write("Trump-like Assistant:", trump_response)
+    
+    # Convert response to audio
+    audio_output_path = generate_audio(trump_response)
+    audio_file = open(audio_output_path, "rb")
+    audio_bytes = audio_file.read()
     
-    # Transcribe audio to text
-    user_input = transcribe_audio(audio_path)
-    if user_input:
-        st.write(f"Transcribed text: {user_input}")
-        
-        # Generate Trump-like response
-        trump_prompt = f"As a bold and confident leader, respond to: {user_input}"
-        trump_response = generate_response(trump_prompt)
-        st.write("Trump-like Assistant:", trump_response)
-        
-        # Generate audio response (choose either Uberduck or gTTS)
-        audio_output_path = generate_audio_uberduck(trump_response)  # Uberduck
-        # audio_output_path = generate_audio_gtts(trump_response)     # gTTS as fallback
-        
-        if audio_output_path:
-            audio_file = open(audio_output_path, "rb")
-            audio_bytes = audio_file.read()
-            st.audio(audio_bytes, format="audio/mp3")
+    # Display audio output
+    st.audio(audio_bytes, format="audio/mp3")