Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,9 +13,9 @@ import random
|
|
13 |
# Load environment variables
|
14 |
load_dotenv()
|
15 |
|
16 |
-
# Initialize Firebase
|
17 |
-
cred = credentials.Certificate("
|
18 |
-
firebase_admin.initialize_app(cred, {"databaseURL": "https://
|
19 |
|
20 |
# Configure Llama index settings
|
21 |
Settings.llm = HuggingFaceInferenceAPI(
|
@@ -38,7 +38,7 @@ PDF_DIRECTORY = 'data'
|
|
38 |
os.makedirs(PDF_DIRECTORY, exist_ok=True)
|
39 |
os.makedirs(PERSIST_DIR, exist_ok=True)
|
40 |
|
41 |
-
#
|
42 |
session_chat_histories = {}
|
43 |
|
44 |
def select_random_name():
|
@@ -103,9 +103,11 @@ def save_chat_message(session_id, message_data):
|
|
103 |
ref.push().set(message_data)
|
104 |
|
105 |
def chat_interface(message, history):
|
106 |
-
|
|
|
|
|
107 |
response = handle_query(session_id, message)
|
108 |
-
return response,
|
109 |
|
110 |
css = '''
|
111 |
.circle-logo {
|
|
|
13 |
# Load environment variables
|
14 |
load_dotenv()
|
15 |
|
16 |
+
# Initialize Firebase with provided credentials and URL
|
17 |
+
cred = credentials.Certificate("redfernstech-fd8fe-firebase-adminsdk-g9vcn-0537b4efd6.json")
|
18 |
+
firebase_admin.initialize_app(cred, {"databaseURL": "https://redfernstech-fd8fe-default-rtdb.firebaseio.com/"})
|
19 |
|
20 |
# Configure Llama index settings
|
21 |
Settings.llm = HuggingFaceInferenceAPI(
|
|
|
38 |
os.makedirs(PDF_DIRECTORY, exist_ok=True)
|
39 |
os.makedirs(PERSIST_DIR, exist_ok=True)
|
40 |
|
41 |
+
# Dictionary to store chat histories for different sessions
|
42 |
session_chat_histories = {}
|
43 |
|
44 |
def select_random_name():
|
|
|
103 |
ref.push().set(message_data)
|
104 |
|
105 |
def chat_interface(message, history):
|
106 |
+
# Retrieve or create a new session ID based on history
|
107 |
+
session_id = history[0][1] if history and history[0][1] else str(uuid.uuid4())
|
108 |
+
history.append((message, session_id)) # Append the session ID to history
|
109 |
response = handle_query(session_id, message)
|
110 |
+
return response, history
|
111 |
|
112 |
css = '''
|
113 |
.circle-logo {
|