Update app.py

app.py

@@ -18,6 +18,12 @@ def calculate_microstrip_patch(frequency, permittivity, thickness):
     patch_width = wavelength / (2 * np.sqrt(1 + permittivity))
     return patch_length, patch_width, thickness
 
+def calculate_dipole_antenna(frequency):
+    c = 3e8  # Speed of light in m/s
+    wavelength = c / frequency
+    dipole_length = wavelength / 2  # Length of dipole antenna is half wavelength
+    return dipole_length
+
 def calculate_s11(frequency):
     # Simulate S11 (just an example function for demonstration)
     s11 = -20 + 5 * np.cos(2 * np.pi * frequency / 10)
@@ -56,6 +62,16 @@ def plot_3d_microstrip_patch(patch_length, patch_width, thickness):
     fig.update_layout(title="3D Microstrip Patch Antenna", showlegend=False)
     return fig
 
+def plot_3d_dipole_antenna(dipole_length):
+    fig = go.Figure()
+    fig.add_trace(go.Scatter3d(
+        x=[0, dipole_length], y=[0, 0], z=[0, 0], marker=dict(size=5, color='red'),
+        line=dict(color='red', width=4), name="Dipole Antenna"
+    ))
+    fig.update_layout(title="3D Dipole Antenna", scene=dict(
+        xaxis_title='Length (m)', yaxis_title='Y', zaxis_title='Z'))
+    return fig
+
 def plot_s11_graph(frequencies, s11_values):
     fig = go.Figure()
     fig.add_trace(go.Scatter(x=frequencies, y=s11_values, mode='lines', name="S11"))
@@ -99,8 +115,28 @@ def design_antenna(antenna_type, frequency, permittivity, thickness):
             f"S11 at Operating Frequency: {s11_values[len(s11_values)//2]:.2f} dB\n"
             f"Patch Dimensions: {patch_length:.2f} m x {patch_width:.2f} m x {thickness:.2f} m\n"
         )
+    
+    elif antenna_type == "Dipole":
+        dipole_length = calculate_dipole_antenna(frequency_hz)
+        s11_values = [calculate_s11(f) for f in frequencies]
+        directivities, gains = zip(*[calculate_directivity_and_gain(f) for f in frequencies])
+        theta = np.linspace(-180, 180, 360)
+        gain_pattern = radiation_pattern(theta, frequency_hz)
+        
+        s11_graph = plot_s11_graph(frequencies, s11_values)
+        directivity_gain_graph = plot_directivity_and_gain(frequencies, directivities, gains)
+        radiation_graph = plot_radiation_pattern(theta, gain_pattern)
+        antenna_3d = plot_3d_dipole_antenna(dipole_length)
+        
+        output = (
+            f"Design Type: Dipole Antenna\n"
+            f"Operating Frequency: {frequency:.2f} GHz\n"
+            f"S11 at Operating Frequency: {s11_values[len(s11_values)//2]:.2f} dB\n"
+            f"Dipole Length: {dipole_length:.2f} m\n"
+        )
+        
     else:
-        output = "Currently, only Microstrip Patch Antenna is supported."
+        output = "Currently, only Microstrip Patch and Dipole Antenna are supported."
         s11_graph, directivity_gain_graph, radiation_graph, antenna_3d = None, None, None, None
     
     return output, s11_graph, directivity_gain_graph, radiation_graph, antenna_3d
@@ -108,7 +144,7 @@ def design_antenna(antenna_type, frequency, permittivity, thickness):
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Antenna Design Tool with Groq API")
-    antenna_type = gr.Dropdown(["Microstrip Patch"], label="Select Antenna Type")
+    antenna_type = gr.Dropdown(["Microstrip Patch", "Dipole"], label="Select Antenna Type")
     frequency = gr.Slider(1.0, 10.0, step=0.1, label="Operating Frequency (GHz)")
     permittivity = gr.Number(value=4.4, label="Substrate Permittivity")
     thickness = gr.Number(value=0.01, label="Substrate Thickness (m)")
@@ -125,4 +161,4 @@ with gr.Blocks() as demo:
         outputs=[output_text, s11_plot, directivity_gain_plot, radiation_pattern_plot, antenna_3d_display]
     )
 
-demo.launch()
+demo.launch()