init commit

README.md

@@ -5,9 +5,12 @@ colorFrom: yellow
 colorTo: blue
 sdk: gradio
 sdk_version: 5.33.0
+python_version: 3.12
 app_file: app.py
 pinned: false
 license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Bode Plotter
+
+> Written by KYLiN

app.py

@@ -0,0 +1,144 @@
+import gradio as gr
+import numpy as np
+from control.matlab import tf, pole, zero, frequency_response, bode
+from sympy.polys.polytools import poly_from_expr
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
+
+
+def poly_handler(poly_str: str) -> list[float]:
+    try:
+        if poly_str.isdigit():
+            poly_coffs = [poly_str]
+        else:
+            poly = poly_from_expr(poly_str)[0]
+            poly_coffs = poly.all_coeffs()
+
+        return list(map(float, poly_coffs))
+
+    except Exception as e:
+        raise ValueError(f"Error parsing polynomial: {e}")
+
+
+def bode_information(poly_upper: str, poly_lower: str) -> tuple:
+    upper_poly_coffs = poly_handler(poly_upper)
+    lower_poly_coffs = poly_handler(poly_lower)
+
+    num, den = np.array(upper_poly_coffs), np.array(lower_poly_coffs)
+
+    try:
+        trans_func_g = tf(num, den)
+    except Exception as e:
+        raise ValueError(f"Error creating transfer function: {e}")
+
+    pole_result = pole(trans_func_g)
+    zero_result = zero(trans_func_g)
+    mag, phase, omega = frequency_response(trans_func_g)
+
+    return mag, phase, omega
+
+
+def to_latex(poly_upper: str, poly_lower: str) -> str:
+    upper_poly_coffs = poly_handler(poly_upper)
+    lower_poly_coffs = poly_handler(poly_lower)
+
+    def process_item(i: float, c: float):
+        c = f"+ {c}" if c >= 0 else f"- {-c}"
+
+        if i == 0:
+            return f"{c}"
+        if i == 1:
+            return f"{c} s"
+
+        return f"{c} s^{i}"
+
+    upper_str = "".join(
+        [process_item(i, c) for i, c in enumerate(upper_poly_coffs[::-1]) if c != 0]
+    )
+    lower_str = "".join(
+        [process_item(i, c) for i, c in enumerate(lower_poly_coffs[::-1]) if c != 0]
+    )
+
+    if upper_str.startswith("+ "):
+        upper_str = upper_str[2:]
+    if lower_str.startswith("+ "):
+        lower_str = lower_str[2:]
+
+    latex_str = r"\frac{" + upper_str + r"}{" + lower_str + r"}"
+
+    return latex_str
+
+
+def bode_graph(mag: np.ndarray, phase: np.ndarray, omega: np.ndarray):
+    mag_db = 20 * np.log10(mag)
+    fig = make_subplots(
+        rows=1,
+        cols=2,
+        subplot_titles=(
+            "Bode Plot - Magnitude",
+            "Bode Plot - Phase",
+        ),
+    )
+
+    fig.add_trace(
+        go.Scatter(x=omega, y=mag_db, mode="lines", name="Magnitude"), row=1, col=1
+    )
+    fig.add_trace(
+        go.Scatter(x=omega, y=phase, mode="lines", name="Phase"), row=1, col=2
+    )
+
+    fig.update_yaxes(title_text="Magnitude (dB)", row=1, col=1)
+    fig.update_yaxes(title_text="Phase (degrees)", row=1, col=2)
+    fig.update_xaxes(type="log", title_text="Frequency (rad/s)", row=1, col=1)
+    fig.update_xaxes(type="log", title_text="Frequency (rad/s)", row=1, col=2)
+    fig.update_layout(title_text="Bode Plot")
+
+    return fig
+
+
+def run_it(upper_poly: str, lower_poly: str):
+    mag, phase, omega = bode_information(upper_poly, lower_poly)
+    latex_str = to_latex(upper_poly, lower_poly)
+    fig = bode_graph(mag, phase, omega)
+
+    return fig, f"$$H(s)={latex_str}$$"
+
+
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Bode Plotter")
+        gr.Markdown("This app allows you to plot Bode plots for transfer functions.")
+        gr.Markdown(
+            "Enter the transfer function in the form of a string, e.g., '1/(s^2 + 2*s + 1)'."
+        )
+        upper_input = gr.Textbox(
+            label="Numerator Polynomial",
+            placeholder="Enter the numerator polynomial (e.g., '1 - 2*s + 3*s^2')",
+            lines=2,
+        )
+        lower_input = gr.Textbox(
+            label="Denominator Polynomial",
+            placeholder="Enter the denominator polynomial (e.g., '1 + 4*s + 5*s^2')",
+            lines=2,
+        )
+        run_btn = gr.Button("Plot Bode", variant="primary")
+        gr.Markdown("### Bode Plot")
+
+        with gr.Column():
+            latex_output = gr.Markdown(label="Transfer Function LaTeX")
+            bode_plot = gr.Plot(label="Bode Plot")
+
+        run_btn.click(
+            run_it,
+            inputs=[
+                upper_input,
+                lower_input,
+            ],
+            outputs=[bode_plot, latex_output],
+        )
+
+    demo.launch()
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -0,0 +1,14 @@
+[project]
+name = "bode-plotter"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "black>=25.1.0",
+    "control>=0.10.1",
+    "gradio>=5.33.0",
+    "loguru>=0.7.3",
+    "plotly>=6.1.2",
+    "sympy>=1.14.0",
+]

requirements.txt

@@ -0,0 +1,71 @@
+aiofiles==24.1.0
+annotated-types==0.7.0
+anyio==4.9.0
+black==25.1.0
+certifi==2025.4.26
+charset-normalizer==3.4.2
+click==8.2.1
+contourpy==1.3.2
+control==0.10.1
+cycler==0.12.1
+fastapi==0.115.12
+ffmpy==0.6.0
+filelock==3.18.0
+fonttools==4.58.1
+fsspec==2025.5.1
+gradio==5.33.0
+gradio-client==1.10.2
+groovy==0.1.2
+h11==0.16.0
+hf-xet==1.1.3
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.32.4
+idna==3.10
+jinja2==3.1.6
+kiwisolver==1.4.8
+loguru==0.7.3
+markdown-it-py==3.0.0
+markupsafe==3.0.2
+matplotlib==3.10.3
+mdurl==0.1.2
+mpmath==1.3.0
+mypy-extensions==1.1.0
+narwhals==1.41.0
+numpy==2.2.6
+orjson==3.10.18
+packaging==25.0
+pandas==2.3.0
+pathspec==0.12.1
+pillow==11.2.1
+platformdirs==4.3.8
+plotly==6.1.2
+pydantic==2.11.5
+pydantic-core==2.33.2
+pydub==0.25.1
+pygments==2.19.1
+pyparsing==3.2.3
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.2
+pyyaml==6.0.2
+requests==2.32.3
+rich==14.0.0
+ruff==0.11.13
+safehttpx==0.1.6
+scipy==1.15.3
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+starlette==0.46.2
+sympy==1.14.0
+tomlkit==0.13.3
+tqdm==4.67.1
+typer==0.16.0
+typing-extensions==4.14.0
+typing-inspection==0.4.1
+tzdata==2025.2
+urllib3==2.4.0
+uvicorn==0.34.3
+websockets==15.0.1