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))