psaiops 0.0.1__tar.gz

psaiops-0.0.1/.github/README.md

@@ -0,0 +1,16 @@
+# psAI ops <img src="images/logo.png" alt="apehex logo" width="32" height="32">
+
+[![License][shield-license]][github-license]
+[![Latest][shield-release]][github-release]
+
+Collection of web apps to inspect & engineer activations.
+
+## License
+
+Licensed under the [aGPLv3][github-license].
+
+[github-license]: LICENSE.md
+[github-release]: https://github.com/apehex/psai-ops/releases/latest
+
+[shield-license]: https://img.shields.io/badge/license-aGPLv3-green?style=flat-square
+[shield-release]: https://img.shields.io/github/release/apehex/psai-ops.svg?style=flat-square

psaiops-0.0.1/PKG-INFO

@@ -0,0 +1,36 @@
+Metadata-Version: 2.3
+Name: psaiops
+Version: 0.0.1
+Summary: Web apps to inspect & engineer NN activations.
+License: .github/LICENSE.md
+Author: apehex
+Author-email: apehex@protonmail.com
+Requires-Python: >=3.10,<3.14
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: deformers (>=0.0)
+Requires-Dist: gradio (>=5.0)
+Requires-Dist: requests (>=2.0)
+Description-Content-Type: text/markdown
+
+# psAI ops <img src="images/logo.png" alt="apehex logo" width="32" height="32">
+
+[![License][shield-license]][github-license]
+[![Latest][shield-release]][github-release]
+
+Collection of web apps to inspect & engineer activations.
+
+## License
+
+Licensed under the [aGPLv3][github-license].
+
+[github-license]: LICENSE.md
+[github-release]: https://github.com/apehex/psai-ops/releases/latest
+
+[shield-license]: https://img.shields.io/badge/license-aGPLv3-green?style=flat-square
+[shield-release]: https://img.shields.io/github/release/apehex/psai-ops.svg?style=flat-square
+

psaiops-0.0.1/psaiops/elements/data.py

@@ -0,0 +1,17 @@
+import requests
+
+# CONSTANTS ####################################################################
+
+HF_URL = 'https://huggingface.co/api/quicksearch?q={target}&type={label}&limit={limit}'
+
+# HUGGING FACE #################################################################
+
+def query_huggingface(target: str, label: str='model', limit: int=16, endpoint: str=HF_URL) -> list:
+    __results = []
+    # query HF
+    __response = requests.get(endpoint.format(target=target, label=label, limit=limit))
+    # no error
+    if __response.status_code == 200:
+        __results = [__d['id'] for __d in __response.json().get(f'{label}s', [])]
+    # list of strings
+    return __results

psaiops-0.0.1/psaiops/elements/dropdown.py

@@ -0,0 +1,19 @@
+import gradio
+
+import psaiops.elements.data
+
+# AUTO-COMPLETE ################################################################
+
+def update_dropdown(query, label: str, data: gradio.KeyUpData):
+    # model_dropdown.key_up(fn=update_dropdown, inputs=[model_dropdown, gradio.State("model")], outputs=model_dropdown, queue=False, show_progress="hidden")
+    datasets = psaiops.elements.data.query_huggingface(target=data.input_value, label=label, limit=16)
+    return gradio.update(choices=datasets, visible=True)
+
+# with gradio.Blocks() as demo:
+#     model_dropdown = gradio.Dropdown(label="Models Auto-Complete", choices=[""], allow_custom_value=True)
+#     dataset_dropdown = gradio.Dropdown(label="Datasets Auto-Complete", choices=[""], allow_custom_value=True)
+#     spaces_dropdown = gradio.Dropdown(label="Spaces Auto-Complete", choices=[""], allow_custom_value=True)
+#     model_dropdown.key_up(fn=update_dropdown, inputs=[model_dropdown, gradio.State("model")], outputs=model_dropdown, queue=False, show_progress="hidden")
+#     dataset_dropdown.key_up(fn=update_dropdown, inputs=[dataset_dropdown, gradio.State("dataset")], outputs=dataset_dropdown, queue=False, show_progress="hidden")
+#     spaces_dropdown.key_up(fn=update_dropdown, inputs=[spaces_dropdown, gradio.State("space")], outputs=spaces_dropdown, queue=False, show_progress="hidden")
+# demo.launch(share=True, debug=True)

psaiops-0.0.1/psaiops/steer/app.py

