from huggingface_hub import from_pretrained_fastai
import gradio as gr
from fastai.text.all import *

cpp_code = [
    """// Esto es código 'Hello World' en C++
#include <iostream>
int main() {
    std::cout << "Hello World" << std::endl;
    return 0;
}"""
]

fortran_code = [
    """! Esto es código 'Hello World' en Fortran
program hello
    print *, "Hello World"
end program hello"""
]

go_code = [
    """// Esto es código 'Hello World' en Go
package main
import "fmt"

func main() {
    fmt.Println("Hello World")
}"""
]

java_code = [
    """// Esto es código 'Hello World' en Java
public class Main {
    public static void main(String[] args) {
        System.out.println("Hello World");
    }
}"""
]

python_code = [
    """# Esto es código 'Hello World' en Python
print("Hello World")"""
]

ruby_code = [
    """# Esto es código 'Hello World' en Ruby
puts 'Hello World'"""
]

c_code = [
    """// Esto es código 'Hello World' en C
#include <stdio.h>

int main() {
    printf("Hello World\\n");
    return 0;
}"""
]

js_code = [
    """// Esto es código 'Hello World' en JavaScript
console.log("Hello World");"""
]


js_code = [
    "// Esto es código 'Hello World' en JavaScript",
    "console.log(\"Hello World\");"
]

examples = [cpp_code, fortran_code, go_code, java_code, python_code, ruby_code, c_code, js_code]

repo_id = "GabrielMI/entregable3"


learner = from_pretrained_fastai(repo_id)
labels = ["C", "C++", "Fortran", "Go", "Java", "JavaScript", "Python", "Ruby"]

def predict(text):
    pred, pred_idx, probs = learner.predict(text)
    return {labels[i]: float(probs[i]) for i in range(len(labels))}

gr.Interface(
    fn=predict,
    inputs=[gr.Textbox(label="Instrucciones")],
    outputs=gr.Label(num_top_classes=3),
    examples=examples
).launch(share=False)