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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -3,7 +3,7 @@ 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):
9
  try:
@@ -19,15 +19,15 @@ def process_youtube_url(url, language):
19
  text_splitter = TokenTextSplitter(chunk_size=10000, chunk_overlap=0)
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,7 +57,7 @@ iface = gr.Interface(
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")
 
3
  from langchain.text_splitter import TokenTextSplitter
4
  from fastapi import FastAPI
5
 
6
+ max_textboxes = 5
7
 
8
  def process_youtube_url(url, language):
9
  try:
 
19
  text_splitter = TokenTextSplitter(chunk_size=10000, chunk_overlap=0)
20
  chunks = text_splitter.split_text(text)
21
 
22
+ output_textboxes = [chunk for i, chunk in enumerate(chunks)]
23
+ output_textboxes += ["" 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 ["" for _ in range(max_textboxes)], 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.Textbox(label=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")