Update app.py

app.py

@@ -18,7 +18,7 @@ from neo4j import GraphDatabase
 from llama_index.core import SimpleDirectoryReader, KnowledgeGraphIndex, Settings
 from llama_index.core.graph_stores import SimpleGraphStore
 from llama_index.core import StorageContext
-from llama_index.embeddings import OpenAIEmbedding
+from llama_index.embeddings.openai import OpenAIEmbedding  # Fix import path
 
 def main():
     st.set_page_config(
@@ -192,8 +192,8 @@ def process_document(file_path, graph, openai_api_key):
     Settings.chunk_size = 512
     Settings.llm = ChatOpenAI(temperature=0, model="gpt-4")
     
-    # Setup embeddings
-    embed_model = OpenAIEmbedding()
+    # Setup embeddings using correct import
+    embed_model = OpenAIEmbedding()  # This should work now
     
     # Create graph store
     graph_store = SimpleGraphStore()