Spaces:
Sleeping
Sleeping
File size: 3,665 Bytes
ae7d660 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import streamlit as st
import logging
from models.summarizer import TextSummarizer
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
...
def main():
# Setup logging
setup_logging()
logging.info("Starting GenAI Lab Report Analyzer with Streamlit.")
# Initialize summarizer
summarizer = TextSummarizer()
# Streamlit UI
st.title("GenAI Lab Report Analyzer")
st.write("Upload a file, record audio, or type text to generate a summary. Select the appropriate input type and provide the input.")
input_type = st.radio(
"Select Input Type:",
options=["Text", "Text File", "PDF", "DOCX", "Audio"],
index=0
)
file = None
text = None
audio = None
if input_type == "Text":
text = st.text_area("Enter your text here:", placeholder="Type your text here...")
elif input_type in ["Text File", "PDF", "DOCX"]:
file = st.file_uploader(f"Upload your {input_type}:", type=["txt", "pdf", "docx"])
elif input_type == "Audio":
audio = st.file_uploader("Upload your audio file:", type=["wav", "mp3", "m4a"])
if st.button("Report Result"):
try:
if input_type == "Text" and text:
logging.info("Processing text input.")
processed_text = handle_text_input(text)
summary = summarizer.summarize(processed_text)
logging.info("Text input processed successfully.")
elif input_type in ["Text File", "PDF", "DOCX"] and file:
if input_type == "Text File":
logging.info(f"Processing text file: {file.name}")
processed_text = read_text_file(file)
elif input_type == "PDF":
logging.info(f"Processing PDF file: {file.name}")
processed_text = read_pdf_file(file)
elif input_type == "DOCX":
logging.info(f"Processing DOCX file: {file.name}")
processed_text = read_docx_file(file)
if processed_text:
summary = summarizer.summarize(processed_text)
logging.info(f"{input_type} processed successfully.")
else:
summary = "Failed to process the file. Check logs for more details."
logging.error(f"Failed to process {input_type}: {file.name}")
elif input_type == "Audio" and audio:
logging.info("Processing audio input.")
processed_text = audio_to_text(audio)
if processed_text:
summary = summarizer.summarize(processed_text)
logging.info("Audio input processed successfully.")
else:
summary = "Failed to convert audio to text. Check logs for more details."
logging.error("Failed to convert audio to text.")
else:
summary = "Invalid input. Please provide a valid file or text."
logging.warning("Invalid input type provided.")
st.text_area("Report Result:", summary, height=200)
except Exception as e:
logging.error(f"Error during summarization: {e}")
st.error("An error occurred during summarization. Please check the logs for more details.")
logging.info("Closing GenAI Lab Report Analyzer with Streamlit.")
...
if __name__ == "__main__":
main() |