nazdridoy commited on
Commit
c663c5b
·
verified ·
1 Parent(s): dd78fc1

refactor(provider): unify default provider constant

Browse files

- [refactor] Remove `DEFAULT_IMAGE_PROVIDER`, `DEFAULT_IMAGE_TO_IMAGE_PROVIDER`, `DEFAULT_TTS_PROVIDER` constants (utils.py:14-16)
- [refactor] Add `DEFAULT_PROVIDER` constant set to "auto" (utils.py:19)
- [docs] Add comment for `DEFAULT_PROVIDER` (utils.py:20)
- [refactor] Update imports for provider constants (ui_components.py:6-9)
- [update] Set `chat_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:52)
- [update] Set `img_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:194)
- [update] Set `img2img_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:314)
- [update] Set `tts_provider` dropdown value to `DEFAULT_PROVIDER` (ui_components.py:455)
- [docs] Remove "(default for images)" from `fal-ai` description in footer (ui_components.py:691)

Files changed (2) hide show
  1. ui_components.py +8 -8
  2. utils.py +3 -3
ui_components.py CHANGED
@@ -5,9 +5,9 @@ Contains functions to create different sections of the Gradio interface.
5
 
6
  import gradio as gr
7
  from utils import (
8
- DEFAULT_CHAT_MODEL, DEFAULT_IMAGE_MODEL, DEFAULT_IMAGE_PROVIDER,
9
- DEFAULT_IMAGE_TO_IMAGE_MODEL, DEFAULT_IMAGE_TO_IMAGE_PROVIDER,
10
- DEFAULT_TTS_MODEL, DEFAULT_TTS_PROVIDER,
11
  CHAT_CONFIG, IMAGE_CONFIG, IMAGE_PROVIDERS, IMAGE_MODEL_PRESETS,
12
  IMAGE_TO_IMAGE_MODEL_PRESETS, TTS_MODEL_PRESETS, TTS_VOICES, TTS_MODEL_CONFIGS,
13
  IMAGE_EXAMPLE_PROMPTS, IMAGE_TO_IMAGE_EXAMPLE_PROMPTS, TTS_EXAMPLE_TEXTS, TTS_EXAMPLE_AUDIO_URLS
@@ -49,7 +49,7 @@ def create_chat_tab(handle_chat_submit_fn, handle_chat_retry_fn=None):
49
  )
50
  chat_provider = gr.Dropdown(
51
  choices=IMAGE_PROVIDERS,
52
- value="auto",
53
  label="Provider",
54
  interactive=True
55
  )
@@ -191,7 +191,7 @@ def create_image_tab(handle_image_generation_fn):
191
  )
192
  img_provider = gr.Dropdown(
193
  choices=IMAGE_PROVIDERS,
194
- value=DEFAULT_IMAGE_PROVIDER,
195
  label="Provider",
196
  interactive=True
197
  )
@@ -311,7 +311,7 @@ def create_image_to_image_tab(handle_image_to_image_generation_fn):
311
  )
312
  img2img_provider = gr.Dropdown(
313
  choices=IMAGE_PROVIDERS,
314
- value=DEFAULT_IMAGE_TO_IMAGE_PROVIDER,
315
  label="Provider",
316
  interactive=True
317
  )
@@ -452,7 +452,7 @@ def create_tts_tab(handle_tts_generation_fn):
452
  )
453
  tts_provider = gr.Dropdown(
454
  choices=IMAGE_PROVIDERS,
455
- value=DEFAULT_TTS_PROVIDER,
456
  label="Provider",
457
  interactive=True
458
  )
@@ -687,7 +687,7 @@ def create_footer():
687
  - Supports both fal-ai and replicate providers
688
 
689
  **Supported Providers:**
690
- - **fal-ai**: High-quality image generation (default for images)
691
  - **hf-inference**: Core API with comprehensive model support
692
  - **cerebras**: High-performance inference
693
  - **cohere**: Advanced language models with multilingual support
 
5
 
6
  import gradio as gr
7
  from utils import (
8
+ DEFAULT_CHAT_MODEL, DEFAULT_IMAGE_MODEL, DEFAULT_PROVIDER,
9
+ DEFAULT_IMAGE_TO_IMAGE_MODEL,
10
+ DEFAULT_TTS_MODEL,
11
  CHAT_CONFIG, IMAGE_CONFIG, IMAGE_PROVIDERS, IMAGE_MODEL_PRESETS,
12
  IMAGE_TO_IMAGE_MODEL_PRESETS, TTS_MODEL_PRESETS, TTS_VOICES, TTS_MODEL_CONFIGS,
13
  IMAGE_EXAMPLE_PROMPTS, IMAGE_TO_IMAGE_EXAMPLE_PROMPTS, TTS_EXAMPLE_TEXTS, TTS_EXAMPLE_AUDIO_URLS
 
49
  )
50
  chat_provider = gr.Dropdown(
51
  choices=IMAGE_PROVIDERS,
52
+ value=DEFAULT_PROVIDER,
53
  label="Provider",
54
  interactive=True
55
  )
 
191
  )
192
  img_provider = gr.Dropdown(
193
  choices=IMAGE_PROVIDERS,
194
+ value=DEFAULT_PROVIDER,
195
  label="Provider",
196
  interactive=True
197
  )
 
311
  )
312
  img2img_provider = gr.Dropdown(
313
  choices=IMAGE_PROVIDERS,
314
+ value=DEFAULT_PROVIDER,
315
  label="Provider",
316
  interactive=True
317
  )
 
452
  )
453
  tts_provider = gr.Dropdown(
454
  choices=IMAGE_PROVIDERS,
455
+ value=DEFAULT_PROVIDER,
456
  label="Provider",
457
  interactive=True
458
  )
 
687
  - Supports both fal-ai and replicate providers
688
 
689
  **Supported Providers:**
690
+ - **fal-ai**: High-quality image generation
691
  - **hf-inference**: Core API with comprehensive model support
692
  - **cerebras**: High-performance inference
693
  - **cohere**: Advanced language models with multilingual support
utils.py CHANGED
@@ -11,11 +11,11 @@ import requests
11
  # Configuration constants
12
  DEFAULT_CHAT_MODEL = "openai/gpt-oss-20b"
13
  DEFAULT_IMAGE_MODEL = "Qwen/Qwen-Image"
14
- DEFAULT_IMAGE_PROVIDER = "fal-ai"
15
  DEFAULT_IMAGE_TO_IMAGE_MODEL = "Qwen/Qwen-Image-Edit"
16
- DEFAULT_IMAGE_TO_IMAGE_PROVIDER = "fal-ai"
17
  DEFAULT_TTS_MODEL = "hexgrad/Kokoro-82M"
18
- DEFAULT_TTS_PROVIDER = "fal-ai"
 
 
19
 
20
  # Chat configuration
21
  CHAT_CONFIG = {
 
11
  # Configuration constants
12
  DEFAULT_CHAT_MODEL = "openai/gpt-oss-20b"
13
  DEFAULT_IMAGE_MODEL = "Qwen/Qwen-Image"
 
14
  DEFAULT_IMAGE_TO_IMAGE_MODEL = "Qwen/Qwen-Image-Edit"
 
15
  DEFAULT_TTS_MODEL = "hexgrad/Kokoro-82M"
16
+
17
+ # Unified default provider used by all non-chat tasks
18
+ DEFAULT_PROVIDER = "auto"
19
 
20
  # Chat configuration
21
  CHAT_CONFIG = {