Update app.py

app.py

@@ -10,6 +10,9 @@ translation_models = {
     'Spanish': "Helsinki-NLP/opus-mt-en-es"
 }
 
+# Initialize summarization pipeline
+summarizer = pipeline("summarization")
+
 # Initialize translation pipeline
 def get_translator(language):
     model_name = translation_models.get(language)
@@ -32,6 +35,11 @@ def generate_bullet_points(text):
     
     return "\n".join(f"- {point}" for point in bullet_points)
 
+# Helper function to summarize text
+def summarize_text(text):
+    summary = summarizer(text, max_length=150, min_length=40, do_sample=False)
+    return summary[0]['summary_text']
+
 # Helper function to translate text
 def translate_text(text, language):
     translator = get_translator(language)
@@ -41,7 +49,8 @@ def translate_text(text, language):
     return text
 
 def process_text(input_text, language):
-    bullet_points = generate_bullet_points(input_text)
+    summary = summarize_text(input_text)
+    bullet_points = generate_bullet_points(summary)
     translated_text = translate_text(bullet_points, language)
     return bullet_points, translated_text
 
@@ -57,7 +66,8 @@ iface = gr.Interface(
         gr.Textbox(label="Translated Bullet Points")
     ],
     title="Text to Bullet Points and Translation",
-    description="Paste any text, and the program will transform it into bullet points. Optionally, translate the bullet points into Vietnamese, Japanese, Thai, or Spanish."
+    description="Paste any text, and the program will summarize it into bullet points. Optionally, translate the bullet points into Vietnamese, Japanese, Thai, or Spanish."
 )
 
 iface.launch()
+