Update app.py

app.py

@@ -15,6 +15,9 @@ GROQ_API_KEY = "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"
 # Initialize Neo4j graph
 graph = Neo4jGraph(url=NEO4J_URI, username=NEO4J_USERNAME, password=NEO4J_PASSWORD)
 
+# Initialize Neo4j graph
+graph = Neo4jGraph(url=NEO4J_URI, username=NEO4J_USERNAME, password=NEO4J_PASSWORD)
+
 # Load movie dataset into Neo4j
 moview_query = """
 LOAD CSV WITH HEADERS FROM
@@ -48,4 +51,7 @@ st.title("Movie Director Finder")
 query = st.text_input("Enter your query:", "Who was the director of the movie Casino")
 if st.button("Find Director"):
     response = chain.invoke({"query": query})
-    st.write(response)
+    if response and response.get("result"):
+        st.write(response["result"])
+    else:
+        st.write("I don't know the answer.")