oort commited on
Commit
5ec775d
·
verified ·
1 Parent(s): cf6fc38

Revert to initial commit of Gradio_UI

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +6 -59
Gradio_UI.py CHANGED
@@ -258,44 +258,12 @@ class GradioUI:
258
  "",
259
  )
260
 
261
- # def launch(self, **kwargs):
262
- # import gradio as gr
263
-
264
- # with gr.Blocks(fill_height=True) as demo:
265
- # stored_messages = gr.State([])
266
- # file_uploads_log = gr.State([])
267
- # chatbot = gr.Chatbot(
268
- # label="Agent",
269
- # type="messages",
270
- # avatar_images=(
271
- # None,
272
- # "https://huggingface.co/datasets/agents-course/course-images/resolve/main/en/communication/Alfred.png",
273
- # ),
274
- # resizeable=True,
275
- # scale=1,
276
- # )
277
- # # If an upload folder is provided, enable the upload feature
278
- # if self.file_upload_folder is not None:
279
- # upload_file = gr.File(label="Upload a file")
280
- # upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=False)
281
- # upload_file.change(
282
- # self.upload_file,
283
- # [upload_file, file_uploads_log],
284
- # [upload_status, file_uploads_log],
285
- # )
286
- # text_input = gr.Textbox(lines=1, label="Chat Message")
287
- # text_input.submit(
288
- # self.log_user_message,
289
- # [text_input, file_uploads_log],
290
- # [stored_messages, text_input],
291
- # ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
292
  def launch(self, **kwargs):
293
  import gradio as gr
294
-
295
  with gr.Blocks(fill_height=True) as demo:
296
  stored_messages = gr.State([])
297
  file_uploads_log = gr.State([])
298
-
299
  chatbot = gr.Chatbot(
300
  label="Agent",
301
  type="messages",
@@ -306,27 +274,7 @@ class GradioUI:
306
  resizeable=True,
307
  scale=1,
308
  )
309
-
310
- # Show temporary hint when chat loads
311
- demo.load(
312
- lambda: [{"role": "assistant", "content": "Draw me...", "temp_hint": True}],
313
- inputs=None,
314
- outputs=chatbot,
315
- )
316
-
317
- # When any message is added, clear the temporary hint
318
- def update_chat_display(messages):
319
- if messages and isinstance(messages[-1], dict) and messages[-1].get("temp_hint"):
320
- return []
321
- return messages
322
-
323
- chatbot.change(
324
- update_chat_display,
325
- stored_messages,
326
- chatbot,
327
- )
328
-
329
- # Rest of your existing code
330
  if self.file_upload_folder is not None:
331
  upload_file = gr.File(label="Upload a file")
332
  upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=False)
@@ -335,15 +283,14 @@ class GradioUI:
335
  [upload_file, file_uploads_log],
336
  [upload_status, file_uploads_log],
337
  )
338
-
339
- text_input = gr.Textbox(lines=1, label="Chat Message", placeholder="Type 'Draw me...' to get started")
340
  text_input.submit(
341
  self.log_user_message,
342
  [text_input, file_uploads_log],
343
  [stored_messages, text_input],
344
- ).then(self.interact_with_agent, stored_messages, chatbot)
345
-
346
- demo.launch(debug=True, share=True, **kwargs)
347
 
348
 
349
  __all__ = ["stream_to_gradio", "GradioUI"]
 
258
  "",
259
  )
260
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
261
  def launch(self, **kwargs):
262
  import gradio as gr
263
+
264
  with gr.Blocks(fill_height=True) as demo:
265
  stored_messages = gr.State([])
266
  file_uploads_log = gr.State([])
 
267
  chatbot = gr.Chatbot(
268
  label="Agent",
269
  type="messages",
 
274
  resizeable=True,
275
  scale=1,
276
  )
277
+ # If an upload folder is provided, enable the upload feature
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
278
  if self.file_upload_folder is not None:
279
  upload_file = gr.File(label="Upload a file")
280
  upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=False)
 
283
  [upload_file, file_uploads_log],
284
  [upload_status, file_uploads_log],
285
  )
286
+ text_input = gr.Textbox(lines=1, label="Chat Message")
 
287
  text_input.submit(
288
  self.log_user_message,
289
  [text_input, file_uploads_log],
290
  [stored_messages, text_input],
291
+ ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
292
+
293
+ demo.launch(debug=True, share=True, **kwargs)
294
 
295
 
296
  __all__ = ["stream_to_gradio", "GradioUI"]