Create app.py

app.py

@@ -0,0 +1,92 @@
+from flask import Flask, request, render_template, jsonify
+import json
+import copy
+
+app = Flask(__name__)
+
+# Basic phoneme classification
+VOWELS = {'a', 'e', 'i', 'o', 'u', 'æ', 'ɪ', 'ʊ', 'ə', 'ɛ', 'ɔ', 'ʌ', 'ɑ', 'ɒ', 'ɘ', 'ɚ', 'ɜ', 'ɞ', 'ɨ', 'ɵ', 'ɶ', 'ʉ', 'ʏ'}
+CONSONANTS = {'b', 'c', 'd', 'f', 'h', 'j', 'k', 'l', 'm', 'n', 'p', 'q', 'r', 's', 't', 'v', 'w', 'x', 'y', 'z', 'ç', 'ð', 'ħ', 'ŋ', 'ɓ', 'ɕ', 'ɖ', 'ɗ', 'ɟ', 'ɠ', 'ɡ', 'ɢ', 'ɣ', 'ɥ', 'ɦ', 'ɧ', 'ɫ', 'ɬ', 'ɭ', 'ɮ', 'ɰ', 'ɱ', 'ɲ', 'ɳ', 'ɴ', 'ɸ', 'ɹ', 'ɺ', 'ɻ', 'ɽ', 'ɾ', 'ʀ', 'ʁ', 'ʂ', 'ʃ', 'ʄ', 'ʈ', 'ʋ', 'ʍ', 'ʎ', 'ʐ', 'ʑ', 'ʒ', 'ʔ', 'ʕ', 'ʘ', 'ʙ', 'ʛ', 'ʜ', 'ʝ', 'ʟ', 'ʡ', 'ʢ', 'ⱱ', 'θ', 'χ', 'β'}
+STRESS_MARKERS = {'ˈ', 'ˌ'}
+
+def classify_phoneme(phoneme):
+    if phoneme in VOWELS:
+        return 'vowel'
+    elif phoneme in CONSONANTS:
+        return 'consonant'
+    elif phoneme in STRESS_MARKERS:
+        return 'stress'
+    return 'special'
+
+def adjust_phonemes(model_data, emotion):
+    modified_data = copy.deepcopy(model_data)
+    phoneme_id_map = modified_data['phoneme_id_map']
+    num_symbols = modified_data['num_symbols']
+    
+    new_phoneme_id_map = {}
+    
+    for phoneme, id_list in phoneme_id_map.items():
+        base_id = id_list[0]  # Assume single ID per phoneme
+        phoneme_type = classify_phoneme(phoneme)
+        
+        if emotion == 'happy':
+            if phoneme_type == 'vowel':
+                new_id = min(num_symbols - 1, base_id + 5)
+            elif phoneme_type == 'consonant':
+                new_id = min(num_symbols - 1, base_id + 2)
+            elif phoneme_type == 'stress':
+                new_id = min(num_symbols - 1, base_id + 3)
+            else:
+                new_id = base_id
+        elif emotion == 'sad':
+            if phoneme_type == 'vowel':
+                new_id = max(0, base_id - 5)
+            elif phoneme_type == 'consonant':
+                new_id = max(0, base_id - 2)
+            elif phoneme_type == 'stress':
+                new_id = max(0, base_id - 1)
+            else:
+                new_id = base_id
+        elif emotion == 'angry':
+            if phoneme_type == 'vowel':
+                new_id = min(num_symbols - 1, base_id + 3)
+            elif phoneme_type == 'consonant':
+                new_id = min(num_symbols - 1, base_id + 5)
+            elif phoneme_type == 'stress':
+                new_id = min(num_symbols - 1, base_id + 10)
+            else:
+                new_id = base_id
+        
+        new_phoneme_id_map[phoneme] = [new_id]
+    
+    modified_data['phoneme_id_map'] = new_phoneme_id_map
+    return modified_data
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/process_json', methods=['POST'])
+def process_json():
+    try:
+        data = request.get_json()
+        model_json = data.get('model_json', '')
+        
+        # Parse the input JSON
+        model_data = json.loads(model_json)
+        
+        # Define emotions to process
+        emotions = ['happy', 'sad', 'angry']
+        results = {}
+        
+        # Generate modified JSON for each emotion
+        for emotion in emotions:
+            modified_data = adjust_phonemes(model_data, emotion)
+            results[emotion] = json.dumps(modified_data, indent=2)
+        
+        return jsonify({'status': 'success', 'results': results})
+    except Exception as e:
+        return jsonify({'status': 'error', 'message': str(e)}), 400
+
+if __name__ == '__main__':
+    app.run(host="0.0.0.0",port=7850, debug=True)