srinuksv commited on
Commit
19e8df5
·
verified ·
1 Parent(s): 7c5507d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -6
app.py CHANGED
@@ -13,9 +13,9 @@ import random
13
  # Load environment variables
14
  load_dotenv()
15
 
16
- # Initialize Firebase
17
- cred = credentials.Certificate("path_to_your_firebase_adminsdk_json.json")
18
- firebase_admin.initialize_app(cred, {"databaseURL": "https://your-project-id.firebaseio.com/"})
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
- # Variable to store chat history
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
- session_id = history[0][1] if history else str(uuid.uuid4())
 
 
107
  response = handle_query(session_id, message)
108
- return response, [(message, 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 {