Update app.py

app.py

@@ -3,32 +3,13 @@ import assemblyai as aai
 from transformers import pipeline
 import pandas as pd
 import os
-import firebase_admin
-from firebase_admin import credentials, db
+from firebase_admin import credentials, db, initialize_app
 
-# Replace with your Firebase credentials JSON
-firebase_credentials = {
-    "type": "service_account",
-    "project_id": "learning-5fd92",
-    "private_key_id": "3d926218194aee3902d5f92b2973effbbf27b4f8",
-    "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQCiZPrNvBtVqjM1\na+/MzyZ86s/Z7fHguR1FIuNM65nxZcG6tiOgFUD5RVq9RJKpHb1eB5NTCaLVVDGg\n1nmjX45Zt/3Wj6FQKFv20iruElaNpRfhhvxCjxb37XtgwAI4BwCi8nwCi3hjuPbN\ni2jd4cCQw0lR+bPDUq/1RF/rZC5OBCmV3KpM1XuZNfpB/0UQmOdhlenba9SYhkoG\nPom/ttJBcS2nfddCvPc4qRm/z97WV+FlBtU9P4ySGSbuU+SObKhBLVEZdD9xyHw/\nYGojIxsAj2C6pahsjD94sfPN+Pve4MnGeAWvp5An0H95inDz0ewgcQW1U9gYwKBF\nDpMsWhwFAgMBAAECggEAO0lOXwnLmOy4IUq0kzsVPAmltFQwN/CdBP6FqTFiVETo\nAzg3rVvVCqwNQofkWyVmWwu891XPQr6oF6X1ZnWeTJblbw2kwNTbLjLwRVYgjzm0\njDtxoQMISZ9DqcCvjrDPxK1uJKeyXgXK5KN+t3KRkMga3nig7jAj2RELS66ZURWT\nC5ul8I1wZtRwIerussw038xkmSfhPJ3OxXovt0cMULbR/TCzWOs7442TEEQ8OVop\ng/hqbJyALiN0Je4Ba8v7Vpf2CMa+m/0XeCvwevsziKrjzsfSl5D2J7uaeaSH3YCt\nfNsKCl4kVI6p1nPCmb7/607jBiHULdumGGnZrguMSQKBgQDPQDattsZoXugsa2kN\nrx1Y+gSdR7O5CWDk41zc7YRBOlV+yZVU0Ql+NStYSUhoZNGcAtvoOE/T1b53CmWf\n2u8R9MOi1+/bha6lWjD9UhFzSC9If6ks8/GLlrQRBLcBtL8D65eR0VNr8jYh4Prj\nnWlvQNlEBnrVQiWAMP3HL+OoIwKBgQDIl7NdtyuXFmMLFEp/vBlbux5LgwtC3UkC\nlCNLRHC26Dd+ofYdH1msK8U+hbVxzeE68G5Z0uUZXkPvwKRBTP6G+G2Rkq0pn2iz\noxbPEo3k4YAU387Hs7bIE4ijBjavAoQXlX6a+KtWD/NyXNxoPITuWcY1/69oUTRY\nSzDaHGCZtwKBgQCp50usJ94KLFwMJlYuWLQngHgN9H07IZwkqNw1ZiZ5TvhOA7Z/\nnQs6Qx5wt0s8AgLw8ksT2i5fTKhjcMuVPGengdBsjLhfb/2YIxNsaT5TEwdoWPf5\nGVeZBwhBt2rMZEAAV8CfU0sfgKauKYGk3iRUh4ZWRTWn5R1XNVKnW/Q3nwKBgHj5\nJC9MnHlOAwVXg2Ztpk1wF+EqcswscC+h73chG2jzuyQfwTHnCkmPTDFcqgSsST0m\naya4zXnzI72fzG+lY4s6n7m/X7dOb/KGVOclJXX3mSr9ntNJSt53CVxoDuqU9dzT\nR7tmZM5OaF98KWCjvveQJL3EKaEb+BmXnyqrGocZAoGBAIMZviUGb7KlmuctERej\ntVSpE7WZLbXLykMFNsgiswsGXANmv8VvBguRLS0C5wCp3Xd0Erz9jld0erPNGiaf\n/8rRN2ibnMhV0BKjRNQHLFKfA24Wm2vHX7scB0Vj9fGpzQlW2UOV66m/zfX5DWua\n7X1bfrWIpyN5dv/oXkMXBnvF\n-----END PRIVATE KEY-----\n",
-    "client_email": "firebase-adminsdk-twa5w@learning-5fd92.iam.gserviceaccount.com",
-    "client_id": "116952036581323046961",
-    "auth_uri": "https://accounts.google.com/o/oauth2/auth",
-    "token_uri": "https://oauth2.googleapis.com/token",
-    "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
-    "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/firebase-adminsdk-twa5w%40learning-5fd92.iam.gserviceaccount.com",
-    "universe_domain": "googleapis.com"
-}
+# Initialize Firebase with your credentials
+cred = credentials.Certificate('credentials.json')
+initialize_app(cred, {'databaseURL': 'https://learning-5fd92-default-rtdb.asia-southeast1.firebasedatabase.app/'})
 
