import gradio as gr
from easyocr import Reader
from PIL import Image
import io
import json
import csv
import openai
import ast
import os


openai.api_key = os.getenv('API_KEY')
reader = Reader(["tr"])


def get_text(input_img):
    result = reader.readtext(input_img, detail=0)
    return " ".join(result)


def save_csv(mahalle, il, sokak, apartman):
    adres_full = [mahalle, il, sokak, apartman]

    with open("adress_book.csv", "a", encoding="utf-8") as f:
        write = csv.writer(f)
        write.writerow(adres_full)
    return adres_full


def get_json(mahalle, il, sokak, apartman):
    adres = {"mahalle": mahalle, "il": il, "sokak": sokak, "apartman": apartman}
    dump = json.dumps(adres, indent=4, ensure_ascii=False)
    return dump


def text_dict_il(input):
    eval_result = ast.literal_eval(input)["il"]

    return eval_result


def text_dict_mahalle(input):
    eval_result = ast.literal_eval(input)["mahalle"]

    return eval_result


def text_dict_ilce(input):
    eval_result = ast.literal_eval(input)["ilçe"]

    return eval_result


def text_dict_sokak(input):
    eval_result = ast.literal_eval(input)["sokak"]

    return eval_result


def text_dict_no(input):
    eval_result = ast.literal_eval(input)["no"]

    return eval_result


def text_dict_tel(input):
    eval_result = ast.literal_eval(input)["tel"]

    return eval_result


def text_dict_isim(input):
    eval_result = ast.literal_eval(input)["isim_soyisim"]
    return eval_result


def text_dict_adres(input):
    eval_result = ast.literal_eval(input)["adres"]

    return eval_result


def openai_response(ocr_input):
    prompt = f"""Tabular Data Extraction
You are a highly intelligent and accurate tabular data extractor from plain text input and especially from emergency text that carries address information, your inputs can be text of arbitrary size, but the output should be in [{{'tabular': {{'entity_type': 'entity'}} }}] JSON format

Force it to only extract keys that are shared as an example in the examples section, if a key value is not found in the text input, then it should be ignored and should be returned as an empty string

Have only il, ilçe, mahalle, sokak, no, tel, isim_soyisim, adres

Examples:


Input: Deprem sırasında evimizde yer alan adresimiz: İstanbul, Beşiktaş, Yıldız Mahallesi, Cumhuriyet Caddesi No: 35, cep telefonu numaram 5551231256, adim Ahmet Yilmaz
Output: [{{'Tabular': '{{'il': 'İstanbul', 'ilçe': 'Beşiktaş', 'mahalle': 'Yıldız Mahallesi', 'sokak': 'Cumhuriyet Caddesi', 'no': 35, 'tel': 5551231256, 'isim_soyisim': 'Ahmet Yılmaz', 'adres': 'İstanbul, Beşiktaş, Yıldız Mahallesi, Cumhuriyet Caddesi No: 35'}}' }}]


Input: {ocr_input}
Output:

"""

    response = openai.Completion.create(
        model="text-davinci-003",
        prompt=prompt,
        temperature=0,
        max_tokens=300,
        top_p=1,
        frequency_penalty=0.0,
        presence_penalty=0.0,
        stop=["\n"],
    )
    resp = response["choices"][0]["text"]
    resp = eval(resp.replace("'{", "{").replace("}'", "}"))
    resp = resp[0]["Tabular"]
    return resp


with gr.Blocks() as demo:
    gr.Markdown("""## Enkaz Bildirme""")
    gr.Markdown("Bu uygulamada ekran görüntüsü sürükleyip bırakarak AFAD'a enkaz bildirimi yapabilirsiniz.")
    with gr.Row():
        img_area = gr.Image(label="Ekran Görüntüsü")
        ocr_result = gr.Textbox(label="Metin")
    open_api_text = gr.Textbox(label="Tam Adres")
    submit_button = gr.Button(label="Görüntüyü Yükle", api_name="upload-file")
    with gr.Column():
        with gr.Row():
            il = gr.Textbox(label="İl")
            ilce = gr.Textbox(label="İlçe")
        with gr.Row():
            mahalle = gr.Textbox(label="Mahalle")
            sokak = gr.Textbox(label="Sokak/Cadde/Bulvar")
        with gr.Row():
            no = gr.Textbox(label="No")
            tel = gr.Textbox(label="Telefon")
        with gr.Row():
            isim_soyisim = gr.Textbox(label="İsim Soyisim")
            adres = gr.Textbox(label="Adres")

    submit_button.click(get_text, img_area, ocr_result)

    ocr_result.change(openai_response, ocr_result, open_api_text)

    open_api_text.change(text_dict_il, [open_api_text], il)
    open_api_text.change(text_dict_ilce, [open_api_text], ilce)
    open_api_text.change(text_dict_mahalle, [open_api_text], mahalle)
    open_api_text.change(text_dict_sokak, [open_api_text], sokak)
    open_api_text.change(text_dict_no, [open_api_text], no)
    open_api_text.change(text_dict_adres, [open_api_text], adres)
    open_api_text.change(text_dict_tel, [open_api_text], tel)
    open_api_text.change(text_dict_isim, [open_api_text], isim_soyisim)

    # json_out = gr.Textbox()
    # csv_out = gr.Textbox()

    # adres_submit = gr.Button()
    # adres_submit.click(get_json, [mahalle, il, sokak, apartman], json_out)
    # adres_submit.click(save_csv, [mahalle, il, sokak, apartman], csv_out)


if __name__ == "__main__":
    demo.launch()