tregu0458 commited on
Commit
2d161b2
·
verified ·
1 Parent(s): fa531c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -10,12 +10,14 @@ def process_youtube_url(url, language):
10
  language=[language],
11
  )
12
  docs = loader.load()
13
- return str(docs), [], ""
 
 
14
  except Exception as e:
15
  error_msg = str(e)
16
  available_languages = extract_available_languages(error_msg)
17
  recommended_language = extract_recommended_language(error_msg)
18
- return f"Error: {error_msg}", available_languages, recommended_language
19
 
20
  def extract_available_languages(error_msg):
21
  languages = []
@@ -45,9 +47,10 @@ iface = gr.Interface(
45
  gr.Dropdown(label="Language",value="ja",choices=["en","en-US", "ja", "fr","de","it"],allow_custom_value=True),
46
  ],
47
  outputs=[
48
- gr.Textbox(label="Loaded Documents / Error Message",show_copy_button=True),
49
  gr.Dropdown(label="Available Languages"),
50
- gr.Textbox(label="Recommended Language")
 
51
  ],
52
  examples = [["https://youtu.be/6Af6b_wyiwI?si=zqD9-kjw24lpRJw3","ja"],["https://youtu.be/9kxL9Cf46VM?si=ADgUmDXb6riA-lgb","ja"]],
53
  title="YouTube Transcript Loader",
 
10
  language=[language],
11
  )
12
  docs = loader.load()
13
+ text = str(docs)
14
+ char_count = len(text)
15
+ return text, [], "", char_count
16
  except Exception as e:
17
  error_msg = str(e)
18
  available_languages = extract_available_languages(error_msg)
19
  recommended_language = extract_recommended_language(error_msg)
20
+ return f"Error: {error_msg}", available_languages, recommended_language, 0
21
 
22
  def extract_available_languages(error_msg):
23
  languages = []
 
47
  gr.Dropdown(label="Language",value="ja",choices=["en","en-US", "ja", "fr","de","it"],allow_custom_value=True),
48
  ],
49
  outputs=[
50
+ gr.Textbox(label="Loaded Documents / Error Message",show_copy_button=True),
51
  gr.Dropdown(label="Available Languages"),
52
+ gr.Textbox(label="Recommended Language"),
53
+ gr.Number(label="Character Count")
54
  ],
55
  examples = [["https://youtu.be/6Af6b_wyiwI?si=zqD9-kjw24lpRJw3","ja"],["https://youtu.be/9kxL9Cf46VM?si=ADgUmDXb6riA-lgb","ja"]],
56
  title="YouTube Transcript Loader",