cutted 0.3.3__tar.gz → 0.4.0__tar.gz

{cutted-0.3.3 → cutted-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cutted
-Version: 0.3.3
+Version: 0.4.0
 Summary: AI-powered audio editor controllable via natural language.
 Author-email: simon0302010 <simon0302010@gmail.com>
 License-Expression: GPL-3.0

cutted-0.4.0/cutted/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "0.4.0"
+__author__ = "simon0302010"

{cutted-0.3.3 → cutted-0.4.0}/cutted/app.py

@@ -109,45 +109,54 @@ class CuttedApp:
     def update_plot(self):
         if self.canvas:
             self.canvas.get_tk_widget().destroy()
-        
+        if hasattr(self, "slider") and self.slider is not None:
+            self.slider.destroy()
+
         fig = self.AudioProcessor.plot_audio()
         self.ax = fig.axes[0]
         self.canvas = FigureCanvasTkAgg(fig, master=self.plot_frame)
         self.canvas.draw()
-        
-        self.audio_lenght = int(round(self.AudioProcessor.get_length()))
-        
+
+        self.audio_length = float(self.AudioProcessor.get_length())
+
         slider_width = self.root.winfo_width() - 40
+        if self.slider_value > self.audio_length:
+            self.slider_value = self.audio_length
+        if self.slider_value < 0:
+            self.slider_value = 0
+
         self.slider = customtkinter.CTkSlider(
-            self.root, from_=0, to=self.audio_lenght, command=self.set_cursor, width=slider_width
+            self.root, from_=0, to=self.audio_length, command=self.set_cursor, width=slider_width
         )
-        self.slider.set(0)
-        
+        self.slider.set(self.slider_value)
         self.slider.place(relx=0.5, rely=1.0, anchor="s", y=-130)
-        
+        self.set_cursor(self.slider_value)
+
         self.canvas.get_tk_widget().pack(
             fill=customtkinter.BOTH, 
             expand=True, 
             padx=10,
             pady=10,
         )
-        
-        self.cursor_line = self.ax.axvline(x=0, color="red", linewidth=2)
+
+        self.cursor_line = self.ax.axvline(x=self.slider_value, color="red", linewidth=2)
         self.canvas.draw_idle()
         
     def set_cursor(self, value):
         now = time.time()
-        if now - self.last_slider_update < 0.1:  # 100ms
+        if now - self.last_slider_update < 0.05:  # 100ms
             return
         self.last_slider_update = now
         
-        self.slider_value = round(value)
+        self.slider_value = round(value, 2)
         
         if self.cursor_line:
             self.cursor_line.set_xdata([self.slider_value, self.slider_value])
             self.canvas.draw_idle()
+            self.slider.set(self.slider_value)
+            self.set_cursor(self.slider_value)
         
-        print(f"Slider Value: {self.slider_value}")
+        print_info(f"Slider Value: {self.slider_value}")
 
     def play_audio(self):
         if not hasattr(self.AudioProcessor, "audio") or self.AudioProcessor.audio is None:
@@ -200,7 +209,7 @@ class CuttedApp:
         
         text = self.entry.get()
         if text.strip():
-            full_prompt = f"You are a audio editing AI. You are controllable via natural language and editing a audio file. The audio file is {round(self.AudioProcessor.get_length())}s long. The cursor of the user is currently at {self.slider_value}s."
+            full_prompt = f"You are a audio editing AI. You are controllable via natural language and editing a audio file. The audio file is {round(self.AudioProcessor.get_length(), 2)}s long. The cursor of the user is currently at {self.slider_value}s."
             full_prompt += "\nHere is a the waveform samples of the audio. You can use them to determine silent parts, loud parts, silences, beats and much more.\nYou are forced to used these if the user requires you to cut out silent of quiet parts for example."
             full_prompt += "\nAll of your tools should be enough to fullfill almost every task.\nNEVER ASK FOR CONFIRMATION FROM THE USER. DO EVERYTHING!"
             full_prompt += f"\n{self.AudioProcessor.get_waveform_summary()}\n"
@@ -228,6 +237,12 @@ class CuttedApp:
                     result = self.AudioProcessor.cut(args["start"], args["end"])
                     if not result:
                         messagebox.showerror("Error", "Please try again.")
+                if function_call.name == "change_volume":
+                    print_info("Change Volume function called")
+                    args = function_call.args
+                    result = self.AudioProcessor.change_volume(args["start"], args["end"], args["volume"])
+                    if not result:
+                        messagebox.showerror("Error", "Please try again.")
                 self.update_plot()
             elif text_result:
                 messagebox.showerror("Error", text_result.strip())

{cutted-0.3.3 → cutted-0.4.0}/cutted/core/audio_processor.py

@@ -112,6 +112,28 @@ class AudioProcessor:
         else:
             return False
         
+    def change_volume(self, start, end, volume):
+        if len(start) == len(end) == len(volume):
+            time_sets = list(zip(start, end, volume))
+            for single_start, single_end, single_volume in time_sets:
+                if single_end <= single_start:
+                    print_fail("End time must be greater than start time.")
+                    continue
+                print_info(f"Changing volume of {single_start} - {single_end} to {str(single_volume)}")
+                
+                start_ms = round(single_start * 1000)
+                end_ms = round(single_end * 1000)
+                part1 = self.audio[:start_ms]
+                part2 = self.audio[start_ms:end_ms]
+                part3 = self.audio[end_ms:]
+                
+                part2 = part2.apply_gain(ratio_to_db(single_volume))
+                
+                self.audio = part1 + part2 + part3
+            return True
+        else:
+            return False
+            
     def play_audio(self, start_time=0):
         if self.audio is None:
             print_fail("No audio loaded.")

cutted-0.4.0/cutted/core/gemini.py

@@ -0,0 +1,149 @@
+import os
+import sys
+import base64
+from google import genai
+from google.genai import types
+
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+
+if not GEMINI_API_KEY:
+    print("Please set the environment variable GEMINI_API_KEY to your Gemini API Key.")
+    sys.exit(0)
+
+class GeminiClient:
+    def __init__(self):
+        self.client = genai.Client(
+            api_key=GEMINI_API_KEY,
+        )
+        self.contents = []
+    
+    def generate(self, prompt: str, model: str = "gemini-2.0-flash", audio_base64 = None):
+        parts=[
+            types.Part.from_text(text=prompt),
+        ]
+        
+        if audio_base64:
+            parts.append(types.Part.from_bytes(
+                mime_type="audio/mpeg",
+                data=base64.b64decode(audio_base64)
+            ))
+        
+        self.contents.append(
+            types.Content(
+                role="user",
+                parts=parts
+            )
+        )
+        tools = [
+            types.Tool(
+                function_declarations=[
+                    types.FunctionDeclaration(
+                        name="cut_audio",
+                        description=(
+                            "Remove one or more segments from the audio by specifying start and end times in seconds. "
+                            "You can cut multiple segments at once by providing lists of start and end values. "
+                            "Each segment defined by a start and end pair will be removed from the audio."
+                        ),
+                        parameters=genai.types.Schema(
+                            type=genai.types.Type.OBJECT,
+                            required=["start", "end"],
+                            properties={
+                                "start": genai.types.Schema(
+                                    type=genai.types.Type.ARRAY,
+                                    items=genai.types.Schema(
+                                        type=genai.types.Type.NUMBER,
+                                    ),
+                                ),
+                                "end": genai.types.Schema(
+                                    type=genai.types.Type.ARRAY,
+                                    items=genai.types.Schema(
+                                        type=genai.types.Type.NUMBER,
+                                    ),
+                                ),
+                            },
+                        ),
+                    ),
+                    types.FunctionDeclaration(
+                        name="change_volume",
+                        description=(
+                            "Adjust the volume of specific segments in the audio by specifying lists of start times, end times, "
+                            "and volume factors. Each segment between a start and end time will have its volume changed by the "
+                            "corresponding factor (e.g., 0.5 for half volume, 2.0 for double volume). Multiple segments can be "
+                            "adjusted at once by providing lists of values."
+                        ),
+                        parameters=genai.types.Schema(
+                            type=genai.types.Type.OBJECT,
+                            required=["start", "end", "volume"],
+                            properties={
+                                "start": genai.types.Schema(
+                                    type=genai.types.Type.ARRAY,
+                                    items=genai.types.Schema(
+                                        type=genai.types.Type.NUMBER,
+                                    ),
+                                ),
+                                "end": genai.types.Schema(
+                                    type=genai.types.Type.ARRAY,
+                                    items=genai.types.Schema(
+                                        type=genai.types.Type.NUMBER,
+                                    ),
+                                ),
+                                "volume": genai.types.Schema(
+                                    type=genai.types.Type.ARRAY,
+                                    items=genai.types.Schema(
+                                        type=genai.types.Type.NUMBER,
+                                    ),
+                                ),
+                            },
+                        ),
+                    ),
+                ]
+            )
+        ]
+        generate_content_config = types.GenerateContentConfig(
+            tools=tools,
+            response_mime_type="text/plain",
+        )
+
+        response = self.client.models.generate_content(
+            model=model,
+            contents=self.contents,
+            config=generate_content_config,
+        )
+
+        function_call = None
+        text_response = None
+        try:
+            for candidate in response.candidates:
+                for part in candidate.content.parts:
+                    if part.function_call:
+                        function_call = part.function_call
+                    if part.text:
+                        text_response = part.text
+        except TypeError:
+            pass
+        
+        model_parts = []
+        if text_response:
+            model_parts.append(
+                types.Part.from_text(text=text_response)
+            )
+        if function_call:
+            model_parts.append(
+                types.Part.from_function_call(
+                    name=function_call.name,
+                    args=function_call.args
+                )
+            )
+            
+        self.contents.append(
+            types.Content(
+                role="model",
+                parts=model_parts
+            )
+        )
+
+        return function_call, text_response
+        
+if __name__ == "__main__":
+    gemini = GeminiClient()
+    print(gemini.generate("cut from 10 to 20.5"))

{cutted-0.3.3 → cutted-0.4.0}/cutted.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cutted
-Version: 0.3.3
+Version: 0.4.0
 Summary: AI-powered audio editor controllable via natural language.
 Author-email: simon0302010 <simon0302010@gmail.com>
 License-Expression: GPL-3.0

{cutted-0.3.3 → cutted-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cutted"
-version = "0.3.3"
+version = "0.4.0"
 description = "AI-powered audio editor controllable via natural language."
 readme = "README.md"
 requires-python = ">=3.9, <=3.13"

cutted-0.3.3/cutted/__init__.py

@@ -1,2 +0,0 @@
-__version__ = "0.3.3"
-__author__ = "simon0302010"

cutted-0.3.3/cutted/core/gemini.py

@@ -1,90 +0,0 @@
-import os
-import sys
-import base64
-from google import genai
-from google.genai import types
-
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
-
-if not GEMINI_API_KEY:
-    print("Please set the environment variable GEMINI_API_KEY to your Gemini API Key.")
-    sys.exit(0)
-
-class GeminiClient:
-    def __init__(self):
-        self.client = genai.Client(
-            api_key=GEMINI_API_KEY,
-        )
-    
-    def generate(self, prompt: str, model: str = "gemini-2.0-flash", audio_base64 = None):
-        parts=[
-            types.Part.from_text(text=prompt),
-        ]
-        
-        if audio_base64:
-            parts.append(types.Part.from_bytes(
-                mime_type="audio/mpeg",
-                data=base64.b64decode(audio_base64)
-            ))
-        
-        contents = [
-            types.Content(
-                role="user",
-                parts=parts
-            ),
-        ]
-        tools = [
-            types.Tool(
-                function_declarations=[
-                    types.FunctionDeclaration(
-                        name="cut_audio",
-                        description="Cuts specified parts out of audio. Multiple parts can be cut if a list of both start and end values is used as property.",
-                        parameters=genai.types.Schema(
-                            type = genai.types.Type.OBJECT,
-                            required = ["start", "end"],
-                            properties = {
-                                "start": genai.types.Schema(
-                                    type = genai.types.Type.ARRAY,
-                                    items = genai.types.Schema(
-                                        type = genai.types.Type.NUMBER,
-                                    ),
-                                ),
-                                "end": genai.types.Schema(
-                                    type = genai.types.Type.ARRAY,
-                                    items = genai.types.Schema(
-                                        type = genai.types.Type.NUMBER,
-                                    ),
-                                    ),
-                            },
-                        ),
-                    ),
-                ])
-        ]
-        generate_content_config = types.GenerateContentConfig(
-            tools=tools,
-            response_mime_type="text/plain",
-        )
-
-        response = self.client.models.generate_content(
-            model=model,
-            contents=contents,
-            config=generate_content_config,
-        )
-
-        function_call = None
-        text_response = None
-        try:
-            for candidate in response.candidates:
-                for part in candidate.content.parts:
-                    if part.function_call:
-                        function_call = part.function_call
-                    if part.text:
-                        text_response = part.text
-        except TypeError:
-            return None, None
-
-        return function_call, text_response
-        
-if __name__ == "__main__":
-    gemini = GeminiClient()
-    print(gemini.generate("cut from 10 to 20.5"))