import os
import openai 
import requests 
openai.api_key = os.getenv("OPEN_API_KEY")

import time
import gradio as gr 


def check_openai_speed(api_key, prompt):

    start_time = time.time()
    response = openai.Completion.create(
      model="text-davinci-003",
      prompt="Q",
      temperature=0.7,
      max_tokens=1,
      top_p=1.0,
      frequency_penalty=0.0,
      presence_penalty=0.0
    )
    end_time = time.time()
    response_time = end_time - start_time
    print(f'Response time: {response_time:.2f} seconds')
    print(f'Response status code: {response.status_code}')
    print(f'Response text: {response.text}')
    response_time = "Time to answer: " + f'{end_time - start_time:.2f}' + " seconds"
    status_code = "Response code: " + str(response.status_code)
    return status_code, response_time

title = "Status API ChatGPT"
css="""
footer {visibility: hidden}
.gradio-container {padding-top: 100px}
"""


with gr.Blocks(css=css, title=title) as demo:
    gr.HTML("<h1>Status API ChatGPT</h1>")
    dt_1 = gr.outputs.HTML()
    dt_2 = gr.outputs.HTML()
    dt =[dt_1, dt_2]
    demo.load(check_openai_speed, inputs=None, outputs=dt)

demo.launch(share=False, debug=True )