add file_types parameter to only allow pdfs, docs, & text files; make sure file count is single for visible uploadbutton elements

app.py

@@ -78,6 +78,8 @@ HEADER = """
 **Getting Started**: Provide a question and document or context given to your model in addition to the answer given by the model and then click submit. The output panel will indicate whether the reponse is a hallucination (Fail) or if it is faithful to the given document or context (Pass) through the score Pass or Fail and provide reasoning behind the score.
 """
 
+UPLOADABLE_FILE_TYPES = [".pdf", ".txt", ".docx", ".doc"]
+
 def update_client_base_url(model_name):
     if model_name == "Patronus Lynx 8B v1.1":
         return "https://yb15a7dy-patronus-lynx-8b-v1-1.tin.lepton.run/api/v1/"
@@ -138,10 +140,10 @@ def upload_file(filepath):
         print("FILEPATH type & file name type", type(filepath), type(name))
         return [gr.UploadButton(visible=False), gr.Group(visible=True), gr.Markdown("**Uploaded file:** {name}".format(name=name))]
     else:
-        return [gr.UploadButton(visible=True), gr.Group(visible=False), gr.Markdown("")]
+        return [gr.UploadButton(visible=True, file_count="single", file_types=UPLOADABLE_FILE_TYPES), gr.Group(visible=False), gr.Markdown("")]
     # return [gr.UploadButton(visible=False), gr.DownloadButton(label=f"Download {name}", value=filepath, visible=True)]
 def reset_buttons():
-    return [gr.UploadButton(visible=True), gr.Group(visible=False), gr.Markdown("")]
+    return [gr.UploadButton(visible=True, file_count="single", file_types=UPLOADABLE_FILE_TYPES), gr.Group(visible=False), gr.Markdown("")]
 
 # def download_file():
 #     return [gr.UploadButton(visible=True), gr.DownloadButton(visible=False)]
@@ -167,7 +169,7 @@ with gr.Blocks(css=css) as demo:
                 question = gr.Textbox(label="Question")
             with gr.Row():
                 document = gr.Textbox(label="Document", scale=9)
-                u = gr.UploadButton("Upload", file_count="single", scale=1)
+                u = gr.UploadButton("Upload", visible=True, file_count="single", file_types=UPLOADABLE_FILE_TYPES, scale=1)
                 file_group = gr.Group(visible=False)
                 with file_group:
                     file_name = gr.Markdown("")