Update app.py

app.py

@@ -1,34 +1,28 @@
 import streamlit as st
 import subprocess
-import pdfplumber
-import camelot
-import tabula
-from transformers import pipeline
 import re
 import plotly.express as px
 import pandas as pd
 
-# Ensure required libraries are installed
-try:
-    import torch
-except ImportError:
-    subprocess.check_call(["pip", "install", "torch", "pandas"])
-try:
-    import camelot
-except ImportError:
-    subprocess.check_call(["pip", "install", "camelot-py[cv]"])
-try:
-    import tabula
-except ImportError:
-    subprocess.check_call(["pip", "install", "tabula-py"])
-try:
-    import pdfplumber
-except ImportError:
-    subprocess.check_call(["pip", "install", "pdfplumber"])
-try:
-    import plotly
-except ImportError:
-    subprocess.check_call(["pip", "install", "plotly"])
+# Ensure required libraries are installed and imported
+def ensure_library_installed(library_name, install_name=None):
+    """Ensure a library is installed and import it."""
+    try:
+        globals()[library_name] = __import__(library_name)
+    except ImportError:
+        install_name = install_name or library_name
+        subprocess.check_call(["pip", "install", install_name])
+        globals()[library_name] = __import__(library_name)
+
+# Ensure required libraries
+ensure_library_installed("torch")
+ensure_library_installed("camelot", "camelot-py[cv]")
+ensure_library_installed("tabula", "tabula-py")
+ensure_library_installed("pdfplumber")
+ensure_library_installed("plotly")
+
+# Import other required modules
+from transformers import pipeline
 
 # NLP Model for summarization
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
@@ -77,8 +71,6 @@ if input_type == "Upload PDF":
                         for i, table in enumerate(tables):
                             st.write(f"Table {i+1}")
                             st.dataframe(table.df)
-                            # Save table to CSV (optional)
-                            # table.to_csv(f"table_{i+1}.csv")
                     else:
                         st.warning("No tables found using Camelot.")
                 except Exception as e:
@@ -92,8 +84,6 @@ if input_type == "Upload PDF":
                         for i, table in enumerate(tabula_tables):
                             st.write(f"Table {i+1}")
                             st.dataframe(table)
-                            # Save table to CSV (optional)
-                            # table.to_csv(f"tabula_table_{i+1}.csv")
                     else:
                         st.warning("No tables found using Tabula.")
                 except Exception as e: