Update app.py

app.py

@@ -10,41 +10,18 @@ load_dotenv()
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
 # Functions for Antenna Calculations
-def calculate_microstrip_patch(frequency, permittivity, thickness, tangent_loss, impedance):
+def calculate_microstrip_patch(frequency, permittivity, thickness, tangent_loss):
     c = 3e8  # Speed of light in m/s
     wavelength = c / frequency
     effective_wavelength = wavelength / np.sqrt(permittivity)
     patch_length = effective_wavelength / 2
     patch_width = wavelength / (2 * np.sqrt(1 + permittivity))
-    
-    # Adjustments for tangent loss (approximation)
-    adjustment_factor = 1 + tangent_loss / 10
-    patch_length *= adjustment_factor
-    patch_width *= adjustment_factor
-    
-    # Consider impedance adjustment (this will be a simplified factor here)
-    if impedance == 50:
-        impedance_factor = 1.0
-    elif impedance == 73:
-        impedance_factor = 1.1
-    else:
-        impedance_factor = 1.0
-    
-    patch_length *= impedance_factor
-    patch_width *= impedance_factor
-    
     return patch_length, patch_width, thickness
 
-def calculate_dipole_antenna(frequency, impedance):
+def calculate_dipole(frequency):
     c = 3e8  # Speed of light in m/s
     wavelength = c / frequency
-    dipole_length = wavelength / 2  # Length of dipole antenna is half wavelength
-    
-    # Consider impedance adjustments (simplified)
-    if impedance == 50:
-        dipole_length *= 1.0
-    elif impedance == 73:
-        dipole_length *= 1.05  # A minor adjustment for 73 ohms impedance
+    dipole_length = wavelength / 2
     return dipole_length
 
 def calculate_s11(frequency):
@@ -85,16 +62,6 @@ 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"))
@@ -121,7 +88,7 @@ def design_antenna(antenna_type, frequency, permittivity, thickness, tangent_los
     frequencies = np.linspace(frequency - 0.5, frequency + 0.5, 100)
     
     if antenna_type == "Microstrip Patch":
-        patch_length, patch_width, thickness = calculate_microstrip_patch(frequency_hz, permittivity, thickness, tangent_loss, impedance)
+        patch_length, patch_width, thickness = calculate_microstrip_patch(frequency_hz, permittivity, thickness, tangent_loss)
         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)
@@ -137,12 +104,10 @@ def design_antenna(antenna_type, frequency, permittivity, thickness, tangent_los
             f"Operating Frequency: {frequency:.2f} GHz\n"
             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"
-            f"Substrate Tangent Loss: {tangent_loss:.2f}\n"
-            f"Input Impedance: {impedance} Ohms\n"
+            f"Input Impedance: {impedance} Ohms"
         )
-    
     elif antenna_type == "Dipole":
-        dipole_length = calculate_dipole_antenna(frequency_hz, impedance)
+        dipole_length = calculate_dipole(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)
@@ -151,19 +116,15 @@ def design_antenna(antenna_type, frequency, permittivity, thickness, tangent_los
         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)
+        antenna_3d = None  # Dipole does not have a 3D plot
         
         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"
-            f"Input Impedance: {impedance} Ohms\n"
+            f"Input Impedance: {impedance} Ohms"
         )
-        
-    else:
-        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
 
@@ -175,4 +136,18 @@ with gr.Blocks() as demo:
     permittivity = gr.Number(value=4.4, label="Substrate Permittivity")
     thickness = gr.Number(value=0.01, label="Substrate Thickness (m)")
     tangent_loss = gr.Number(value=0.02, label="Substrate Tangent Loss (tan δ)", step=0.01)
-    impedance = gr.Dropdown([50, 73], label="Input Impedance (Ohms)", value
+    impedance = gr.Dropdown([50, 73], label="Input Impedance (Ohms)", value=50)
+    design_button = gr.Button("Design Antenna")
+    output_text = gr.Textbox(label="Design Results")
+    s11_plot = gr.Plot(label="S11 Plot")
+    directivity_gain_plot = gr.Plot(label="Directivity and Gain Plot")
+    radiation_pattern_plot = gr.Plot(label="Radiation Pattern")
+    antenna_3d_display = gr.Plot(label="3D Antenna Visualization")
+
+    design_button.click(
+        design_antenna,
+        inputs=[antenna_type, frequency, permittivity, thickness, tangent_loss, impedance],
+        outputs=[output_text, s11_plot, directivity_gain_plot, radiation_pattern_plot, antenna_3d_display]
+    )
+
+demo.launch()