




# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from typing import TYPE_CHECKING

from ..utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_torch_available,
)


a_      						=   {
    "agents": ["Agent", "AzureOpenAiAgent", "HfAgent", "LocalAgent", "OpenAiAgent"],
    "base": ["PipelineTool", "RemoteTool", "Tool", "launch_gradio_demo", "load_tool"],
}

try:
 if not is_torch_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 a_      						=   ["DocumentQuestionAnsweringTool"]
 a_      						=   ["ImageCaptioningTool"]
 a_      						=   ["ImageQuestionAnsweringTool"]
 a_      						=   ["ImageSegmentationTool"]
 a_      						=   ["SpeechToTextTool"]
 a_      						=   ["TextClassificationTool"]
 a_      						=   ["TextQuestionAnsweringTool"]
 a_      						=   ["TextSummarizationTool"]
 a_      						=   ["TextToSpeechTool"]
 a_      						=   ["TranslationTool"]

if TYPE_CHECKING:
 from .agents import Agent, AzureOpenAiAgent, HfAgent, LocalAgent, OpenAiAgent
 from .base import PipelineTool, RemoteTool, Tool, launch_gradio_demo, load_tool

 try:
  if not is_torch_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .document_question_answering import DocumentQuestionAnsweringTool
  from .image_captioning import ImageCaptioningTool
  from .image_question_answering import ImageQuestionAnsweringTool
  from .image_segmentation import ImageSegmentationTool
  from .speech_to_text import SpeechToTextTool
  from .text_classification import TextClassificationTool
  from .text_question_answering import TextQuestionAnsweringTool
  from .text_summarization import TextSummarizationTool
  from .text_to_speech import TextToSpeechTool
  from .translation import TranslationTool
else:
 import sys

 a_      						=   _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)












from typing import List

import datasets
from datasets.tasks import ImageClassification

from ..folder_based_builder import folder_based_builder


_lowerCamelCase			 =						datasets.utils.logging.get_logger(__name__)



class 			a     (   folder_based_builder.FolderBasedBuilderConfig			):

     '''simple docstring'''





     lowerCAmelCase				:       bool   		=  None
     lowerCAmelCase				:       bool   		=  None



class 			a     (   folder_based_builder.FolderBasedBuilder			):

     '''simple docstring'''





     lowerCAmelCase				:       List[Any]   		=  datasets.Image()
     lowerCAmelCase				:       List[Any]   		=  'image'
     lowerCAmelCase				:       List[Any]   		=  ImageFolderConfig
     lowerCAmelCase				:       List[str]  # definition at the bottom of the script
     lowerCAmelCase				:       Dict   		=  ImageClassification(image_column='image'	,   label_column='label'			)


_lowerCamelCase			 =						[
    '.blp',
    '.bmp',
    '.dib',
    '.bufr',
    '.cur',
    '.pcx',
    '.dcx',
    '.dds',
    '.ps',
    '.eps',
    '.fit',
    '.fits',
    '.fli',
    '.flc',
    '.ftc',
    '.ftu',
    '.gbr',
    '.gif',
    '.grib',
    '.h5',
    '.hdf',
    '.png',
    '.apng',
    '.jp2',
    '.j2k',
    '.jpc',
    '.jpf',
    '.jpx',
    '.j2c',
    '.icns',
    '.ico',
    '.im',
    '.iim',
    '.tif',
    '.tiff',
    '.jfif',
    '.jpe',
    '.jpg',
    '.jpeg',
    '.mpg',
    '.mpeg',
    '.msp',
    '.pcd',
    '.pxr',
    '.pbm',
    '.pgm',
    '.ppm',
    '.pnm',
    '.psd',
    '.bw',
    '.rgb',
    '.rgba',
    '.sgi',
    '.ras',
    '.tga',
    '.icb',
    '.vda',
    '.vst',
    '.webp',
    '.wmf',
    '.emf',
    '.xbm',
    '.xpm',
]
_lowerCamelCase			 =						IMAGE_EXTENSIONS





