Update app.py

app.py

@@ -1,4 +1,4 @@
-import streamlit as st
+import gradio as gr
 import logging
 from transformers import pipeline
 from services.text_input_handler import handle_text_input
@@ -16,78 +16,77 @@ class TextSummarizer:
         summary = self.summarizer(text, max_length=150, min_length=30, do_sample=False)
         return summary[0]['summary_text']
 
+def process_input(input_type, input_data):
+    try:
+        logging.info(f"Processing input type: {input_type}")
+        
+        if input_type == "Text":
+            processed_text = handle_text_input(input_data)
+        elif input_type == "Text File":
+            processed_text = read_text_file(input_data)
+        elif input_type == "PDF":
+            processed_text = read_pdf_file(input_data)
+        elif input_type == "DOCX":
+            processed_text = read_docx_file(input_data)
+        elif input_type == "Audio":
+            processed_text = audio_to_text(input_data)
+        else:
+            return "Invalid input type."
+
+        if processed_text:
+            summary = summarizer.summarize(processed_text)
+            logging.info(f"{input_type} processed successfully.")
+            return summary
+        else:
+            logging.error(f"Failed to process {input_type}.")
+            return "Failed to process the input. Check logs for more details."
+    except Exception as e:
+        logging.error(f"Error during summarization: {e}")
+        return "An error occurred during summarization. Please check the logs for more details."
+
 def main():
     # Setup logging
     setup_logging()
-    logging.info("Starting GenAI Lab Report Analyzer with Streamlit.")
+    logging.info("Starting GenAI Lab Report Analyzer with Gradio.")
     
     # Initialize summarizer
+    global 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
+    # Gradio interface
+    input_type = gr.inputs.Radio(choices=["Text", "Text File", "PDF", "DOCX", "Audio"], label="Select Input Type")
+    input_data = gr.inputs.Textbox(lines=5, label="Enter your text here")  # Default for text input
 
-    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"])
+    def update_input_type(input_type):
+        if input_type == "Text":
+            return gr.update(value="", placeholder="Type your text here...")
+        elif input_type == "Text File":
+            return gr.update(value=None, type="file", label="Upload your text file")
+        elif input_type == "PDF":
+            return gr.update(value=None, type="file", label="Upload your PDF file")
+        elif input_type == "DOCX":
+            return gr.update(value=None, type="file", label="Upload your DOCX file")
+        elif input_type == "Audio":
+            return gr.update(value=None, type="file", label="Upload your audio file")
+        else:
+            return gr.update(value="", placeholder="Invalid input type")
 
-    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.")
+    input_data = gr.inputs.Textbox(lines=5, label="Enter your text here")  # Default for text input
+    output = gr.outputs.Textbox(label="Report Result")
 
-            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.")
+    interface = gr.Interface(
+        fn=process_input,
+        inputs=[input_type, input_data],
+        outputs=output,
+        title="GenAI Lab Report Analyzer",
+        description="Upload a file, record audio, or type text to generate a summary. Select the appropriate input type and provide the input.",
+        live=True,
+        theme="default",
+        layout="vertical",
+        allow_flagging="never"
+    )
 
-    logging.info("Closing GenAI Lab Report Analyzer with Streamlit.")
+    interface.launch()
 
 if __name__ == "__main__":
     main()