use simple time

app.py

@@ -5,6 +5,10 @@ from PIL import Image
 import cProfile
 import pstats
 from io import StringIO
+import time
+
+def print_time(start, end):
+    st.session_state.messages.append({"role": "assistant", "content": f"Execution time: {end - start} seconds}")
 
 # Streamlit app
 def main():
@@ -30,24 +34,29 @@ def main():
 
     uploaded_file = st.sidebar.file_uploader("Choose a PDF file", type="pdf")
     if uploaded_file is not None:
-        profiler = cProfile.Profile()
-        profiler.enable()
+        # profiler = cProfile.Profile()
+        # profiler.enable()
         st.sidebar.success("File uploaded successfully.")
+        start_time = time.time()
         vector_store = PDF_4_QA(uploaded_file)
+        end_time = time.time()
+        print_time(start_time, end_time)
+        start_time = time.time()
         QA_Bot(vector_store)
-        profiler.disable()
-        #stats.print_stats()
-        s = StringIO()
-        ps = pstats.Stats(profiler, stream=s).sort_stats('cumulative')
+        end_time = time.time()
+        print_time(start_time, end_time)
+        # profiler.disable()
+        # s = StringIO()
+        # ps = pstats.Stats(profiler, stream=s).sort_stats('cumulative')
         
         # Print the profiling results to the StringIO object
-        ps.print_stats()
+        # ps.print_stats()
         
         # Get the profiling results as a string
-        profiling_results = s.getvalue()
+        # profiling_results = s.getvalue()
         
         # Print the profiling results
-        st.session_state.messages.append({"role": "assistant", "content": profiling_results})
+        # st.session_state.messages.append({"role": "assistant", "content": profiling_results})
         
 
 if __name__ == '__main__':