Update app.py

app.py

@@ -1,9 +1,11 @@
 from transformers import pipeline
 asr_pipe = pipeline("automatic-speech-recognition", model="Abdullah17/whisper-small-urdu")
+transcript_pipe = pipeline("automatic-speech-recognition", model="ihanif/whisper-medium-urdu")
 from difflib import SequenceMatcher
 import json
 import socket
-
+import soundfile as sf
+import gradio as gr
 def get_local_ip():
     try:
         # Create a socket connection to a remote host (here, google.com)
@@ -50,6 +52,7 @@ def find_most_similar_command(statement, command_list):
     return best_match,reply
 
 transcript_only=["1","3","4"]
+match_and_save=["2"]
 col_names={'1':"name",'3':"address",'4':"order"}
 def send_data_to_db(menu_id,col_value,order_id):
     import requests
@@ -71,19 +74,28 @@ def transcribe_the_command(audio,menu_id,order_id):
          print(f"Local IP Address: {local_ip}")
       else:
          print("Local IP could not be determined.")
-      import soundfile as sf
+
       sample_rate, audio_data = audio
       file_name = "recorded_audio.wav"
       sf.write(file_name, audio_data, sample_rate)
     # Convert stereo to mono by averaging the two channels
       print(menu_id)
 
-      transcript = asr_pipe(file_name)["text"]
       if menu_id in transcript_only:
+        transcript = transcript_pipe(file_name)["text"]
         col_value=transcript
         send_data_to_db(menu_id,col_value,order_id)
         print("data uploaded successfully!")
+      elif menu_id in match_and_save:
+        transcript = asr_pipe(file_name)["text"]
+        commands=urdu_data[menu_id]
+        most_similar_command,reply = find_most_similar_command(transcript, commands)
+        print(f"Given Statement: {transcript}")
+        print(f"Most Similar Command: {most_similar_command}\n")
+        print(reply)
+        send_data_to_db(menu_id,reply,order_id)
       else:
+        transcript = asr_pipe(file_name)["text"]
         commands=urdu_data[menu_id]
         print(commands)
         most_similar_command,reply = find_most_similar_command(transcript, commands)
@@ -92,7 +104,6 @@ def transcribe_the_command(audio,menu_id,order_id):
         print(reply)
         return reply
 # get_text_from_voice("urdu.wav")
-import gradio as gr
 
 
 iface = gr.Interface(