Update app.py

app.py

@@ -7,10 +7,12 @@ from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
-NEO4J_URI = "neo4j+s://64beefae.databases.neo4j.io"
-NEO4J_USERNAME = "neo4j"
-NEO4J_PASSWORD = "OTj5yGnWLF59yx4UX1g3xABarVOvVKiM3CT9L4bNkF8"
-GROQ_API_KEY = "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"
+
+# Set environment variables
+NEO4J_URI = os.getenv("NEO4J_URI")
+NEO4J_USERNAME = os.getenv("NEO4J_USERNAME")
+NEO4J_PASSWORD = os.getenv("NEO4J_PASSWORD")
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
 # Initialize Neo4j graph
 graph = Neo4jGraph(url=NEO4J_URI, username=NEO4J_USERNAME, password=NEO4J_PASSWORD)
@@ -52,9 +54,17 @@ if st.button("Find Director"):
     # Debug: Print the raw response to check its structure
     st.write("Raw Response:", response)
     
-    # Process and display the response
-    if response and 'result' in response:
-        result = response['result']
-        st.write(f"Answer: {result}")
+    # Extract the director's name from the full context
+    if response and 'full_context' in response:
+        full_context = response['full_context']
+        if full_context and isinstance(full_context, list) and len(full_context) > 0:
+            director_info = full_context[0]
+            director_name = director_info.get('p.name', None)
+            if director_name:
+                st.write(f"The director of the movie is {director_name}.")
+            else:
+                st.write("Could not find the director's name in the response.")
+        else:
+            st.write("No full context found in the response.")
     else:
-        st.write("I don't know the answer.")
+        st.write("I don't know the answer.")