Update app.py

app.py

@@ -4,30 +4,33 @@ import torch
 from diffusers import DiffusionPipeline
 from datasets import load_dataset
 
-headline_gen = pipeline("text2text-generation", model="Michau/t5-base-en-generate-headline", tokenizer="t5-base")
-ar_to_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en")
-en_to_ar_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar")
-
+headline_gen = pipeline("text2text-generation", model="Michau/t5-base-en-generate-headline", tokenizer="t5-base") #Headline Generator Pipeline
+ar_to_en_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ar-en") #Arabic to English translator Pipeline
+en_to_ar_translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ar") #English to Arabic translator Pipeline
 
+#main function
 def generate_headline(selected_language, text):
     if selected_language == "Arabic":
-        translated_text = translate_ar_to_en(text)  # Translate Arabic to English
-        english_headline = generate_headline_english(translated_text)  # Generate headline in English
-        arabic_headline = translate_en_to_ar(english_headline)  # Translate headline back to Arabic
+        translated_text = translate_ar_to_en(text)  #Translate Arabic text to English (so the headlines pipeline will understand the input)
+        english_headline = generate_headline_english(translated_text) #Generate headline in English based on the translated text
+        arabic_headline = translate_en_to_ar(english_headline)  # Translate headline to Arabic (output will be arabic)
         return arabic_headline
 
     elif selected_language == "English":
-        english_headline = generate_headline_english(text)
+        english_headline = generate_headline_english(text) #Generate headline in English based on the text
         return english_headline
 
+#function to translate Arabic Text to English
 def translate_ar_to_en(text):
   var_ar_to_en = ar_to_en_translator(text)[0]['translation_text']
   return var_ar_to_en
 
+#function to translate English Headline to Arabic
 def translate_en_to_ar(text):
   var_en_to_ar = en_to_ar_translator(text)[0]['translation_text']
   return var_en_to_ar
 
+#function to generate headline in english
 def generate_headline_english(text):
     result1 = headline_gen(text, max_length=100, truncation=True)
     result2 = result1[0]['generated_text']
@@ -78,7 +81,7 @@ interface = gr.Interface(
     fn=generate_headline,
     inputs=[
         gr.Dropdown(choices=["Arabic", "English"], label="Select Language"),
-        gr.Textbox(lines=5, placeholder="Enter article text here...", label="Text/Article") 
+        gr.Textbox(lines=5, placeholder="Enter article text here", label="Text/Article") 
     ],
     outputs=gr.Textbox(label="Generated Headline"),