-
Notifications
You must be signed in to change notification settings - Fork 0
/
GPT.py
64 lines (59 loc) · 2.98 KB
/
GPT.py
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import os
import openai
import time
import streamlit as st
from dotenv import load_dotenv
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")
def chatGPT(userinput,temperature=0,max_tokens=1000):
response = openai.ChatCompletion.create(
model='gpt-3.5-turbo',
messages= [{"role":"user","content":userinput}],
temperature=temperature,
max_tokens=max_tokens
)
return response['choices'][0]['message']['content']
#Example Styles I came up with
styleList = ['None','Plagiarism Checker','Shakespearean Response','Python Interpreter','Java Code Generation','Music Suggestions','Hackathon Idea Generator']
colsec, col1, col2 = st.columns(3)
with colsec:
#temperature and max_tokens sliders
temperature = st.slider('The higher the temperature, the more random the response',0.0,2.0,0.0)
max_tokens = st.slider('Maximum number of tokens that ChatGPT will generate as a response',100,2500,100)
with col1:
style_option = st.selectbox("Choose a style: ",styleList)
if style_option == 'Plagiarism Checker':
user_input = st.text_input("Write the text you want to check for plagiarism: ")
if user_input:
res = chatGPT("Check the following text for plagiarism: "+user_input,temperature,max_tokens)
elif style_option == 'Shakespearean Response':
user_input = st.text_input("Write the text you want to be rewritten in a Shakespearean way: ")
if user_input:
res = chatGPT("Rewrite the following text in a Shakespearean way: "+user_input,temperature,max_tokens)
elif style_option == 'Python Interpreter':
user_input = st.text_input("Write down the coding problem: ")
if user_input:
res = chatGPT("Write a Python program that performs the following description: "+user_input,temperature,max_tokens)
elif style_option == 'Java Code Generation':
user_input = st.text_input("Write down the coding problem: ")
if user_input:
res = chatGPT("Write a Java program that performs the following description: "+user_input,temperature,max_tokens)
elif style_option == 'Music Suggestions':
user_input = st.text_input("Write down the name of the song and the artist who performed it: ")
if user_input:
res = chatGPT("Suggest songs similar to the given song: "+user_input,temperature,max_tokens)
elif style_option == 'Hackathon Idea Generator':
user_input = st.text_input("Write down the theme you want ideas for: ")
if user_input:
res = chatGPT("Generate five ideas based on this theme: "+user_input,temperature,max_tokens)
else:
user_input = st.text_input("Ask ChatGPT anything :) ")
if user_input:
res = chatGPT(user_input,temperature,max_tokens)
with col2:
if user_input:
with st.spinner('Processing...'):
time.sleep(1)
st.write("ChatGPT Response: ")
st.write(res)
st.success("Response Received!",icon="✅")