update ui

app.py

@@ -5,32 +5,26 @@ import io
 import json
 import csv
 import openai
+import ast
+import os
 
 
-
-openai.api_key = os.getenv("API_KEY")
+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]
 
-    adres_full = f"{mahalle}, {il}, {sokak}, {apartman}"
     with open("adress_book.csv", "a", encoding="utf-8") as f:
-        book = f.read()
-        if adres_full not in book:
-            with open("adress_book.csv", "a", encoding="utf-8") as f:
-                write = csv.writer(f)
-                write.writerow(adres_full)
-        else:
-            adres_full = "Bu adres daha önce raporlanmış."
-        
-        return adres_full
+        write = csv.writer(f)
+        write.writerow(adres_full)
+    return adres_full
 
 
 def get_json(mahalle, il, sokak, apartman):
@@ -39,64 +33,131 @@ def get_json(mahalle, il, sokak, apartman):
     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"""Address Extractor
-    You are a highly intelligent and accurate address  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
-    Examples:
+    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
 
-    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: 'Sehir:İstanbul', 'Ilce:Beşiktaş', 'Mahalle:Yıldız Mahallesi', 'Cadde: Cumhuriyet Caddesi', 'Apartman:no:35', 'Telefon: 5551231256', 'isim:Ahmet Yılmaz'
+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
 
-    Input: {ocr_input}
-    Output:"""
+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'}}' }}]
 
-    response = openai.Completion.create(
-    model="text-davinci-003",
-    prompt=prompt,
-    temperature=0,
-    max_tokens=100,
-    top_p=1,
-    frequency_penalty=0.0,
-    presence_penalty=0.0,
-    stop=["\n"],
-)
 
-    return response["choices"][0]['text']
+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 Raporlama")
-    gr.Markdown("Bu aplikasyonda üzerinde adres olan bir görüntüyü sürükleyip bırakarak konumu afet koordinasyona raporlayabilirsiniz.")
+    gr.Markdown(""" # Image to Text - Adres""")
     with gr.Row():
         img_area = gr.Image()
-        ocr_result = gr.Textbox(label="Okunan Metin")
-        open_api_text = gr.Textbox(label="Ayrılmış Çıktı")
-    
-    submit_button = gr.Button()
-    submit_button.click(get_text, img_area, ocr_result)
-
-    ocr_result.change(openai_response, ocr_result, open_api_text)
+        ocr_result = gr.Textbox(label="OCR")
+    open_api_text = gr.Textbox(label="OPENAI")
 
     with gr.Column():
         with gr.Row():
-            mahalle = gr.Textbox(label="Mahalle")
-            sokak = gr.Textbox(label="Sokak")
+            il = gr.Textbox(label="il")
+            ilce = gr.Textbox(label="ilce")
+        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="tel")
         with gr.Row():
-            apartman = gr.Textbox(label="Apartman")
-            il = gr.Textbox(label="İl")
-        tarif = gr.Textbox(label="Tarif")
+            isim_soyisim = gr.Textbox(label="isim_soyisim")
+            adres = gr.Textbox(label="adres")
 
-    json_out = gr.Textbox("Ayrılmış Adres")
-    csv_out = gr.Textbox("Adres Kaydetme")
+    submit_button = gr.Button()
+    submit_button.click(get_text, img_area, ocr_result)
+
+    ocr_result.change(openai_response, ocr_result, open_api_text)
 
-    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)
+    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)
 
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()