app.py

app.py

@@ -1,85 +1,115 @@
 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
 from datasets import load_dataset
 
+
+theme = gr.themes.Base(
+    secondary_hue="cyan",
+).set(
+    button_primary_background_fill='*primary_100',
+    button_primary_text_color='*neutral_950',
+    border_color_accent="*primary_100",
+    body_text_color="*neutral_950",
+    block_border_color="*primary_100",
+    block_title_text_color="*neutral_950",
+    block_title_text_size="text_xl",
+    panel_border_color="*primary_950",
+    form_gap_width="2px"
+    
+
+
+)
+
+# Daten laden
 directory_path = load_dataset("HuberDa/Labeltest")
-print("Spaltennamen:", directory_path['train'].column_names)
-print("Erstes Beispiel im Train-Split:", directory_path['train'][0])
-count = 0   
 
+# Bilder generieren
 def image_generator(hf_dataset, split="train"):
-    image_column = None
-    for col in hf_dataset[split].column_names:
-        if isinstance(hf_dataset[split][0][col], np.ndarray) or isinstance(hf_dataset[split][0][col], Image.Image):
-            image_column = col
-            break
-           
-    if image_column is None:
-        raise ValueError("Keine Bildspalte gefunden!")
-
-    # Schleife durch das Dataset und generiere Bild-Arrays
-    for example in hf_dataset[split]:
-        image = example[image_column]  # Greife auf das Bild zu
-        image_array = np.array(image)  # Konvertiere es in ein NumPy-Array
-        filename = example['id'] if 'id' in example else "unknown"
-        yield filename, image_array
-
-# Create the generator
+    for index, example in enumerate(hf_dataset[split]):
+        image = example['image']  # Access the image column
+        image_array = np.array(image)  # Convert to a NumPy array
+        yield index, image_array  # Yield the index and image
+
+# Generator aktivieren
 image_gen = image_generator(directory_path)
 
-def record_input(fname:str, label:str):
-  if label != "Pass":
-    output = {"fname": fname, "label": label}
-    with open("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
+# Input der Bilder sammeln
+def record_input(label: str, index: int, free_text: str):
+    # Den Output mit index, label, and free text sammeln
+    output = {
+        "index": index,
+        "label": label,
+        "free_text": free_text
+    }
+    # Output notieren
+    with open("output.txt", 'a') as f:
+        json.dump(output, f)
+        f.write("\n")
+    return f"Bild {index} wurde als {label} klassifiziert. \nZusätzliche Notizen: {free_text}"
 
 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)
+    try:
+        # Nächstes Bild und Index holen
+        index, image = next(image_gen)  # Index und Bild entpacken
+        return str(index), image, ""  # Rückgabe von index, image, and leeren der Textbox
+    except StopIteration:
+        return "No more images", None, ""  # Falls keine Bilder übrig
+    except Exception as e:
+        return str(e), None, ""  # Für andere Probleme
+
+with gr.Blocks(fill_height=True, fill_width=True, theme = theme) as demo:
+    gr.Markdown("OCTis Training Center")
+       
+    with gr.Row():
+        # Bildanzeige
+        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)
+
+    with gr.Row():
+        output_message = gr.Textbox(label="Ergebnis", interactive=False)
+        free_text_input = gr.Textbox(label="zusätzliche Informationen", placeholder="Hier Text eingeben...", lines=2)  
+        
+        # versteckte textbox für index
+        index_box = gr.Textbox(label="Index", visible=False)
+       
+    with gr.Row():
+        # Buttons 1 - 3 
+        disease1_btn = gr.Button(value="Disease 1",variant="primary")
+        disease2_btn = gr.Button(value="Disease 2",variant="primary")
+        disease3_btn = gr.Button(value="Disease 3",variant="primary")
+        
+    with gr.Row():
+        # Button 4 - 6
+        disease4_btn = gr.Button(value="Disease 4",variant="primary")
+        disease5_btn = gr.Button(value="Disease 5",variant="primary")
+        nodisease_btn = gr.Button(value="No Disease",variant="primary")
+
+    with gr.Row():
+        # Start und skip Button
+        start_btn = gr.Button("Start")
+        skip_btn = gr.Button("Überspringen")
+
+    # Starten der App
+    start_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+    skip_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+
+    # Labeling mit Buttons
+    disease1_btn.click(fn=lambda idx, free_text: record_input("Disease 1", idx, free_text), inputs=[index_box, free_text_input], outputs=output_message)
+    disease2_btn.click(fn=lambda idx, free_text: record_input("Disease 2", idx, free_text), inputs=[index_box, free_text_input], outputs=output_message)
+    disease3_btn.click(fn=lambda idx, free_text: record_input("Disease 3", idx, free_text), inputs=[index_box, free_text_input], outputs=output_message)
+    disease4_btn.click(fn=lambda idx, free_text: record_input("Disease 4", idx, free_text), inputs=[index_box, free_text_input], outputs=output_message)
+    disease5_btn.click(fn=lambda idx, free_text: record_input("Disease 5", idx, free_text), inputs=[index_box, free_text_input], outputs=output_message)
+    nodisease_btn.click(fn=lambda idx, free_text: record_input("No Disease", idx, free_text), inputs=[index_box, free_text_input], outputs=output_message)
+
+    # Nächstes Bild
+    disease1_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+    disease2_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+    disease3_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+    disease4_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+    disease5_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+    nodisease_btn.click(fn=start, outputs=[index_box, img_block, free_text_input])
+
 
 demo.queue()
 demo.launch(share=True, debug=True)