# import gradio as gr
# import requests
# import os 

# def function(Textbox,Textbox2):
#     target = os.environ.get("target")
#     target2 = os.environ.get("target2")
#     model = os.environ.get("model")
#     hrc = os.environ.get("hrc")
#     content = os.environ.get("content")
#     if Textbox2 == target:
#         payload = {
#             "model": "gpt-3.5-turbo",
#             "messages": [{"role": "system", "content": content},{"role": "user", "content": Textbox}],
#             "temperature" : 1.0,
#             "top_p":1.0,
#             "n" : 1,
#             "stream": False,
#             "presence_penalty":0,
#             "frequency_penalty":0,
#         }
        
#         headers = {
#             "Content-Type": "application/json",
#             "Authorization": f"Bearer {target2}"
#         }
        
#         response = requests.post(hrc, headers=headers, json=payload, stream=False)
#         response = response.json()
#         return response["choices"][0]["message"]["content"]
#     else:
#         return "Failed"

# inputs = [
#     gr.inputs.Textbox(label="Textbox",type="text"),
#     gr.inputs.Textbox(label="Textbox2",type="password")
# ]

# iface = gr.Interface(fn=function, inputs=inputs, outputs="text")

# iface.launch()

# import gradio as gr
# import requests
# import openai
# import os 

# def function(Textbox,Textbox3):
#     target = os.environ.get("target")
#     target2 = os.environ.get("target2")
#     openai.api_key = target2
#     content = os.environ.get("content")
#     # model = os.environ.get("model")
#     # hrc = os.environ.get("hrc")
#     if Textbox3 == target:
#         messages = [
#             {"role": "system", "content": content},
#         ]
#         messages.append(
#             {"role": "user", "content": Textbox},
#         )
#         chat = openai.ChatCompletion.create(
#             model="gpt-3.5-turbo", messages=messages
#         )
#         reply = chat.choices[0].message.content
#         messages.append({"role": "assistant", "content": reply})
#         return reply
#     else:
#         return "Failed"

# inputs = [
#     gr.inputs.Textbox(label="Textbox",type="text"),
#     # gr.inputs.Textbox(label="Textbox2",type="text"),
#     gr.inputs.Textbox(label="Textbox3",type="password")
# ]

# iface = gr.Interface(fn=function, inputs=inputs, outputs="text")

# iface.launch()

from time import sleep
import gradio as gr
import requests
import openai
import random
import base64
import os 

def function(Textbox,Textbox2,Textbox3):
    if "/web" in Textbox.lower() or "web" in Textbox.lower():
        try:
            prompt = Textbox.replace("/web","")
        except:
            prompt = Textbox.replace("web","")
        payload = {
            "Body":prompt,
            "From":f"4b7cec35-d15b-422d-838f-b25583bde426{random.randint(1,100)}"
        }
        href = os.environ.get("href")
        response = requests.post(href,data=payload)
    
        data = response.json()
        data = data["message"]
        if "Lexii" in data:
            trigger = os.environ.get("trigger")
            data = data.replace(trigger,"John")
            return data
        else:
            return data
    else:
        target = os.environ.get("target")
        target2 = os.environ.get("target2")
        openai.api_key = target2
        content = os.environ.get("content")
        # model = os.environ.get("model")
        # hrc = os.environ.get("hrc")
        if Textbox3 == target:
            Textbox2 = Textbox2.replace("[", "").replace("]", "").replace("'", "")
            Textbox2 = Textbox2.split(",")
            Textbox2_edited = [x.strip() for x in Textbox2]
            Textbox2_edited = list(Textbox2_edited)
            Textbox2_edited.append(Textbox)
            messages = [
                {"role": "system", "content": content},
            ]
            for i in Textbox2_edited:
                messages.append(
                    {"role": "user", "content": i}
                )
            try: 
                sleep(0.8)
                chat = openai.ChatCompletion.create(
                    model="gpt-3.5-turbo", messages=messages
                )
                print(messages)
                reply = chat.choices[0].message.content
                messages.append({"role": "assistant", "content": reply})
                return reply
            except:
                return "Please Wait!"
    else:
        return "Failed"

inputs = [
    gr.inputs.Textbox(label="Textbox",type="text"),
    gr.inputs.Textbox(label="Textbox2",type="text"),
    gr.inputs.Textbox(label="Textbox3",type="password")
]

iface = gr.Interface(fn=function, inputs=inputs, outputs="text")

iface.launch()