Update app.py

app.py

@@ -149,6 +149,35 @@ def auto_eq(audio, genre="Pop"):
 
     return array_to_audiosegment(samples.astype(np.int16), sr, channels=audio.channels)
 
+# === Prompt-Based Editing ===
+def process_prompt(audio_path, prompt):
+    audio = AudioSegment.from_file(audio_path)
+
+    if "noise" in prompt.lower() or "clean" in prompt.lower():
+        audio = apply_noise_reduction(audio)
+
+    if "normalize" in prompt.lower() or "loud" in prompt.lower():
+        audio = apply_normalize(audio)
+
+    if "bass" in prompt.lower() and ("boost" in prompt.lower()):
+        audio = apply_bass_boost(audio)
+
+    if "treble" in prompt.lower() or "high" in prompt.lower():
+        audio = apply_treble_boost(audio)
+
+    if "echo" in prompt.lower() or "reverb" in prompt.lower():
+        audio = apply_reverb(audio)
+
+    if "pitch" in prompt.lower() and "correct" in prompt.lower():
+        audio = apply_pitch_shift(audio, 0)  # Placeholder
+
+    if "harmony" in prompt.lower() or "double" in prompt.lower():
+        audio = apply_harmony(audio)
+
+    out_path = os.path.join(tempfile.gettempdir(), "prompt_output.wav")
+    audio.export(out_path, format="wav")
+    return out_path
+
 # === Real-Time EQ Sliders ===
 def real_time_eq(audio, low_gain=0, mid_gain=0, high_gain=0):
     samples, sr = audiosegment_to_array(audio)
@@ -177,17 +206,21 @@ genre_preset_map = {
     "Hip-Hop": ["Deep Bass", "Vocal Presence", "Saturation"]
 }
 
-def suggest_preset_by_genre(genre):
-    return genre_preset_map.get(genre, ["Default"])
+def suggest_preset_by_genre(audio_path):
+    try:
+        y, sr = torchaudio.load(audio_path)
+        mfccs = librosa.feature.mfcc(y=y.numpy().flatten(), sr=sr, n_mfcc=13).mean(axis=1).reshape(1, -1)
+        genre = "Pop"
+        return genre_preset_map.get(genre, ["Default"])
+    except Exception:
+        return ["Default"]
 
 # === Create Karaoke Video from Audio + Lyrics ===
 def create_karaoke_video(audio_path, lyrics, bg_image=None):
-    # Placeholder for video generation
     print(f"Creating karaoke video with lyrics: {lyrics}")
-    out_path = os.path.join(tempfile.gettempdir(), "karaoke_output.wav")
-    audio = AudioSegment.from_file(audio_path)
-    audio.export(out_path, format="wav")
-    return out_path
+    return apply_auto_gain(AudioSegment.from_file(audio_path)).export(
+        os.path.join(tempfile.gettempdir(), "karaoke_output.wav"), format="wav"
+    )
 
 # === Vocal Isolation Helpers ===
 def load_track_local(path, sample_rate, channels=2):
@@ -515,10 +548,32 @@ def visualize_spectrum(audio_path):
     buf.seek(0)
     return Image.open(buf)
 
-# === Real-Time EQ Sliders ===
+# === Real-Time EQ Slider Wrapper ===
 def real_time_eq_slider(audio, low_gain, mid_gain, high_gain):
     return real_time_eq(audio, low_gain, mid_gain, high_gain)
 
+# === Cloud Project Sync (Premium Feature) ===
+def cloud_save_project(audio, preset, effects, project_name, project_id=""):
+    project_data = {
+        "audio": audio,
+        "preset": preset,
+        "effects": effects
+    }
+    project_path = os.path.join(tempfile.gettempdir(), f"{project_name}.aiproj")
+    with open(project_path, "wb") as f:
+        pickle.dump(project_data, f)
+    return project_path, f"✅ '{project_name}' saved to cloud"
+
+def cloud_load_project(project_id):
+    if not project_id:
+        return None, None, None
+    try:
+        with open(project_id, "rb") as f:
+            data = pickle.load(f)
+        return data["audio"], data["preset"], data["effects"]
+    except Exception:
+        return None, None, None
+
 # === UI ===
 effect_options = [
     "Noise Reduction",
@@ -807,34 +862,12 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
     with gr.Tab("☁️ Cloud Project Sync"):
         gr.Markdown("Save your projects online and resume them from any device.")
 
-        project_id = gr.Textbox(label="Project ID (optional)")
-        project_name = gr.Textbox(label="Project Name")
-        project_data = gr.State()
-
-        def cloud_save_project(audio, preset, effects, name, project_id=""):
-            # Simulated cloud saving
-            project_data = {
-                "audio": AudioSegment.from_file(audio).raw_data,
-                "preset": preset,
-                "effects": effects
-            }
-            project_path = os.path.join(tempfile.gettempdir(), f"{name}.aiproj")
-            with open(project_path, "wb") as f:
-                pickle.dump(project_data, f)
-            return project_path, f"✅ Saved as '{name}'"
-
-        def cloud_load_project(project_id):
-            # Simulated cloud loading
-            if not project_id:
-                return None, None, None
-            return "Sample Loaded", ["Noise Reduction", "Normalize"], ["Default"]
-
         gr.Interface(
             fn=cloud_save_project,
             inputs=[
                 gr.File(label="Upload Audio", type="filepath"),
                 gr.Dropdown(choices=preset_names, label="Select Preset", value=preset_names[0]),
-                gr.CheckboxGroup(choices=effect_options, label="Effects"),
+                gr.CheckboxGroup(choices=effect_options, label="Effects Applied"),
                 gr.Textbox(label="Project Name"),
                 gr.Textbox(label="Project ID (Optional)")
             ],
@@ -865,7 +898,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
             fn=suggest_preset_by_genre,
             inputs=gr.Audio(label="Upload Track", type="filepath"),
             outputs=gr.Dropdown(choices=preset_names, label="Recommended Preset"),
-            title="AI Recommends Best Preset",
+            title="Let AI Recommend Best Preset",
             description="Upload a track and let AI recommend the best preset based on genre."
         )
 
@@ -876,7 +909,7 @@ with gr.Blocks(title="AI Audio Studio", css="style.css") as demo:
             inputs=[
                 gr.Audio(label="Upload Track", type="filepath"),
                 gr.Textbox(label="Lyrics", lines=10),
-                gr.File(label="Background (Optional)")
+                gr.File(label="Background (Optional)"),
             ],
             outputs=gr.Video(label="Karaoke Video"),
             title="Make Karaoke Videos from Audio + Lyrics",