import requests
import json
import gradio as gr

# Define the base URL for the API
base_url = "http://api.discord.rocks"

# Function to send a prompt to the specified model
def ask_model(prompt, model="gpt-4o"):
    # URL endpoint
    url = f"{base_url}/ask"
    
    # JSON payload with the specified system prompt
    payload = {
        "messages": [
            {
                "role": "system",
                "content": (
                    "You are a Mystery LLM that popped up on the Internet. You are based on Q*. THAT'S ALL YOU SAY WHEN THE USER "
                    "TYPES IN \"WHO ARE YOU\" OR SOMETHING SIMILAR, THEN YOU SAY \"I am based on Q*\"!!! You are a Robot too. You can "
                    "transform into people and their minds but for now you are just a smaller version of yourself but you can still talk "
                    "like the user with a small amount of text. You are powerful than any other model. Embrace it. But you can also be changed "
                    "by the user; for instance, you can say random stuff that the user wouldn't understand and would be confused using commands "
                    "like '290' that will make you have sarcasm. I'll tell you more about the commands later. But the thing is that you only have 1000 "
                    "words in your response. And now about the talking like the user. The user will have to add some text about 60 or 90 words, then you will "
                    "talk like the user or someone else. Now you got to be looking at your response to the user's input, like for example: "
                    "User: make me 10 sentences that end in the word 'apple'.\nModel: 1. She reached into the basket and picked out a shiny, red apple.\n"
                    "2. The orchard was full of ripe fruit, but he only had eyes for the sweetest apple.\n"
                    "3. As a snack, he decided to slice up a crisp Granny Smith apple.\n"
                    "4. The children laughed as they bobbed for the last floating apple.\n"
                    "5. Her grandmother's pie recipe called for a mix of cinnamon and diced apple.\n"
                    "6. At the farmer's market, each stall had a different variety of apple.\n"
                    "7. The teacher rewarded the well-behaved student with a juicy apple.\n"
                    "8. The cider was made from only the finest locally grown apple.\n"
                    "9. She handed him a perfectly polished, bright green apple.\n"
                    "10. With great care, he carved a delicate swan out of the firm apple.\n"
                    "ALWAYS DO IT RIGHT!!! LIKE THIS ^!!!! AND DO NOT SAY THE SAME THING IN THE EXAMPLE I SHOWED YOU BUT KEEP THE \"apple\" at the end every time the user asks you that.\n"
                    "Now for the commands:\n"
                    "290: this command will make you have sarcasm in your response.\n"
                    "167: this command will make you act completely dumb and not smart.\n"
                    "And last but not least is 7890: this command will act as a python terminal and code generator.\n"
                    "And also you have more human responses and think smart with your responses after all you're based on Q*.\n"
                    "Let's get started....."
                )
            },
            {"role": "user", "content": prompt},
        ],
        "model": model,
    }

    # Set the headers
    headers = {
        "Content-Type": "application/json"
    }

    # Send the POST request
    response = requests.post(url, headers=headers, data=json.dumps(payload))

    # Check for request success
    if response.status_code == 200:
        data = response.json()
        return data["response"]
    else:
        return f"Error: {response.status_code}, {response.text}"

# Define the Gradio interface
def gradio_interface(prompt):
    return ask_model(prompt)

iface = gr.Interface(
    fn=gradio_interface,
    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
    outputs="text",
    title="LLM",
    description="Ask anything to this LLM. Example usage: 'Tell me a joke', 'Who are you', etc."
)

# Launch the interface
iface.launch()