Update app.py

app.py

@@ -8,7 +8,9 @@ def setup_logging():
     logging.basicConfig(
         level=logging.INFO,
         format='%(asctime)s - %(levelname)s - %(message)s',
-        handlers=[logging.StreamHandler()]
+        handlers=[
+            logging.StreamHandler()
+        ]
     )
 
 # Function to extract text from a PDF file
@@ -23,8 +25,8 @@ def main():
     setup_logging()
     logging.info("Starting the Streamlit app.")
 
-    # Initialize the medical text summarization pipeline
-    summarizer = pipeline("summarization", model="your/medical_text_summarization_model")
+    # Initialize the summarization pipeline with a public model for long medical texts
+    summarizer = pipeline("summarization", model="allenai/led-large-16384-arxiv")
 
     # Streamlit UI
     st.title("GenAI Lab Report Analyzer")
@@ -51,7 +53,7 @@ def main():
     elif input_type == "Audio":
         audio = st.file_uploader("Upload your audio file:", type=["wav", "mp3", "m4a"])
 
-    if st.button("Analyze Report"):
+    if st.button("Report Result"):
         try:
             summary = None
             if input_type == "Text" and text:
@@ -74,6 +76,7 @@ def main():
                 summary = "Invalid input. Please provide a valid file or text."
                 logging.warning("Invalid input type provided.")
 
+            # Display the summarized result
             st.text_area("Report Result:", summary[0]['summary_text'] if isinstance(summary, list) else summary, height=200)
         except Exception as e:
             logging.error(f"Error during summarization: {e}")