added input file type filter

app.py

@@ -80,10 +80,16 @@ def runner(in_pdb, model_type):
 
 
 with gr.Blocks() as app:
-    gr.Markdown("# cg2all: conversion of coarse-grained protein structure model to all-atom structure")
+    gr.Markdown(
+        "# cg2all: conversion of coarse-grained protein structure model to all-atom structure"
+    )
     with gr.Row():
         with gr.Column():
-            input_pdb = gr.File(file_count="single", label="Input CG structure")
+            input_pdb = gr.File(
+                file_count="single",
+                label="Input CG structure",
+                file_types=[".pdb", ".PDB", ".txt", ".TXT"],
+            )
             model_type = gr.Radio(
                 [
                     "CalphaBasedModel",
@@ -141,7 +147,9 @@ with gr.Blocks() as app:
         button.click(fn=runner, inputs=[input_pdb, model_type], outputs=[output_pdb, viewer])
     #
     gr.Markdown("---")
-    gr.Markdown("### GitHub repository: [https://github.com/huhlim/cg2all](https://github.com/huhlim/cg2all)")
+    gr.Markdown(
+        "### GitHub repository: [https://github.com/huhlim/cg2all](https://github.com/huhlim/cg2all)"
+    )
     gr.Markdown("### Local installation: `pip install git+http://github.com/huhlim/cg2all`")
     gr.Markdown("### Supported coarse-grained models")
     gr.Markdown("- CalphaBasedModel: CA-trace")