nazdridoy commited on
Commit
52fc803
·
verified ·
1 Parent(s): c2e6d7e

feat(handlers): pass Hugging Face profile to handlers

Browse files

- [feat] Update `handle_chat_submit()` and `handle_chat_retry()` signatures to accept `hf_profile` and initialize `username` (chat_handler.py:168,178, 210,219)
- [feat] Update `handle_image_to_image_generation()` and `handle_image_generation()` signatures to accept `hf_profile` and initialize `username` (image_handler.py:279,289, 307,318)
- [feat] Update `handle_text_to_speech_generation()` signature to accept `hf_profile` and initialize `username` (tts_handler.py:155,169)
- [feat] Update `handle_video_generation()` signature to accept `hf_profile` and initialize `username` (video_handler.py:132,140)

Files changed (4) hide show
  1. chat_handler.py +4 -4
  2. image_handler.py +4 -4
  3. tts_handler.py +2 -2
  4. video_handler.py +2 -2
chat_handler.py CHANGED
@@ -168,7 +168,7 @@ def chat_respond(
168
  yield format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
169
 
170
 
171
- def handle_chat_submit(message, history, system_msg, model_name, provider, max_tokens, temperature, top_p, hf_token: gr.OAuthToken = None):
172
  """
173
  Handle chat submission and manage conversation history with streaming.
174
  """
@@ -178,7 +178,7 @@ def handle_chat_submit(message, history, system_msg, model_name, provider, max_t
178
 
179
  # Require sign-in: if no token present, prompt login
180
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
181
- username = None
182
  if not access_token:
183
  assistant_response = format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
184
  current_history = history + [{"role": "assistant", "content": assistant_response}]
@@ -210,14 +210,14 @@ def handle_chat_submit(message, history, system_msg, model_name, provider, max_t
210
  yield current_history, ""
211
 
212
 
213
- def handle_chat_retry(history, system_msg, model_name, provider, max_tokens, temperature, top_p, hf_token: gr.OAuthToken = None, retry_data=None):
214
  """
215
  Retry the assistant response for the selected message.
216
  Works with gr.Chatbot.retry() which provides retry_data.index for the message.
217
  """
218
  # Require sign-in: if no token present, prompt login
219
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
220
- username = None
221
  if not access_token:
222
  assistant_response = format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
223
  current_history = (history or []) + [{"role": "assistant", "content": assistant_response}]
 
168
  yield format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
169
 
170
 
171
+ def handle_chat_submit(message, history, system_msg, model_name, provider, max_tokens, temperature, top_p, hf_token: gr.OAuthToken = None, hf_profile: gr.OAuthProfile = None):
172
  """
173
  Handle chat submission and manage conversation history with streaming.
174
  """
 
178
 
179
  # Require sign-in: if no token present, prompt login
180
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
181
+ username = getattr(hf_profile, "username", None) if hf_profile is not None else None
182
  if not access_token:
183
  assistant_response = format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
184
  current_history = history + [{"role": "assistant", "content": assistant_response}]
 
210
  yield current_history, ""
211
 
212
 
213
+ def handle_chat_retry(history, system_msg, model_name, provider, max_tokens, temperature, top_p, hf_token: gr.OAuthToken = None, hf_profile: gr.OAuthProfile = None, retry_data=None):
214
  """
215
  Retry the assistant response for the selected message.
216
  Works with gr.Chatbot.retry() which provides retry_data.index for the message.
217
  """
218
  # Require sign-in: if no token present, prompt login
219
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
220
+ username = getattr(hf_profile, "username", None) if hf_profile is not None else None
221
  if not access_token:
222
  assistant_response = format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
223
  current_history = (history or []) + [{"role": "assistant", "content": assistant_response}]
image_handler.py CHANGED
@@ -279,7 +279,7 @@ def generate_image_to_image(
279
  return None, format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
280
 
281
 
282
- def handle_image_to_image_generation(input_image_val, prompt_val, model_val, provider_val, negative_prompt_val, steps_val, guidance_val, seed_val, hf_token: gr.OAuthToken = None):
283
  """
284
  Handle image-to-image generation request with validation.
285
  """
@@ -289,7 +289,7 @@ def handle_image_to_image_generation(input_image_val, prompt_val, model_val, pro
289
 
290
  # Require sign-in via HF OAuth token
291
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
292
- username = None
293
  if not access_token:
294
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
295
 
@@ -307,7 +307,7 @@ def handle_image_to_image_generation(input_image_val, prompt_val, model_val, pro
307
  )
308
 
309
 
310
- def handle_image_generation(prompt_val, model_val, provider_val, negative_prompt_val, width_val, height_val, steps_val, guidance_val, seed_val, hf_token: gr.OAuthToken = None):
311
  """
312
  Handle image generation request with validation.
313
  """
@@ -318,7 +318,7 @@ def handle_image_generation(prompt_val, model_val, provider_val, negative_prompt
318
 
319
  # Require sign-in via HF OAuth token
320
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
321
- username = None
322
  if not access_token:
323
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
324
 
 
279
  return None, format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
280
 
281
 
282
+ def handle_image_to_image_generation(input_image_val, prompt_val, model_val, provider_val, negative_prompt_val, steps_val, guidance_val, seed_val, hf_token: gr.OAuthToken = None, hf_profile: gr.OAuthProfile = None):
283
  """
284
  Handle image-to-image generation request with validation.
285
  """
 
289
 
290
  # Require sign-in via HF OAuth token
291
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
292
+ username = getattr(hf_profile, "username", None) if hf_profile is not None else None
293
  if not access_token:
294
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
295
 
 
307
  )
308
 
309
 
310
+ def handle_image_generation(prompt_val, model_val, provider_val, negative_prompt_val, width_val, height_val, steps_val, guidance_val, seed_val, hf_token: gr.OAuthToken = None, hf_profile: gr.OAuthProfile = None):
311
  """
312
  Handle image generation request with validation.
313
  """
 
318
 
319
  # Require sign-in via HF OAuth token
320
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
321
+ username = getattr(hf_profile, "username", None) if hf_profile is not None else None
322
  if not access_token:
323
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
324
 
tts_handler.py CHANGED
@@ -155,7 +155,7 @@ def generate_text_to_speech(
155
  return None, format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
156
 
157
 
158
- def handle_text_to_speech_generation(text_val, model_val, provider_val, voice_val, speed_val, audio_url_val, exaggeration_val, temperature_val, cfg_val, hf_token: gr.OAuthToken = None):
159
  """
160
  Handle text-to-speech generation request with validation.
161
  """
@@ -169,7 +169,7 @@ def handle_text_to_speech_generation(text_val, model_val, provider_val, voice_va
169
 
170
  # Require sign-in via HF OAuth token
171
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
172
- username = None
173
  if not access_token:
174
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
175
 
 
155
  return None, format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
156
 
157
 
158
+ def handle_text_to_speech_generation(text_val, model_val, provider_val, voice_val, speed_val, audio_url_val, exaggeration_val, temperature_val, cfg_val, hf_token: gr.OAuthToken = None, hf_profile: gr.OAuthProfile = None):
159
  """
160
  Handle text-to-speech generation request with validation.
161
  """
 
169
 
170
  # Require sign-in via HF OAuth token
171
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
172
+ username = getattr(hf_profile, "username", None) if hf_profile is not None else None
173
  if not access_token:
174
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
175
 
video_handler.py CHANGED
@@ -132,7 +132,7 @@ def generate_video(
132
  return None, format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
133
 
134
 
135
- def handle_video_generation(prompt_val, model_val, provider_val, steps_val, guidance_val, seed_val, hf_token: gr.OAuthToken = None):
136
  """
137
  Handle text-to-video generation request with validation and org access.
138
  """
@@ -140,7 +140,7 @@ def handle_video_generation(prompt_val, model_val, provider_val, steps_val, guid
140
  return None, format_error_message("Validation Error", "Please enter a prompt for video generation")
141
 
142
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
143
- username = None
144
  if not access_token:
145
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
146
 
 
132
  return None, format_error_message("Unexpected Error", f"An unexpected error occurred: {error_msg}")
133
 
134
 
135
+ def handle_video_generation(prompt_val, model_val, provider_val, steps_val, guidance_val, seed_val, hf_token: gr.OAuthToken = None, hf_profile: gr.OAuthProfile = None):
136
  """
137
  Handle text-to-video generation request with validation and org access.
138
  """
 
140
  return None, format_error_message("Validation Error", "Please enter a prompt for video generation")
141
 
142
  access_token = getattr(hf_token, "token", None) if hf_token is not None else None
143
+ username = getattr(hf_profile, "username", None) if hf_profile is not None else None
144
  if not access_token:
145
  return None, format_error_message("Access Required", "Please sign in with Hugging Face (sidebar Login button).")
146