HenriqueBraz commited on
Commit
c68e191
·
verified ·
1 Parent(s): 14452d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -1
app.py CHANGED
@@ -653,7 +653,33 @@ def render_use_cases(models):
653
  ):
654
  handle_use_case_demo(models, key, case)
655
  else:
656
- st.warning("Demo requires specific file upload")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
657
 
658
  st.divider()
659
 
 
653
  ):
654
  handle_use_case_demo(models, key, case)
655
  else:
656
+ # Automatic uploader depending on case type
657
+ if "image" in key or "visual" in key:
658
+ file = st.file_uploader("📸 Upload an image", type=["jpg", "jpeg", "png"], key=f"upload_{key}")
659
+ if file:
660
+ case["demo_input"] = file
661
+ handle_use_case_demo(models, key, case)
662
+
663
+ elif "speech" in key or "audio" in key or "call" in key:
664
+ file = st.file_uploader("🎙️ Upload an audio file", type=["wav", "mp3", "m4a"], key=f"upload_{key}")
665
+ if file:
666
+ case["demo_input"] = file
667
+ handle_use_case_demo(models, key, case)
668
+
669
+ elif "medical" in key:
670
+ file = st.file_uploader("🏥 Upload a medical image (PNG/JPG/DICOM)", type=["png", "jpg", "jpeg", "dcm"], key=f"upload_{key}")
671
+ if file:
672
+ case["demo_input"] = file
673
+ handle_use_case_demo(models, key, case)
674
+
675
+ elif "document" in key or "summarization" in key or "extraction" in key:
676
+ file = st.file_uploader("📄 Upload a document", type=["pdf", "docx"], key=f"upload_{key}")
677
+ if file:
678
+ case["demo_input"] = file
679
+ handle_use_case_demo(models, key, case)
680
+
681
+ else:
682
+ st.warning("⚠️ This use case requires a specific file input")
683
 
684
  st.divider()
685