File size: 799 Bytes
dfce866
101ae78
dfce866
101ae78
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c1b3cb1
101ae78
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import streamlit as st
from openai import OpenAI

st.title("AnjaGPT")

user_api_key = st.text_input("Enter your openai key",  type="password")

user_prompt = st.text_input("Enter your prompt")

if user_api_key:
    client = OpenAI(api_key=user_api_key)

    try:
      response = client.chat.completions.create(
          model="gpt-3.5-turbo",
          messages=[
              {"role": "system", "content": "You are a helpful AI called Anja. Answer user in a sarcastic manner and be very concise."},
              {"role": "user", "content": user_prompt}
              ]
      )
      st.write("AnjaGPT Response:")
      st.write(response.choices[0].message.content)

    except Exception as e:
      st.error(f"An error occured: {e}")

else:
  st.info("Please enter your OpenAI key to continue")