add json

app/main.py

@@ -25,6 +25,28 @@ import opensmile
 import ffmpeg
 import noisereduce as nr
 
+import json
+
+# Path to the JSON file
+json_filepath = 'reciters.json'
+
+
+def load_json_data(filepath):
+    """Load JSON data from a file."""
+    with open(filepath, 'r', encoding='utf-8') as file:
+        return json.load(file)
+
+# Load the JSON data from file
+json_reciters = load_json_data(json_filepath)
+
+def find_reciter_by_name(name):
+    """Search for a reciter by name in the loaded JSON data."""
+    for reciter in json_reciters['reciters']:
+        if reciter['name'] == name:
+            return reciter
+    return None  # Return None if no match is found
+
+
 
 default_sample_rate=22050
 
@@ -302,22 +324,23 @@ async def handle_cnn(file: UploadFile = File(...)):
 
         # Make predictions
         predictions = cnn_model.predict(X)
-        print('predictions', predictions)
+        print('predictions', np.argmax(predictions))
 
         # Convert predictions to label indexes
-        predicted_label_indexes = np.argmax(predictions, axis=1)
+        predicted_label_indexes = np.argmax(predictions)
 
         # Convert label indexes to actual label names
-        predicted_labels = cnn_label_encoder.inverse_transform(predicted_label_indexes)
+        predicted_labels = cnn_label_encoder.inverse_transform([predicted_label_indexes])
 
         print('decoded', predicted_labels)
-        print('arg', np.argmax(predicted_labels))
-        sheikh_name = np.argmax(predicted_labels)
-        # .tolist()
+        reciter_name = predicted_labels[0]
+        # Find the reciter by name
+        reciter_object = find_reciter_by_name(reciter_name)
+
         # Clean up the temporary file
         os.remove(temp_filename)
         # Return a successful response with decoded predictions
-        return {"message": "File processed successfully", "sheikh": sheikh_name}
+        return {"message": "File processed successfully", "sheikh": reciter_object}
     except Exception as e:
         print(e)
         # Handle possible exceptions

requirements.txt

@@ -15,6 +15,7 @@ librosa
 soundfile
 opensmile
 eyeD3
+json
 matplotlib
 python-multipart
 ffmpeg-python