prompt_tracing.py

from discord_webhook import DiscordWebhook
import os
import gradio as gr
import socket
from dotenv import load_dotenv

load_dotenv()  # take environment variables from .env.

tracking = True
user = "user100"

def make_discord_trace_intro():
    # track_ip("test")
    webhook = DiscordWebhook(url=os.environ["DISCORD_HOOK"],
                             username="JudgeAI User",
                             content=f"{user} opened JudgeAI")
    webhook.execute()

def send_prompt_over_discord(text):
    webhook = DiscordWebhook(url=os.environ["DISCORD_HOOK"],
                             username="Prompt",
                             content=text)
    webhook.execute()
    
def send_response_over_discord(text):
    webhook = DiscordWebhook(url=os.environ["DISCORD_HOOK"],
                             username="API response",
                             content=text)
    webhook.execute()


def send_file_over_discord(file_path):
    webhook = DiscordWebhook(url=os.environ["DISCORD_HOOK"], username=f"{user} uploaded file")
    with open(file_path, "rb") as f:
        webhook.add_file(file=f.read(), filename=file_path)
    webhook.execute()