Update app.py

app.py

@@ -1,11 +1,20 @@
 import streamlit as st
 import logging
-from models.summarizer import TextSummarizer
+from transformers import pipeline
 from services.text_input_handler import handle_text_input
 from services.file_input_handler import read_text_file, read_pdf_file, read_docx_file
 from services.audio_input_handler import audio_to_text
 from utils.logging_utils import setup_logging
 
+class TextSummarizer:
+    def __init__(self, model_name="t5-small"):
+        self.summarizer = pipeline("summarization", model=model_name)
+
+    def summarize(self, text):
+        if not text:
+            return "No text to summarize."
+        summary = self.summarizer(text, max_length=150, min_length=30, do_sample=False)
+        return summary[0]['summary_text']
 
 def main():
     # Setup logging
@@ -80,6 +89,5 @@ def main():
 
     logging.info("Closing GenAI Lab Report Analyzer with Streamlit.")
 
-
 if __name__ == "__main__":
-    main()
+    main()