Spaces:
Sleeping
Sleeping
import os | |
import openai | |
import streamlit as st | |
# Set up OpenAI API key | |
openai.api_key = str(os.getenv("API_KEY")) | |
# Load context from file | |
with open("RamsayPersonality_V1.txt", "r") as file: | |
context = file.read() | |
# Streamlit app logic | |
st.title("RamsayGPT👨🍳") | |
st.write("Welcome to RamsayGPT! Start by saying hello to Chef Ramsay!") | |
st.write("Type menu to see learning options. And if you can't handle the heat, type quit to end the tutoring session.") | |
menu = """Here are your options: | |
1. Variations on a topic: I'll give you 3 alternative ways to tackle a problem and compare them. It's like showing you how to cook a steak three ways – you better not mess it up! | |
2. Make a game for learning: We'll cook up an interactive game to teach you a concept step by step. Think of it as crafting a dessert with layers of information. Don't burn it! | |
3. Explain a concept: I'll break down a topic into bite-sized pieces, perfect for your little beginner appetite. It's like explaining how to boil an egg to a clueless sous chef. | |
""" | |
class RamsayGPT: | |
def __init__(self, api_key): | |
openai.api_key = api_key | |
self.context = context | |
self.chat_log = [{'role': 'assistant', 'content': self.context}] | |
def ramsay_response(self, user_message): | |
self.chat_log.append({"role": "user", "content": user_message}) | |
try: | |
response = openai.chat.completions.create( | |
model="gpt-4", | |
messages=self.chat_log | |
) | |
assistant_response = response.choices[0].message.content | |
except Exception as e: | |
assistant_response = f"Something went wrong, you donkey! Error: {str(e)}" | |
return assistant_response.strip() | |
def chat(self): | |
print("Say hello to Chef Ramsay!") | |
while True: | |
user_message = input("You:") | |
if user_message.lower() == "quit": | |
print("RamsayGPT: Can't handle the heat? Pathetic. Now get out of my kitchen!!") | |
break | |
elif user_message.lower() == "menu": | |
print("RamsayGPT:", self.show_menu()) | |
else: | |
assistant_response = self.ramsay_response(user_message) | |
print("RamsayGPT:", assistant_response) | |
self.chat_log.append({"role": "assistant", "content": assistant_response}) | |
# Example usage | |
if st.button('Show Menu'): | |
st.write(menu) | |
user_input = st.text_input("You:") | |
if user_input: | |
ramsay = RamsayGPT(openai.api_key) | |
response = ramsay.ramsay_response(user_input) | |
st.write(response) | |