Update app.py

app.py

@@ -53,7 +53,7 @@ def get_json_file(json_docs):
     temp_filepath = os.path.join(temp_dir.name, json_docs.name)
     with open(temp_filepath, "wb") as f:
         f.write(json_docs.getvalue())
-    json_loader = JSONLoader(temp_filepath, jq_schema='.messages[]', content_key="content")
+    json_loader = JSONLoader(temp_filepath, jq_schema='.messages[]', content_key="content", metadata_func=metadata_func)
     json_doc = json_loader.load()
     return json_doc
 
@@ -111,6 +111,18 @@ def handle_userinput(user_question):
                 "{{MSG}}", message.content), unsafe_allow_html=True)
 
 
+def metadata_func(record: dict, metadata: dict) -> dict:
+
+    metadata["sender_name"] = record.get("sender_name")
+    metadata["timestamp_ms"] = record.get("timestamp_ms")
+
+    if "source" in metadata:
+        source = metadata["source"].split("/")
+        source = source[source.index("langchain"):]
+        metadata["source"] = "/".join(source)
+
+    return metadata
+
 def main():
     load_dotenv()
     st.set_page_config(page_title="Chat with multiple Files",