Update chat_with_doc.py

chat_with_doc.py

@@ -1,7 +1,6 @@
 import os
 import openai
 import streamlit as st
-import json
 
 # Load environment variables
 from dotenv import load_dotenv
@@ -25,10 +24,10 @@ def get_openai_response(prompt, model="gpt-4", max_tokens=150):
 # Streamlit UI
 st.title("Chat with Your Document")
 
-# Load chunked data from the file (chunked_data.json)
+# Load chunked data from the .txt file (chunked_data.txt)
 try:
-    with open("chunked_data.json", "r") as f:
-        chunked_data = json.load(f)
+    with open("chunked_data.txt", "r") as f:
+        chunked_data = f.read().split("\n---\n")  # Split by delimiter to separate chunks
         st.write("Document has been chunked into the following parts:")
         for i, chunk_part in enumerate(chunked_data, 1):
             st.write(f"**Chunk {i}:**\n{chunk_part}\n")
@@ -48,4 +47,3 @@ if st.button("Get Answer"):
         st.write(f"**Answer**: {response}")
     else:
         st.error("Please provide a question.")
-