Return textboxes on file upload errors (#214)
This commit is contained in:
parent
fdf4fe49ba
commit
a4ec1e5be3
|
@ -5,7 +5,7 @@ from smolagents import (
|
|||
)
|
||||
|
||||
agent = CodeAgent(
|
||||
tools=[], model=HfApiModel(), max_steps=4, verbosity_level=0
|
||||
tools=[], model=HfApiModel(), max_steps=4, verbosity_level=1
|
||||
)
|
||||
|
||||
GradioUI(agent, file_upload_folder='./data').launch()
|
||||
|
|
|
@ -120,15 +120,21 @@ class GradioUI:
|
|||
"""
|
||||
|
||||
if file is None:
|
||||
return "No file uploaded"
|
||||
return gr.Textbox(
|
||||
"No file uploaded", visible=True
|
||||
), file_uploads_log
|
||||
|
||||
try:
|
||||
mime_type, _ = mimetypes.guess_type(file.name)
|
||||
except Exception as e:
|
||||
return f"Error: {e}"
|
||||
return gr.Textbox(
|
||||
f"Error: {e}", visible=True
|
||||
), file_uploads_log
|
||||
|
||||
if mime_type not in allowed_file_types:
|
||||
return "File type disallowed"
|
||||
return gr.Textbox(
|
||||
"File type disallowed", visible=True
|
||||
), file_uploads_log
|
||||
|
||||
# Sanitize file name
|
||||
original_name = os.path.basename(file.name)
|
||||
|
@ -181,7 +187,7 @@ class GradioUI:
|
|||
)
|
||||
# If an upload folder is provided, enable the upload feature
|
||||
if self.file_upload_folder is not None:
|
||||
upload_file = gr.File(label="Upload a file", height=1)
|
||||
upload_file = gr.File(label="Upload a file")
|
||||
upload_status = gr.Textbox(
|
||||
label="Upload Status", interactive=False, visible=False
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue