Update app.py

app.py

@@ -10,10 +10,11 @@ import chromadb
 chromadb.api.client.SharedSystemClient.clear_system_cache()
 
 import os
-os.environ["OPENAI_API_KEY"]=os.getenv('OPENAI_API_KEY')
+os.environ["OPENAI_API_KEY"] = os.getenv('OPENAI_API_KEY')
+
 # Initialize the embeddings and model
 embd = OpenAIEmbeddings()
-llm = ChatOpenAI(model_name="gpt-4", temperature=0)
+llm = ChatOpenAI(model_name="gpt-4o", temperature=0)
 
 # Initialize conversation history
 if "conversation_history" not in st.session_state:
@@ -60,29 +61,29 @@ if uploaded_file:
     query = st.text_input("Ask a question:")
     
     if query:
-    # Process the query
-    result = qa_chain({"query": query})
-    answer = result["result"]
-    sources = result["source_documents"]
-    
-    # Append to conversation history
-    st.session_state.conversation_history.append((query, answer, sources))
-    
-    # Display the current answer
-    st.write("**Answer:**", answer)
-    
-    # Display the sources
-    st.subheader("Source Documents")
-    for i, doc in enumerate(sources, start=1):
-        st.write(f"**Source {i}:** {doc.metadata.get('source', 'Unknown Source')}")
-        st.write(doc.page_content[:500])  # Display the first 500 characters of the source content
-    
-    # Display conversation history
-    st.subheader("Conversation History")
-    for idx, (q, a, s) in enumerate(st.session_state.conversation_history, 1):
-        st.write(f"**Q{idx}:** {q}")
-        st.write(f"**A{idx}:** {a}")
-        st.write(f"**Sources for Q{idx}:**")
-        for i, doc in enumerate(s, start=1):
+        # Process the query
+        result = qa_chain({"query": query})
+        answer = result["result"]
+        sources = result["source_documents"]
+        
+        # Append to conversation history
+        st.session_state.conversation_history.append((query, answer, sources))
+        
+        # Display the current answer
+        st.write("**Answer:**", answer)
+        
+        # Display the sources
+        st.subheader("Source Documents")
+        for i, doc in enumerate(sources, start=1):
             st.write(f"**Source {i}:** {doc.metadata.get('source', 'Unknown Source')}")
-            st.write(doc.page_content[:300])  # Show a snippet for brevity
+            st.write(doc.page_content[:500])  # Display the first 500 characters of the source content
+        
+        # Display conversation history
+        st.subheader("Conversation History")
+        for idx, (q, a, s) in enumerate(st.session_state.conversation_history, 1):
+            st.write(f"**Q{idx}:** {q}")
+            st.write(f"**A{idx}:** {a}")
+            st.write(f"**Sources for Q{idx}:**")
+            for i, doc in enumerate(s, start=1):
+                st.write(f"**Source {i}:** {doc.metadata.get('source', 'Unknown Source')}")
+                st.write(doc.page_content[:300])  # Show a snippet for brevity