app.py

app.py

@@ -0,0 +1,73 @@
+import numpy as np
+import pandas as pd
+import gradio as gr
+import time
+import os
+import tkinter as tk
+from PIL import Image
+import json
+
+
+def image_generator(directory_path):
+    for filename in os.listdir(directory_path):
+        if filename.endswith(".jpg") or filename.endswith(".png"):
+            image_path = os.path.join(directory_path, filename)
+            image = Image.open(image_path)
+            image_array = np.array(image)
+            yield filename,image_array
+
+directory_path = "/content/drive/MyDrive/Testdateien"
+
+# Create the generator
+image_gen = image_generator(directory_path)
+
+def record_input(fname:str, label:str):
+  if label != "Pass":
+    output = {"fname": fname, "label": label}
+    with open("/content/output.txt", 'a') as f:
+      json.dump(output, f)
+      f.write("\n")
+  fname, image = next(image_gen)
+  return fname, image
+
+def increment():
+  global count # Access the global variable 'count'
+  count += 1  # Increment 'count' by 1
+  return f"{count}" # Return the updated value of 'count' as a string
+
+def start():
+  fname, image = next(image_gen)
+  return fname, image
+
+with gr.Blocks(fill_height=True, fill_width=True) as demo:
+  gr.Markdown("#DiseaseDetection")
+  output = gr.Number()
+  with gr.Row():
+    img_block = gr.Image(visible = True, width = 500, height = 500, show_fullscreen_button = True, show_share_button = False, show_download_button=False, show_label=False)
+    file_name = gr.Textbox(info="File name", visible = False)
+  with gr.Row():
+    disease1_btn = gr.Button(value = "disease1", interactive=True, visible = True)
+    disease2_btn = gr.Button(value = "disease2",interactive=True, visible = True)
+    disease3_btn = gr.Button(value = "disease3",interactive=True, visible = True)
+  with gr.Row():
+    disease4_btn = gr.Button(value = "disease4",interactive=True, visible = True)
+    disease5_btn = gr.Button(value = "disease5",interactive=True, visible = True)
+    nodisease_btn = gr.Button(value = "no disease",interactive=True, visible = True)
+  with gr.Row():
+    start_btn = gr.Button("Start")
+    disease1_btn.click(fn = record_input, inputs = [file_name, disease1_btn], outputs=[file_name,img_block])
+    disease2_btn.click(fn = record_input, inputs = [file_name, disease2_btn], outputs=[file_name,img_block])
+    disease3_btn.click(fn = record_input, inputs = [file_name, disease3_btn], outputs=[file_name,img_block])
+    disease4_btn.click(fn = record_input, inputs = [file_name, disease4_btn], outputs=[file_name,img_block])
+    disease5_btn.click(fn = record_input, inputs = [file_name, disease5_btn], outputs=[file_name,img_block])
+    nodisease_btn.click(fn = record_input, inputs = [file_name, nodisease_btn], outputs=[file_name,img_block])
+    start_btn.click(fn = start, outputs = [file_name,img_block])
+    disease1_btn.click(fn=increment, outputs=output)
+    disease2_btn.click(fn=increment, outputs=output)
+    disease3_btn.click(fn=increment, outputs=output)
+    disease4_btn.click(fn=increment, outputs=output)
+    disease5_btn.click(fn=increment, outputs=output)
+    nodisease_btn.click(fn=increment, outputs=output)
+
+demo.queue()
+demo.launch(share=True, debug=True)