@@ -0,0 +1,202 @@
+import gradio
+
+# MODEL ########################################################################
+
+def create_model_tab():
+    pass
+
+# PROMPT #######################################################################
+
+# SAMPLING #####################################################################
+
+class GradioInterface:
+    """Manages the Gradio user interface."""
+    
+    def __init__(self, orchestra: LLMForestOrchestra):
+        """Initialize the interface."""
+        self.orchestra = orchestra
+    
+    def create_interface(self) -> gr.Blocks:
+        """Create the Gradio interface."""
+        with gr.Blocks(title="LLM Forest Orchestra", theme=gr.themes.Soft()) as demo:
+            gr.Markdown(self.DESCRIPTION)
+            
+            with gr.Tabs():
+                with gr.TabItem("🎵 Generate Music"):
+                    self._create_generation_tab()
+            
+            return demo
+    
+    def _create_generation_tab(self):
+        """Create the main generation tab."""
+        with gr.Row():
+            with gr.Column(scale=1):
+                text_input = gr.Textbox(
+                    value=self.EXAMPLE_TEXT,
+                    label="Input Text",
+                    lines=8,
+                    placeholder="Enter text to sonify..."
+                )
+                
+                model_name = gr.Textbox(
+                    value=self.orchestra.DEFAULT_MODEL,
+                    label="Hugging Face Model",
+                    info="Model must support output_hidden_states and output_attentions"
+                )
+                
+                compute_mode = gr.Radio(
+                    choices=["Full model", "Mock latents"],
+                    value="Mock latents",
+                    label="Compute Mode",
+                    info="Mock latents for quick CPU-only demo"
+                )
+                
+                with gr.Row():
+                    instrument_preset = gr.Dropdown(
+                        choices=self.orchestra.instrument_manager.list_presets(),
+                        value="Ensemble (melody+bass+pad etc.)",
+                        label="Instrument Preset"
+                    )
+                    
+                    scale_choice = gr.Dropdown(
+                        choices=self.orchestra.scale_manager.list_scales() + ["Custom"],
+                        value="C pentatonic",
+                        label="Musical Scale"
+                    )
+                
+                custom_scale = gr.Textbox(
+                    value="",
+                    label="Custom Scale Notes",
+                    placeholder="60,62,65,67,70",
+                    visible=False
+                )
+                
+                with gr.Row():
+                    base_tempo = gr.Slider(
+                        120, 960,
+                        value=480,
+                        step=1,
+                        label="Tempo (ticks per beat)"
+                    )
+                    
+                    num_layers = gr.Slider(
+                        1, 6,
+                        value=6,
+                        step=1,
+                        label="Max Layers"
+                    )
+                
+                with gr.Row():
+                    velocity_low = gr.Slider(
+                        1, 126,
+                        value=40,
+                        step=1,
+                        label="Min Velocity"
+                    )
+                    
+                    velocity_high = gr.Slider(
+                        2, 127,
+                        value=90,
+                        step=1,
+                        label="Max Velocity"
+                    )
+                
+                seed = gr.Number(
+                    value=42,
+                    precision=0,
+                    label="Random Seed"
+                )
+                
+                generate_btn = gr.Button(
+                    "🎼 Generate MIDI",
+                    variant="primary",
+                    size="lg"
+                )
+            
+            with gr.Column(scale=1):
+                midi_output = gr.File(
+                    label="Generated MIDI File",
+                    file_types=[".mid", ".midi"]
+                )
+                
+                stats_display = gr.Markdown(label="Quick Stats")
+                
+                metadata_json = gr.Code(
+                    label="Metadata (JSON)",
+                    language="json"
+                )
+                
+                with gr.Row():
+                    play_instructions = gr.Markdown(
+                        """
+                        ### 🎧 How to Play
+                        1. Download the MIDI file
+                        2. Open in any DAW or MIDI player
+                        3. Adjust instruments and effects as desired
+                        4. Export to audio format
+                        """
+                    )
+        
+        # Set up interactions
+        def update_custom_scale_visibility(choice):
+            return gr.update(visible=(choice == "Custom"))
+        
+        scale_choice.change(
+            update_custom_scale_visibility,
+            inputs=[scale_choice],
+            outputs=[custom_scale]
+        )
+        
+        def generate_wrapper(
+            text, model_name, compute_mode, base_tempo,
+            velocity_low, velocity_high, scale_choice,
+            custom_scale, num_layers, instrument_preset, seed
+        ):
+            """Wrapper for generation with error handling."""
+            try:
+                # Parse custom scale if needed
+                custom_notes = None
+                if scale_choice == "Custom" and custom_scale:
+                    custom_notes = [int(x.strip()) for x in custom_scale.split(",")]
+                
+                # Generate
+                filename, metadata = self.orchestra.generate(
+                    text=text,
+                    model_name=model_name,
+                    compute_mode=compute_mode,
+                    base_tempo=int(base_tempo),
+                    velocity_range=(int(velocity_low), int(velocity_high)),
+                    scale_name=scale_choice,
+                    custom_scale_notes=custom_notes,
+                    num_layers=int(num_layers),
+                    instrument_preset=instrument_preset,
+                    seed=int(seed)
+                )
+                
+                # Format stats
+                stats = metadata.get("stats", {})
+                stats_text = f"""
+                ### Generation Statistics
+                - **Layers Used**: {stats.get('num_layers', 'N/A')}
+                - **Tokens Processed**: {stats.get('num_tokens', 'N/A')}
+                - **Total Notes**: {stats.get('total_notes', 'N/A')}
+                - **Notes per Layer**: {stats.get('notes_per_layer', [])}
+                - **Scale**: {stats.get('scale', [])}
+                - **Tempo**: {stats.get('tempo_ticks_per_beat', 'N/A')} ticks/beat
+                """
+                
+                return filename, stats_text, json.dumps(metadata, indent=2)
+                
+            except Exception as e:
+                error_msg = f"### ❌ Error\n{str(e)}"
+                return None, error_msg, json.dumps({"error": str(e)}, indent=2)
+        
+        generate_btn.click(
+            fn=generate_wrapper,
+            inputs=[
+                text_input, model_name, compute_mode, base_tempo,
+                velocity_low, velocity_high, scale_choice,
+                custom_scale, num_layers, instrument_preset, seed
+            ],
+            outputs=[midi_output, stats_display, metadata_json]
+        )

psaiops-0.0.1/pyproject.toml

@@ -0,0 +1,21 @@
+[tool.poetry]
+name = "psaiops"
+version = "0.0.1"
+description = "Web apps to inspect & engineer NN activations."
+license = ".github/LICENSE.md"
+readme = ".github/README.md"
+authors = ["apehex <apehex@protonmail.com>"]
+packages = [{include = "psaiops"}]
+
+[tool.poetry.dependencies]
+python = ">=3.10, <3.14"
+deformers = ">=0.0"
+gradio = ">=5.0"
+requests = ">=2.0"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "*"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"