Update app.py

app.py

@@ -1,30 +1,26 @@
 
-# Set your OpenAI API key
-openai.api_key = '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'  # Replace with your actual OpenAI API key
 import streamlit as st
-import openai  
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from gtts import gTTS
+import torch
 import os
 
-# Set your OpenAI API key
-openai.api_key = 'your_openai_api_key'  # Replace with your actual OpenAI API key
+# Load the pre-trained language model and tokenizer from Hugging Face
+model_name = "gpt2-medium"  # Replace with a specific Trump-like model if available on Hugging Face
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Function to generate Trump-like response using OpenAI's GPT-3.5 or GPT-4
+# Set up a text generation pipeline
+generator = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
+
+# Function to generate Trump-like response
 def generate_response(prompt):
-    # 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}"
+    # Craft the prompt to encourage a Trump-like response
+    trump_prompt = f"Donald Trump says: {prompt}"
     
-    # Call OpenAI API to generate response using the updated format
-    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']
+    # Generate the response
+    response = generator(trump_prompt, max_length=100, num_return_sequences=1, temperature=0.8)
+    return response[0]['generated_text']
 
 # Function to convert text to audio using gTTS
 def generate_audio(text):
@@ -34,7 +30,7 @@ def generate_audio(text):
     return audio_path
 
 # Streamlit app UI
-st.title("Trump-like ChatGPT")
+st.title("Trump-like Chat Assistant")
 st.write("Type in a question or statement, and receive a 'Trump-style' response in text and audio!")
 
 # Text input from user