arkaprav0's picture
Update app.py
68678da
raw
history blame contribute delete
No virus
1.6 kB
import openai
import os
import sys
import gradio as gr
from utils import process_dict, run_gpt_3, call3, call4, clean_and_concatenate_dict_values
def app(file):
try:
openai.api_key = os.environ['OPENAI_API_KEY']
except KeyError:
sys.stderr.write("""
You haven't set up your API key yet.
If you don't have an API key yet, visit:
You're a moron.
""")
exit(1)
with open(file.name, 'r') as f:
text = f.read()
batch_dict = process_dict(text, 20)
topic_dict = run_gpt_3(batch_dict, call3)
topic_text = clean_and_concatenate_dict_values(topic_dict)
result = call4(topic_text)
return result
iface = gr.Interface(fn=app, inputs="file", outputs="text")
iface.launch()
# response = openai.ChatCompletion.create(
# model="gpt-4", # only available if OpenAI has given you early access, otherwise use: "gpt-3.5-turbo"
# # 32K context gpt-4 model: "gpt-4-32k"
# messages=[
# {"role": "system", "content": "You are a helpful assistant."},
# {"role": "user", "content": "Who won the world series in 2020?"},
# {"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
# {"role": "user", "content": "Where was it played?"}
# ]
# )
text = """
6.08 seconds - Yeah, the Jack Carr one was pretty fun.
11.32 seconds - He's super nice.
16.56 seconds - I'm really enjoying this book.
21.80 seconds - I can't wait to see what happens next.
27.04 seconds - This is a great read.
32.28 seconds - I highly recommend it to anyone who enjoys thrillers.
"""
print(process_dict(text))