# coding: utf-8

# Author: Du Mingzhe (mingzhe@nus.edu.sg)
# Date: 2025-03-01

import time
import requests
import pandas as pd
import streamlit as st
from code_editor import code_editor

lang_map = {
    "Python": ["python", "python", "# Don't Worry, You Can't Break It. We Promise.\nprint('Hello World')\n"],
    "CPP": ["c_cpp", "cpp", "// Don't Worry, You Can't Break It. We Promise.\n// For Cpp, please make sure the program lasts at least 1 ms.\n#include <iostream>\nint main() {\n    std::cout << \"Hello, World!\" << std::endl;\n    return 0;\n}\n"],
    "Java": ["java", "java", "// Don't Worry, You Can't Break It. We Promise.\nclass HelloWorld {\n    public static void main(String[] args) {\n        System.out.println(\"Hello, World!\"); \n    }\n}\n"],
    "JavaScript": ["javascript", "javascript", "// Don't Worry, You Can't Break It. We Promise.\nconsole.log('Hello World!');\n"],
    "Golang": ["golang", "go", "// Don't Worry, You Can't Break It. We Promise.\npackage main\n\nimport \"fmt\"\n\nfunc main() {\n    fmt.Println(\"Hello, World!\")\n}\n"],
    "Ruby": ["ruby", "ruby", "# Don't Worry, You Can't Break It. We Promise.\nputs 'Hello World'\n"],
    "Rust": ["rust", "rust", "// Don't Worry, You Can't Break It. We Promise.\nfn main() {\n\tprintln!('Hello World!'); \n}\n"],
}

def post_task(lang, code, libs=None, timeout=30, run_profiling=False):
    url = 'https://monolith.cool/execute'
    data = {'language': lang, 'code': code, 'libraries': libs, 'timeout': timeout, 'run_profiling': run_profiling}
    response = requests.post(url, json=data)
    return response.json()

# Title
st.title("_Monolith_ is :blue[Cool] :sunglasses:")
st.markdown("[![GitHub Badge](https://img.shields.io/badge/GitHub-Elfsong/Monolith-blue)](https://github.com/Elfsong/Monolith)")
# st.write("Please feel free to have a try. Should you have any question, contact mingzhe@nus.edu.sg")
# st.write("[UPDATE] Hosting the service is costly, so I’ve enabled the spot instance feature. DM me if you need access.")

# Language
lang = st.selectbox("Language?", lang_map.keys(), help="the language for submission.")
editor_language = lang_map[lang][0]
monolith_language = lang_map[lang][1]

# Libraries
lib_str = st.text_input("Library?", placeholder="Package A, Package B, ... , Package N", help="if any libraries are needed. Seperate with a comma.")
libraries = [lib.strip() for lib in lib_str.split(",")] if lib_str else []

# Memory Profile
profiling = st.checkbox("Time/Memory Integral?", help="Enable time-memory integral profiling for the code execution.")

# Timeout
timeout = st.number_input("Timeout?", min_value=1, max_value=120, value=30, help="the maximum time allowed for execution.")

# Code Editor
editor_buttons = [{
    "name": "Submit", 
    "feather": "Play",
    "primary": True, 
    "hasText": True, 
    "showWithIcon": True, 
    "commands": ["submit"], 
    "style": {"bottom": "0.44rem","right": "0.4rem"}
}]
code_prompt = lang_map[lang][2]
response_dict = code_editor(code_prompt, lang=editor_language, height=[15,15], options={"wrap": False}, buttons=editor_buttons)

if response_dict['type'] == 'submit':
    code = response_dict['text']

    with st.spinner('Ok, give me a sec...', show_time=True):
        response = post_task(monolith_language, code, libraries, timeout, profiling)
        task_id = response['task_id']
        st.write(f"Task ID: {task_id}")
    
    if response['output_dict'] and 'stdout' in response['output_dict']:
        st.success(response['output_dict']['stdout'])
    
    if response['output_dict'] and 'stderr' in response['output_dict']:
        st.warning(response['output_dict']['stderr'])
    
    if response['status'] == "success":
        if profiling:
            st.write(f"**Execution Time:** :blue[{response['output_dict']['duration']}] ms, **Peak Memory:** :blue[{response['output_dict']['peak_memory']}] kb, **Integral:** :blue[{response['output_dict']['integral']}] kb*ms")
            # st.write(response)
            st.area_chart(pd.DataFrame(response['output_dict']['log'], columns=["timestemp", "memory"]), x='timestemp', y='memory')
        else:
            st.write(f"**Elapsed Time:** :blue[{response['output_dict']['time_v']['elapsed_time_seconds']}], **System Time:** :blue[{response['output_dict']['time_v']['system_time']}], **User Time:** :blue[{response['output_dict']['time_v']['user_time']}], **Peak Memory:** :blue[{response['output_dict']['time_v']['max_resident_set_kb']}] kb")

    else:
        st.error(response)