npc0 commited on
Commit
3d47212
Β·
verified Β·
1 Parent(s): e392670

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -14
app.py CHANGED
@@ -210,7 +210,7 @@ def scan_uploaded_zip(zip_file, scan_type="vuln,secret,misconfig"):
210
  def create_ui():
211
  """Create Gradio UI"""
212
 
213
- with gr.Blocks(title="Trivy Security Scanner", theme=gr.themes.Soft()) as demo:
214
  gr.Markdown(
215
  """
216
  # πŸ”’ Trivy Security Scanner
@@ -229,16 +229,17 @@ def create_ui():
229
  with gr.Column(scale=2):
230
  docker_image = gr.Textbox(
231
  label="Docker Image",
232
- placeholder="e.g., python:3.4-alpine, nginx:latest, ubuntu:20.04",
233
- info="Enter a Docker image name and tag"
234
  )
 
235
  with gr.Column(scale=1):
236
  docker_scanners = gr.Dropdown(
237
- choices=["vuln", "secret", "misconfig", "vuln,secret", "vuln,secret,misconfig"],
238
  value="vuln",
239
  label="Scanners",
240
- info="Select scan types"
241
  )
 
242
 
243
  docker_button = gr.Button("πŸ” Scan Docker Image", variant="primary")
244
 
@@ -262,16 +263,17 @@ def create_ui():
262
  with gr.Column(scale=2):
263
  github_url = gr.Textbox(
264
  label="GitHub Repository URL",
265
- placeholder="e.g., https://github.com/username/repository",
266
- info="Enter a public GitHub repository URL"
267
  )
 
268
  with gr.Column(scale=1):
269
  github_scanners = gr.Dropdown(
270
- choices=["vuln", "secret", "misconfig", "vuln,secret", "vuln,secret,misconfig"],
271
  value="vuln,secret,misconfig",
272
  label="Scanners",
273
- info="Select scan types"
274
  )
 
275
 
276
  github_button = gr.Button("πŸ” Scan Repository", variant="primary")
277
 
@@ -294,16 +296,17 @@ def create_ui():
294
  zip_upload = gr.File(
295
  label="Upload ZIP File",
296
  file_types=[".zip"],
297
- type="filepath",
298
- info="Upload a ZIP file containing your project"
299
  )
 
300
  with gr.Column(scale=1):
301
  zip_scanners = gr.Dropdown(
302
- choices=["vuln", "secret", "misconfig", "vuln,secret", "vuln,secret,misconfig"],
303
  value="vuln,secret,misconfig",
304
  label="Scanners",
305
- info="Select scan types"
306
  )
 
307
 
308
  zip_button = gr.Button("πŸ” Scan ZIP File", variant="primary")
309
 
@@ -357,4 +360,4 @@ if __name__ == "__main__":
357
  server_name="0.0.0.0",
358
  server_port=7860,
359
  share=False
360
- )
 
210
  def create_ui():
211
  """Create Gradio UI"""
212
 
213
+ with gr.Blocks(title="Trivy Security Scanner") as demo:
214
  gr.Markdown(
215
  """
216
  # πŸ”’ Trivy Security Scanner
 
229
  with gr.Column(scale=2):
230
  docker_image = gr.Textbox(
231
  label="Docker Image",
232
+ placeholder="e.g., python:3.4-alpine, nginx:latest, ubuntu:20.04"
 
233
  )
234
+ gr.Markdown("Enter a Docker image name and tag")
235
  with gr.Column(scale=1):
236
  docker_scanners = gr.Dropdown(
237
+ choices=["vuln", "secret", "misconfig", "vuln,secret", "vuln,secret,misconfig", "vuln,misconfig"],
238
  value="vuln",
239
  label="Scanners",
240
+ allow_custom_value=True
241
  )
242
+ gr.Markdown("Select scan types")
243
 
244
  docker_button = gr.Button("πŸ” Scan Docker Image", variant="primary")
245
 
 
263
  with gr.Column(scale=2):
264
  github_url = gr.Textbox(
265
  label="GitHub Repository URL",
266
+ placeholder="e.g., https://github.com/username/repository"
 
267
  )
268
+ gr.Markdown("Enter a public GitHub repository URL")
269
  with gr.Column(scale=1):
270
  github_scanners = gr.Dropdown(
271
+ choices=["vuln", "secret", "misconfig", "vuln,secret", "vuln,secret,misconfig", "vuln,misconfig"],
272
  value="vuln,secret,misconfig",
273
  label="Scanners",
274
+ allow_custom_value=True
275
  )
276
+ gr.Markdown("Select scan types")
277
 
278
  github_button = gr.Button("πŸ” Scan Repository", variant="primary")
279
 
 
296
  zip_upload = gr.File(
297
  label="Upload ZIP File",
298
  file_types=[".zip"],
299
+ type="filepath"
 
300
  )
301
+ gr.Markdown("Upload a ZIP file containing your project")
302
  with gr.Column(scale=1):
303
  zip_scanners = gr.Dropdown(
304
+ choices=["vuln", "secret", "misconfig", "vuln,secret", "vuln,secret,misconfig", "vuln,misconfig"],
305
  value="vuln,secret,misconfig",
306
  label="Scanners",
307
+ allow_custom_value=True
308
  )
309
+ gr.Markdown("Select scan types")
310
 
311
  zip_button = gr.Button("πŸ” Scan ZIP File", variant="primary")
312
 
 
360
  server_name="0.0.0.0",
361
  server_port=7860,
362
  share=False
363
+ )