app.py

app.py

@@ -15,14 +15,18 @@ theme = gr.themes.Base(
     block_title_text_color="*neutral_950",
     block_title_text_size="text_xl",
     panel_border_color="*primary_950",
-    form_gap_width="2px"
+    form_gap_width="2px",
     
-
-
 )
 
 # Daten laden
 directory_path = load_dataset("HuberDa/OCTTestdaten")
+disease_1 = "Alterbedinge Makulardegeneration (AMD)"
+disease_2 = "Diabetische Netzhauterkrankung (DR)"
+disease_3 = "Drusen"
+disease_4 = "Chorodiale Neovaskularisation (CNV)"
+disease_5 = "Diabetisches Makulaödem (DME)"
+disease_6 = "Normal / Keine Krankheit"
 
 # Bilder generieren
 def image_generator(hf_dataset, split="train"):
@@ -60,7 +64,7 @@ def start():
         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")
+    gr.Markdown("# OCTis Training Center")
        
     with gr.Row():
         # Bildanzeige
@@ -75,15 +79,15 @@ with gr.Blocks(fill_height=True, fill_width=True, theme = theme) as demo:
        
     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")
+        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")
+        disease4_btn = gr.Button(value=disease_4,variant="primary")
+        disease5_btn = gr.Button(value=disease_5,variant="primary")
+        nodisease_btn = gr.Button(value=disease_6,variant="primary")
 
     with gr.Row():
         # Start und skip Button
@@ -95,12 +99,12 @@ with gr.Blocks(fill_height=True, fill_width=True, theme = theme) as demo:
     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)
+    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(disease_6, 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])
@@ -110,6 +114,5 @@ with gr.Blocks(fill_height=True, fill_width=True, theme = theme) as demo:
     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])
 
-gr.Interface().launch(debug=false, auth=("USERNAME", "PASSWORD"))
 demo.queue()
 demo.launch(share=True, debug=True)