tregu0458 commited on
Commit
3ae41e8
·
verified ·
1 Parent(s): 7eefb5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from langchain_community.document_loaders import YoutubeLoader
3
  from langchain.text_splitter import TokenTextSplitter
4
  from fastapi import FastAPI
 
5
  max_textboxes = 10
6
 
7
  def process_youtube_url(url, language):
@@ -19,14 +20,14 @@ def process_youtube_url(url, language):
19
  chunks = text_splitter.split_text(text)
20
 
21
  output_textboxes = [gr.Textbox(label=f"Chunk {i}", value=chunk, show_copy_button=True, visible=True) for i, chunk in enumerate(chunks)]
22
- + [gr.Textbox(visible=False) for _ in range(max_textboxes - len(chunks))]
23
-
24
  return output_textboxes, [], "", char_count
25
  except Exception as e:
26
  error_msg = str(e)
27
  available_languages = extract_available_languages(error_msg)
28
  recommended_language = extract_recommended_language(error_msg)
29
- return [gr.Textbox(label="Error", value=error_msg, show_copy_button=True) for ind in range(max_textboxes)], available_languages, recommended_language, 0
30
 
31
  def extract_available_languages(error_msg):
32
  languages = []
@@ -56,7 +57,7 @@ iface = gr.Interface(
56
  gr.Dropdown(label="Language",value="ja",choices=["en","en-US", "ja", "fr","de","it"],allow_custom_value=True),
57
  ],
58
  outputs=[
59
- [gr.Textbox(f"chunk{ind}") for ind in range(max_textboxes)],
60
  gr.Dropdown(label="Available Languages", allow_custom_value=True),
61
  gr.Textbox(label="Recommended Language"),
62
  gr.Number(label="Character Count")
 
2
  from langchain_community.document_loaders import YoutubeLoader
3
  from langchain.text_splitter import TokenTextSplitter
4
  from fastapi import FastAPI
5
+
6
  max_textboxes = 10
7
 
8
  def process_youtube_url(url, language):
 
20
  chunks = text_splitter.split_text(text)
21
 
22
  output_textboxes = [gr.Textbox(label=f"Chunk {i}", value=chunk, show_copy_button=True, visible=True) for i, chunk in enumerate(chunks)]
23
+ output_textboxes += [gr.Textbox(visible=False) for _ in range(max_textboxes - len(chunks))]
24
+
25
  return output_textboxes, [], "", char_count
26
  except Exception as e:
27
  error_msg = str(e)
28
  available_languages = extract_available_languages(error_msg)
29
  recommended_language = extract_recommended_language(error_msg)
30
+ return [gr.Textbox(label="Error", value=error_msg, show_copy_button=True)] + [gr.Textbox(visible=False) for _ in range(max_textboxes - 1)], available_languages, recommended_language, 0
31
 
32
  def extract_available_languages(error_msg):
33
  languages = []
 
57
  gr.Dropdown(label="Language",value="ja",choices=["en","en-US", "ja", "fr","de","it"],allow_custom_value=True),
58
  ],
59
  outputs=[
60
+ gr.Column(children=[gr.Textbox(f"chunk{ind}") for ind in range(max_textboxes)]),
61
  gr.Dropdown(label="Available Languages", allow_custom_value=True),
62
  gr.Textbox(label="Recommended Language"),
63
  gr.Number(label="Character Count")