-# Initialize Firebase
-cred = credentials.Certificate(firebase_credentials)
-firebase_admin.initialize_app(cred, {
-    'databaseURL': 'https://learning-5fd92-default-rtdb.asia-southeast1.firebasedatabase.app/'
-})
-ref = db.reference("/")
-
-# Initialize AssemblyAI
+# Replace with your AssemblyAI API key
 aai.settings.api_key = "62acec891bb04c339ec059b738bedac6"
 
 # Initialize question answering pipeline
@@ -52,7 +33,7 @@ questions = [
     "Please mention if tooth decay is present with tooth number(s), else just say the word 'none' and nothing else",
     "Have any teeth been fractured? If yes, please mention the tooth number(s), else just say 'none' and nothing else",
     "Is there any pre-shedding mobility of teeth? If yes, please specify, else just say 'none' and nothing else",
-    "Does the child have malocclusion? If yes, please provide details, else just say the word 'none' and nothing",
+    "Does the child have malocclusion? If yes, please provide details, else just say the word 'none' and nothing else",
     "Does the child experience pain, swelling, or abscess? If yes, please provide details, else just say 'none' and nothing else",
     "Are there any other findings you would like to note?",
     "What treatment plan do you recommend? Choose only from Options: (Scaling, Filling, Pulp therapy/RCT, Extraction, Medication, Referral) and nothing else"
@@ -90,17 +71,15 @@ def generate_answer(question, context):
     return result['answer']
 
 # Function to handle audio recording and transcription
-def transcribe_audio(audio_path):
-    print(f"Received audio file at: {audio_path}")
-    
-    # Check if the file exists and is not empty
-    if not os.path.exists(audio_path):
-        return "Error: Audio file does not exist."
-    
-    if os.path.getsize(audio_path) == 0:
-        return "Error: Audio file is empty."
-    
+def transcribe_audio(audio_data):
     try:
+        # Save audio data to a temporary file
+        audio_path = '/tmp/audio.wav'
+        with open(audio_path, 'wb') as f:
+            f.write(audio_data)
+        
+        print(f"Saved audio file at: {audio_path}")
+
         # Transcribe the audio file using AssemblyAI
         transcriber = aai.Transcriber()
         print("Starting transcription...")
@@ -131,12 +110,16 @@ def fill_dataframe(context):
 
 # Function to push data to Firebase
 def push_to_firebase(data):
-    ref.push(data)
-    print("Data pushed to Firebase successfully.")
+    try:
+        ref = db.reference("/")
+        ref.push(data)
+        print("Data pushed to Firebase successfully.")
+    except Exception as e:
+        print(f"Error pushing data to Firebase: {e}")
 
 # Main Gradio app function
-def main(audio):
-    context = transcribe_audio(audio)
+def main(audio_data):
+    context = transcribe_audio(audio_data)
     
     if "Error" in context:
         return context
@@ -163,12 +146,12 @@ def main(audio):
     submit_button.click(fn=submit_data)
     
     return gr.Interface(
-        [audio],
+        [gr.Audio(type="bytes", label="Record your audio")],
         [output_html, submit_button],
         title="Audio Transcription and Question Answering App",
         live=False
     )
 
 # Launch the app
-main_app = main(gr.Audio(type="filepath", label="Record your audio"))
+main_app = main()
 main_app.launch()