Update

app-ngrok.py

@@ -23,20 +23,20 @@ initial_model = "WizardLM/WizardCoder-15B-V1.0"
 lora_model = "richardr1126/spider-skeleton-wizard-coder-qlora"
 dataset = "richardr1126/spider-skeleton-context-instruct"
 
-def log_to_firestore(input_message, db_info, temperature, response_text):
-    # For firebase
-    base64_string = os.getenv('FIREBASE')
-    base64_bytes = base64_string.encode('utf-8')
-    json_bytes = base64.b64decode(base64_bytes)
-    json_data = json_bytes.decode('utf-8')
+# Initialize Firebase
+base64_string = os.getenv('FIREBASE')
+base64_bytes = base64_string.encode('utf-8')
+json_bytes = base64.b64decode(base64_bytes)
+json_data = json_bytes.decode('utf-8')
 
-    firebase_auth = json.loads(json_data)
+firebase_auth = json.loads(json_data)
 
-    # Load credentials and initialize Firestore
-    cred = credentials.Certificate(firebase_auth)
-    firebase_admin.initialize_app(cred)
-    db = firestore.client()
+# Load credentials and initialize Firestore
+cred = credentials.Certificate(firebase_auth)
+firebase_admin.initialize_app(cred)
+db = firestore.client()
 
+def log_to_firestore(input_message, db_info, temperature, response_text):
     doc_ref = db.collection('logs').document()
     log_data = {
         'timestamp': firestore.SERVER_TIMESTAMP,
@@ -46,7 +46,7 @@ def log_to_firestore(input_message, db_info, temperature, response_text):
         'output': response_text
     }
     doc_ref.set(log_data)
-
+# End Firebase code
 
 def format(text):
     # Split the text by "|", and get the last element in the list which should be the final query