Update app.py

app.py

@@ -364,6 +364,28 @@ def summarize(input_text):
     summarization_result = summarizer(input_text)
     return summarization_result[0]['summary_text']
 
+def pdf_summarize(company, model, topic):
+    # Check if the selected company and model are valid
+    if company not in ["Beko", "Siemen"]:
+        raise ValueError("Invalid company selected.")
+    if model not in ["WC712", "WV985", "WG44G290BB", "WG37b18KGF"]:
+        raise ValueError("Invalid model number selected.")
+
+    # Define the summarized text for different topics
+    summarization_texts = {
+        "Summarization of General Safety Instructions": "Never place the product on a carpet-covered floor. Unplug the product if it is not in use. The water supply and draining hoses must be securely fastened and remain undamaged. While there is still water inside the product, never open the loading door or remove the filter. Do not force open the locked loading door.",
+        "Summarization of Children's Safety": "Children may breathe in or swallow small parts, causing them to suffocate. Children should not play with the device. Packaging materials may be dangerous for children. Keep packaging materials in a safe place away from the reach of children.",
+        "Summarization of Restriction on User Group": "This product can be used by children who are at the age of 8 and over and people whose physical, sensory, or mental skills are not fully developed or who do not have the necessary required experience and knowledge as long as they are supervised or trained about the safe use of the product and its risks. Children of less than 3 years should be kept away unless continuously supervised.",
+        "Summarization of Energy Efficiency Tips": "To maximize energy efficiency, always use the washing machine with a full load of laundry. Avoid overloading the machine, as it can lead to decreased performance and higher energy consumption. Use cold water for washing whenever possible, as it saves energy and reduces the cost of heating water. Additionally, consider using the appropriate wash cycle for the type of laundry to minimize unnecessary energy usage. Regularly clean the lint filter and perform maintenance tasks to ensure the machine operates at its best efficiency."
+    }
+
+    # Check if the selected topic is valid
+    if topic not in summarization_texts:
+        raise ValueError("Invalid topic selected.")
+
+    # Return the summarized text for the selected topic
+    return summarization_texts[topic]
+
 # Define the function for translation
 def translate(input_text, target_lang):
     # Map the target language selected by the user to the corresponding language code
@@ -384,13 +406,15 @@ def translate(input_text, target_lang):
     # Generate the translation
     generated_tokens = translation_model.generate(
         **model_inputs,
-        forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_code]
+        forced_bos_token_id=translation_tokenizer.lang_code_to_id[lang_code],
+        max_new_tokens=50  # Adjust the value of max_new_tokens as per your preference
     )
 
     # Decode the generated tokens and return the translation
     return translation_tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
 
 
+
 # Create the Gradio interfaces for question-answering, summarization, and translation
 qa_interface = gr.Interface(
     fn=answer_question,
@@ -455,11 +479,40 @@ translation_interface = gr.Interface(
     flagging_options=["Inappropriate Answer"]
 )
 
+summarization_topic_interface = gr.Interface(
+    fn=pdf_summarize,
+    inputs=[
+        gr.Dropdown(
+            ["Beko", "Siemen"], 
+            label="Select Company", 
+            default="Beko"  
+        ),
+        gr.Dropdown(
+            ["WC712", "WV985", "WG44G290BB", "WG37b18KGF"],
+            label="Select Model Number",
+            default="WC712"
+        ),
+        gr.Dropdown(
+            ["Summarization of General Safety Instructions", "Summarization of Children's Safety", "Summarization of Restriction on User Group"],
+            label="Select Topic for Summarization"
+        )
+    ],
+    outputs=gr.Textbox(label="Summary", lines=3),
+    title="Washing Machine Manual Guide Help Desk - Summarization by Topic",
+    description="Summarize specific topics related to the washing machine manual guide!",
+    examples=[
+        ["Beko", "WC712", "Summarization of General Safety Instructions"],
+        ["Siemen", "WG37b18KGF", "Summarization of Children's Safety"],
+        ["Beko", "WV985", "Summarization of Restriction on User Group"],
+        ["Siemen", "WG44G290BB", "Summarization of Energy Efficiency Tips"]
+        
+    ],
+    flagging_options=["Inappropriate Answer"]
+)
+
 # Combine the interfaces into a tabbed interface
-tabbed_interface = gr.TabbedInterface([qa_interface, summarization_interface, translation_interface], ["Question-Answering", "Text Summarization", "Translation"])
+tabbed_interface = gr.TabbedInterface([qa_interface, summarization_interface, translation_interface, summarization_topic_interface], 
+                                      ["Question-Answering", "Text Summarization", "Translation", "Summarization by Topic"])
 
 # Launch the tabbed interface
 tabbed_interface.launch()
-
-
-