codeteach commited on
Commit
980b3f8
1 Parent(s): 7cd8f48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -3
app.py CHANGED
@@ -57,21 +57,35 @@ def split_text(text, max_tokens=1024):
57
  # Helper function to summarize text
58
  def summarize_text(text):
59
  chunks = split_text(text)
60
- summaries = [summarizer(chunk, max_length=150, min_length=40, do_sample=False)[0]['summary_text'] for chunk in chunks]
 
 
 
 
 
 
61
  return " ".join(summaries)
62
 
63
  # Helper function to translate text
64
  def translate_text(text, language):
65
  translator = get_translator(language)
66
  if translator:
67
- translated_text = translator(text)[0]['translation_text']
68
- return translated_text
 
 
 
 
69
  return text
70
 
71
  def process_text(input_text, language):
 
72
  summary = summarize_text(input_text)
 
73
  bullet_points = generate_bullet_points(summary)
 
74
  translated_text = translate_text(bullet_points, language)
 
75
  return bullet_points, translated_text
76
 
77
  # Create Gradio interface
 
57
  # Helper function to summarize text
58
  def summarize_text(text):
59
  chunks = split_text(text)
60
+ summaries = []
61
+ for chunk in chunks:
62
+ try:
63
+ summary = summarizer(chunk, max_length=150, min_length=40, do_sample=False)[0]['summary_text']
64
+ summaries.append(summary)
65
+ except Exception as e:
66
+ print(f"Error summarizing chunk: {chunk}\nError: {e}")
67
  return " ".join(summaries)
68
 
69
  # Helper function to translate text
70
  def translate_text(text, language):
71
  translator = get_translator(language)
72
  if translator:
73
+ try:
74
+ translated_text = translator(text)[0]['translation_text']
75
+ return translated_text
76
+ except Exception as e:
77
+ print(f"Error translating text: {text}\nError: {e}")
78
+ return text
79
  return text
80
 
81
  def process_text(input_text, language):
82
+ print(f"Input text: {input_text[:500]}...") # Show only the first 500 characters for brevity
83
  summary = summarize_text(input_text)
84
+ print(f"Summary: {summary[:500]}...") # Show only the first 500 characters for brevity
85
  bullet_points = generate_bullet_points(summary)
86
+ print(f"Bullet Points: {bullet_points}")
87
  translated_text = translate_text(bullet_points, language)
88
+ print(f"Translated Text: {translated_text}")
89
  return bullet_points, translated_text
90
 
91
  # Create Gradio interface