Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import logging
 from src.pipeline import QAPipeline
 from langchain_community.document_loaders import CSVLoader
 import os
@@ -6,28 +7,25 @@ import tempfile
 import asyncio
 import nest_asyncio
 
+# Apply nest_asyncio
 nest_asyncio.apply()
 
+# Set Streamlit watcher ignore pattern for torch
 os.environ["STREAMLIT_WATCHER_IGNORE_PATTERNS"] = "*/torch/*"
 
+# Configure logger
+logging.basicConfig(
+    filename="app.log",
+    filemode="a",
+    format="%(asctime)s - %(levelname)s - %(message)s",
+    level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
 pipeline = QAPipeline()
 
 st.title("Recipe Q&A")
-
-# File upload section
-# st.header("Upload CSV")
-# uploaded_file = st.file_uploader("Choose a CSV file", type="csv")
-# if uploaded_file is not None:
-#     if st.button("Add Documents"):
-#         with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmp_file:
-#             tmp_file.write(uploaded_file.read())
-#             tmp_path = tmp_file.name
-#         loader = CSVLoader(file_path=tmp_path)
-#         data = loader.load()
-#         with st.spinner("Uploading documents..."):
-#             asyncio.run(pipeline.upload_documents(data))
-#         os.remove(tmp_path)
-#         st.success("Documents uploaded successfully.")
+logger.info("App started.")
 
 # Query section
 st.header("Ask a Question")
@@ -38,19 +36,32 @@ if "response" not in st.session_state:
 
 # Get answer
 if st.button("Get Answer") and query:
+    logger.info(f"Received query: {query}")
     with st.spinner("Getting answer..."):
-        response = asyncio.run(pipeline.answer_query_(query))
-        st.session_state.response = response  
-        st.write("**Answer:**")
-        st.write(response.answer)
+        try:
+            response = asyncio.run(pipeline.answer_query_(query))
+            st.session_state.response = response  
+            logger.info(f"Response: {response.answer}")
+            st.write("**Answer:**")
+            st.write(response.answer)
+        except Exception as e:
+            logger.error(f"Error while getting answer: {e}")
+            st.error("An error occurred while processing your query.")
 
 if st.session_state.response:
     if st.session_state.response.web_search:
         if st.button("Search the web for this?"):
+            logger.info(f"Initiating web search for query: {query}")
             with st.spinner("Searching web..."):
-                web_response = asyncio.run(pipeline.search_web(query))
-                if web_response:
-                    st.write("**Web Search Result:**")
-                    st.write(web_response)
-                else:
-                    st.write("No web search result found.")
+                try:
+                    web_response = asyncio.run(pipeline.search_web(query))
+                    if web_response:
+                        logger.info(f"Web response: {web_response}")
+                        st.write("**Web Search Result:**")
+                        st.write(web_response)
+                    else:
+                        logger.info("No web search results found.")
+                        st.write("No web search result found.")
+                except Exception as e:
+                    logger.error(f"Error during web search: {e}")
+                    st.error("Web search failed.")