ai: Allowed ext files for documents scanning.

app.py

@@ -33,7 +33,7 @@ from PIL import Image
 LINUX_SERVER_HOSTS = [host for host in json.loads(os.getenv("LINUX_SERVER_HOST", "[]")) if host]
 LINUX_SERVER_PROVIDER_KEYS = [key for key in json.loads(os.getenv("LINUX_SERVER_PROVIDER_KEY", "[]")) if key]
 
-AI_TYPES = {f"AI_TYPE_{i}": os.getenv(f"AI_TYPE_{i}") for i in range(1, 6)}
+AI_TYPES = {f"AI_TYPE_{i}": os.getenv(f"AI_TYPE_{i}") for i in range(1, 7)}
 RESPONSES = {f"RESPONSE_{i}": os.getenv(f"RESPONSE_{i}") for i in range(1, 10)}
 
 MODEL_MAPPING = json.loads(os.getenv("MODEL_MAPPING", "{}"))
@@ -43,6 +43,8 @@ DEFAULT_CONFIG = json.loads(os.getenv("DEFAULT_CONFIG", "{}"))
 
 META_TAGS = os.getenv("META_TAGS")
 
+ALLOWED_EXTENSIONS = json.loads(os.getenv("ALLOWED_EXTENSIONS"))
+
 stop_event = threading.Event()
 session = requests.Session()
 
@@ -206,7 +208,14 @@ def stop_response():
 def change_model(new_model_display):
     return [], new_model_display
 
+def file_type_validation(file):
+    if file:
+        ext = Path(file.name).suffix.lower()
+        return ext in ALLOWED_EXTENSIONS
+    return False
+
 def check_send_button_enabled(msg, file):
+    is_file_valid = file_type_validation(file)
     return gr.update(interactive=bool(msg.strip()) or bool(file))
 
 with gr.Blocks(fill_height=True, fill_width=True, title=AI_TYPES["AI_TYPE_4"], head=META_TAGS) as demo:
@@ -215,12 +224,12 @@ with gr.Blocks(fill_height=True, fill_width=True, title=AI_TYPES["AI_TYPE_4"], h
 
     chatbot = gr.Chatbot(label=AI_TYPES["AI_TYPE_1"], show_copy_button=True, show_share_button=False, scale=1, elem_id=AI_TYPES["AI_TYPE_2"])
     model_dropdown = gr.Dropdown(label=AI_TYPES["AI_TYPE_3"], show_label=False, choices=MODEL_CHOICES, value=MODEL_CHOICES[0], interactive=True)
-    msg = gr.Textbox(label=RESPONSES["RESPONSE_4"], show_label=False, placeholder=RESPONSES["RESPONSE_5"])
+    msg = gr.Textbox(label=RESPONSES["RESPONSE_4"], show_label=False, placeholder=RESPONSES["RESPONSE_5"], interactive=True)
     send_btn = gr.Button(RESPONSES["RESPONSE_6"], visible=True, interactive=False)
     stop_btn = gr.Button(RESPONSES["RESPONSE_7"], variant=RESPONSES["RESPONSE_9"], visible=False)
 
-    with gr.Accordion("See more...", open=False):
-        file_upload = gr.File(label=AI_TYPES["AI_TYPE_5"], height=100, file_count="single", type="filepath")
+    with gr.Accordion(AI_TYPES["AI_TYPE_6"], open=False):
+        file_upload = gr.File(label=AI_TYPES["AI_TYPE_5"], file_count="single", type="filepath", file_types=ALLOWED_EXTENSIONS)
 
     model_dropdown.change(fn=change_model, inputs=[model_dropdown], outputs=[user_history, selected_model])
     send_btn.click(respond, inputs=[msg, file_upload, user_history, selected_model], outputs=[chatbot, msg, send_btn, stop_btn])