KaiserShultz commited on
Commit
41be2fe
·
verified ·
1 Parent(s): 6b5e0d9

Update src/tools/tools.py

Browse files
Files changed (1) hide show
  1. src/tools/tools.py +6 -0
src/tools/tools.py CHANGED
@@ -107,6 +107,12 @@ def preprocess_files(files: List[str]) -> Dict[str, Dict[str, Any]]:
107
  elif file_ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']:
108
  info["type"] = "image"
109
  info["suggested_tool"] = "vision_qa_gemma"
 
 
 
 
 
 
110
  else:
111
  info["type"] = "unknown"
112
  info["suggested_tool"] = "analyze_txt_file (fallback)"
 
107
  elif file_ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']:
108
  info["type"] = "image"
109
  info["suggested_tool"] = "vision_qa_gemma"
110
+ elif file_ext in [".mp3"]:
111
+ info["type"] = "audio"
112
+ info["suggested_tool"] = "transcribe_audio"
113
+ elif file_ext in [".mp4"]:
114
+ info["type"] = "video"
115
+ info["suggested_tool"] = "video_qa_gemma"
116
  else:
117
  info["type"] = "unknown"
118
  info["suggested_tool"] = "analyze_txt_file (fallback)"