Update app.py

app.py

@@ -98,6 +98,20 @@ if uploaded_file:
                 # ✅ Limit number of retrieved documents
                 retriever = langchain_vector_store.as_retriever(search_kwargs={"k": 5})
 
+                # ✅ Create LangChain Query Execution Pipeline
+                system_prompt = (
+                    "You are an assistant for question-answering tasks. "
+                    "Use the following pieces of retrieved context to answer "
+                    "the question. Keep the answer concise.\n\n{context}"
+                )
+
+                prompt = ChatPromptTemplate.from_messages(
+                    [("system", system_prompt), ("human", "{input}")]
+                )
+
+                question_answer_chain = create_stuff_documents_chain(ChatOpenAI(model="gpt-4o"), prompt)
+                langchain_rag_chain = create_retrieval_chain(retriever, question_answer_chain)
+
                 # ✅ Query Processing
                 query = st.text_input("Ask a question about your data (LangChain):")
 
@@ -106,6 +120,7 @@ if uploaded_file:
                         retrieved_context = "\n\n".join([doc.page_content for doc in retriever.get_relevant_documents(query)])
                         retrieved_context = retrieved_context[:3000]  
 
+                        # ✅ Ensure that we use the retrieved context
                         system_prompt = (
                             "You are an assistant for question-answering tasks. "
                             "Use the following pieces of retrieved context to answer "
@@ -118,9 +133,13 @@ if uploaded_file:
                     except Exception as e:
                         error_message = traceback.format_exc()
                         st.error(f"Error processing query: {e}")
-                        st.text(error_message)
+                        st.text(error_message)  
 
             except Exception as e:
                 error_message = traceback.format_exc()
                 st.error(f"Error processing with LangChain: {e}")
-                st.text(error_message)
+                st.text(error_message)  
+    except Exception as e:
+        error_message = traceback.format_exc()
+        st.error(f"Error reading uploaded file: {e}")
+        st.text(error_message)  #