shuff-chat / app.py
TRaw's picture
Create app.py
bd12395
from flask import Flask, request, render_template
import openai
import time
import random
# Set up your OpenAI API credentials
openai.api_key = 'YOUR_API_KEY'
# Initialize the Flask application
app = Flask(__name__)
# Helper function for the TypeShuffle animation
def type_shuffle(text, delay=0.03):
result = ""
for char in text:
random_chars = random.choice(['!', '@', '#', '$', '%', '^', '&', '*', '-', '=', '+', '<', '>'])
result += random_chars
time.sleep(delay)
time.sleep(1)
return result + text
# Generate a response from the ChatGPT model
def generate_response(prompt):
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{
"role": "system",
"content": "You are a helpful assistant."
},
{
"role": "user",
"content": prompt
}
],
)
return response['choices'][0]['message']['content']
@app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'POST':
user_input = request.form['user_input']
# Generate response
response = generate_response(user_input)
# Get the response with TypeShuffle animation
animated_response = type_shuffle(response)
return render_template('index.html', user_input=user_input, animated_response=animated_response)
return render_template('index.html')
if __name__ == "__main__":
app.run(port=5000, debug=True)