import os
from gpt_index import GPTSimpleVectorIndex
import gradio as gr
API_URL='https://api.openai.com/v1/chat/completions'
openai_api_key=os.environ['OPENAI_API_KEY']
if openai_api_key is None:raise ValueError('Did not find openai_api_key, please add an environment variable `OPENAI_API_KEY` which contains it, or pass `openai_api_key` as a named parameter.')
top_p_chatgpt=1.
temperature_chatgpt=1.
def predict_chatgpt(inputs,chat_counter_chatgpt,chatbot_chatgpt=[],history=[]):
        M='user';J='content';I='role';D=chat_counter_chatgpt;C=inputs;A=history
        if D!=0:
                E=[]
                for K in chatbot_chatgpt:F={};F[I]=M;F[J]=K[0];G={};G[I]='assistant';G[J]=K[1];E.append(F);E.append(G)
                H={};H[I]=M;H[J]=C;E.append(H)
        D+=1;A.append('You asked: '+C);N=GPTSimpleVectorIndex.load_from_disk('IPCIndex.json');O=N.query(C);P=O.response.split();L=0;B='';S=0
        for Q in P:
                B=B+' '+Q
                if L==0:A.append(' '+B)
                else:A[-1]=B
                R=[(A[B],A[B+1])for B in range(0,len(A)-1,2)];L+=1;yield(R,A,D)
def reset_textbox():return gr.update(value='')
def reset_chat(chatbot,state):return None,[]
with gr.Blocks(css='#col_container {width: 1000px; margin-left: auto; margin-right: auto;}\n                #chatgpt {height: 500px; overflow: auto;}} ',theme=gr.themes.Default(primary_hue='slate'))as LegalProdigy:
        with gr.Row():
                with gr.Column(scale=14):
                        with gr.Box():
                                with gr.Row():
                                        with gr.Column(scale=13):inputs=gr.Textbox(label='Ask any legal query ⤵️    Try : Explain arbitration process')
                                        with gr.Column(scale=1):b1=gr.Button('Submit',elem_id='submit').style(full_width=True);b2=gr.Button('Clear',elem_id='clear').style(full_width=True)
                                        state_chatgpt=gr.State([])
                        with gr.Box():
                                with gr.Row():chatbot_chatgpt=gr.Chatbot(elem_id='chatgpt',label='Legal Prodigy')
        chat_counter_chatgpt=gr.Number(value=0,visible=False,precision=0);inputs.submit(reset_textbox,[],[inputs]);b1.click(predict_chatgpt,[inputs,chat_counter_chatgpt,chatbot_chatgpt,state_chatgpt],[chatbot_chatgpt,state_chatgpt]);b2.click(reset_chat,[chatbot_chatgpt,state_chatgpt],[chatbot_chatgpt,state_chatgpt]);LegalProdigy.queue(concurrency_count=16).launch(height=2500,debug=True)