Spaces:
Runtime error
Runtime error
| from PIL import ImageFilter, Image | |
| from easyocr import Reader | |
| import gradio as gr | |
| import numpy as np | |
| import openai | |
| import ast | |
| from transformers import pipeline | |
| import os | |
| from openai_api import OpenAI_API | |
| import utils | |
| openai.api_key = os.getenv("API_KEY") | |
| reader = Reader(["tr"]) | |
| def get_text(input_img): | |
| img = Image.fromarray(input_img) | |
| detailed = np.asarray(img.filter(ImageFilter.DETAIL)) | |
| result = reader.readtext(detailed, detail=0, paragraph=True) | |
| return " ".join(result) | |
| # Submit button | |
| def get_parsed_address(input_img): | |
| address_full_text = get_text(input_img) | |
| return ner_response(address_full_text) | |
| def save_deta_db(input): | |
| eval_result = ast.literal_eval(input) | |
| utils.write_db(eval_result) | |
| return | |
| def update_component(): | |
| return gr.update(value="Gönderildi, teşekkürler.", visible=True) | |
| def clear_textbox(value): | |
| return gr.update(value="") | |
| def text_dict(input): | |
| eval_result = ast.literal_eval(input) | |
| return ( | |
| str(eval_result["il"]), | |
| str(eval_result["ilce"]), | |
| str(eval_result["mahalle"]), | |
| str(eval_result["sokak"]), | |
| str(eval_result["Apartman/site"]), | |
| str(eval_result["no"]), | |
| str(eval_result["ad-soyad"]), | |
| str(eval_result["dis kapi no"]), | |
| ) | |
| def ner_response(ocr_input): | |
| ner_pipe = pipeline("token-classification","deprem-ml/deprem-ner", aggregation_strategy="first") | |
| predictions = ner_pipe(ocr_input) | |
| resp = {} | |
| for item in predictions: | |
| print(item) | |
| key = item["entity_group"] | |
| resp[key] = item["word"] | |
| resp["input"] = ocr_input | |
| dict_keys = ["il", "ilce", "mahalle", "sokak", "Apartman/site", "no", "ad-soyad", "dis kapi no"] | |
| for key in dict_keys: | |
| if key not in resp.keys(): | |
| resp[key] = "" | |
| return resp | |
| # User Interface | |
| with gr.Blocks() as demo: | |
| gr.Markdown( | |
| """ | |
| # Enkaz Bildirme Uygulaması | |
| """ | |
| ) | |
| gr.Markdown( | |
| "Bu uygulamada ekran görüntüsü sürükleyip bırakarak AFAD'a enkaz bildirimi yapabilirsiniz. Mesajı metin olarak da girebilirsiniz, tam adresi ayrıştırıp döndürür. API olarak kullanmak isterseniz sayfanın en altında use via api'ya tıklayın." | |
| ) | |
| with gr.Row(): | |
| with gr.Column(): | |
| img_area = gr.Image(label="Ekran Görüntüsü yükleyin 👇") | |
| img_area_button = gr.Button(value="Görüntüyü İşle", label="Submit") | |
| with gr.Column(): | |
| text_area = gr.Textbox(label="Metin yükleyin 👇 ", lines=8) | |
| text_area_button = gr.Button(value="Metni Yükle", label="Submit") | |
| open_api_text = gr.Textbox(label="Tam Adres") | |
| with gr.Column(): | |
| with gr.Row(): | |
| il = gr.Textbox(label="İl", interactive=True, show_progress=False) | |
| ilce = gr.Textbox(label="İlçe", interactive=True, show_progress=False) | |
| with gr.Row(): | |
| mahalle = gr.Textbox( | |
| label="Mahalle", interactive=True, show_progress=False | |
| ) | |
| sokak = gr.Textbox( | |
| label="Sokak/Cadde/Bulvar", interactive=True, show_progress=False | |
| ) | |
| with gr.Row(): | |
| no = gr.Textbox(label="Telefon", interactive=True, show_progress=False) | |
| with gr.Row(): | |
| ad_soyad = gr.Textbox( | |
| label="İsim Soyisim", interactive=True, show_progress=False | |
| ) | |
| apartman = gr.Textbox(label="apartman", interactive=True, show_progress=False) | |
| with gr.Row(): | |
| dis_kapi_no = gr.Textbox(label="Kapı No", interactive=True, show_progress=False) | |
| img_area_button.click( | |
| get_parsed_address, | |
| inputs=img_area, | |
| outputs=open_api_text, | |
| api_name="upload-image", | |
| ) | |
| text_area_button.click( | |
| ner_response, text_area, open_api_text, api_name="upload-text" | |
| ) | |
| open_api_text.change( | |
| text_dict, | |
| open_api_text, | |
| [il, ilce, mahalle, sokak, no, apartman, ad_soyad, dis_kapi_no], | |
| ) | |
| ocr_button = gr.Button(value="Sadece OCR kullan") | |
| ocr_button.click( | |
| get_text, | |
| inputs=img_area, | |
| outputs=text_area, | |
| api_name="get-ocr-output", | |
| ) | |
| submit_button = gr.Button(value="Veriyi Birimlere Yolla") | |
| submit_button.click(save_deta_db, open_api_text) | |
| done_text = gr.Textbox(label="Done", value="Not Done", visible=False) | |
| submit_button.click(update_component, outputs=done_text) | |
| for txt in [il, ilce, mahalle, sokak, apartman, no, ad_soyad, dis_kapi_no]: | |
| submit_button.click(fn=clear_textbox, inputs=txt, outputs=txt) | |
| if __name__ == "__main__": | |
| demo.launch() |