Update app.py

app.py

@@ -78,7 +78,13 @@ with gr.Blocks() as iface:
     with gr.Tab("CIF-SLICES Conversion"):
         with gr.Row():
             with gr.Column():
-                cif_input = gr.File(label="Upload CIF file", file_types=[".cif"], value="NdSiRu.cif")
+                file_choice = gr.Radio(
+                    ["Use example CIF (NdSiRu.cif)", "Upload custom CIF"],
+                    label="Choose CIF source",
+                    value="Use example CIF (NdSiRu.cif)"
+                )
+                example_file = gr.File(value="NdSiRu.cif", visible=False, interactive=False)
+                custom_file = gr.File(label="Upload CIF file", file_types=[".cif"], visible=False)
                 convert_cif_button = gr.Button("Convert CIF to SLICES")
                 slices_input = gr.Textbox(label="Enter SLICES String")
                 convert_slices_button = gr.Button("Convert SLICES to CIF")
@@ -96,23 +102,36 @@ with gr.Blocks() as iface:
         augment_button = gr.Button("Augment and Canonicalize")
         aug_slices_output = gr.Textbox(label="Augmented SLICES Strings")
         canon_slices_output = gr.Textbox(label="Canonical SLICES Strings")
-    
+
     # Event handlers
+    def update_file_visibility(choice):
+        return gr.update(visible=choice == "Use example CIF (NdSiRu.cif)"), gr.update(visible=choice == "Upload custom CIF")
+
+    file_choice.change(
+        update_file_visibility,
+        inputs=[file_choice],
+        outputs=[example_file, custom_file]
+    )
+
+    def get_active_file(choice, example, custom):
+        return example if choice == "Use example CIF (NdSiRu.cif)" else custom
+
     convert_cif_button.click(
-        cif_to_slices,
-        inputs=[cif_input],
+        lambda choice, example, custom: cif_to_slices(get_active_file(choice, example, custom)),
+        inputs=[file_choice, example_file, custom_file],
         outputs=[slices_output, cif_image, conversion_status, slices_input, aug_slices_input]
     )
+
     convert_slices_button.click(
         slices_to_cif,
         inputs=[slices_input],
         outputs=[cif_output, slices_image, conversion_status]
     )
+
     augment_button.click(
         augment_and_canonicalize_slices,
         inputs=[aug_slices_input, num_augmentations],
         outputs=[aug_slices_output, canon_slices_output]
     )
 
-# Launch the interface
 iface.launch(share=True)