Update app.py

app.py

@@ -10,7 +10,7 @@ load_dotenv()
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
 # Functions for Antenna Calculations
-def calculate_microstrip_patch(frequency, permittivity, thickness, tangent_loss):
+def calculate_microstrip_patch(frequency, permittivity, thickness, tangent_loss, impedance):
     c = 3e8  # Speed of light in m/s
     wavelength = c / frequency
     effective_wavelength = wavelength / np.sqrt(permittivity)
@@ -18,17 +18,33 @@ def calculate_microstrip_patch(frequency, permittivity, thickness, tangent_loss)
     patch_width = wavelength / (2 * np.sqrt(1 + permittivity))
     
     # Adjustments for tangent loss (approximation)
-    # The tangent loss will slightly affect the width and length of the patch
     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):
+def calculate_dipole_antenna(frequency, impedance):
     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
     return dipole_length
 
 def calculate_s11(frequency):
@@ -100,12 +116,12 @@ def plot_radiation_pattern(theta, gain_pattern):
     return fig
 
 # Main Function
-def design_antenna(antenna_type, frequency, permittivity, thickness, tangent_loss):
+def design_antenna(antenna_type, frequency, permittivity, thickness, tangent_loss, impedance):
     frequency_hz = frequency * 1e9
     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)
+        patch_length, patch_width, thickness = calculate_microstrip_patch(frequency_hz, permittivity, thickness, tangent_loss, impedance)
         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)
@@ -122,10 +138,11 @@ def design_antenna(antenna_type, frequency, permittivity, thickness, tangent_los
             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"
         )
     
     elif antenna_type == "Dipole":
-        dipole_length = calculate_dipole_antenna(frequency_hz)
+        dipole_length = calculate_dipole_antenna(frequency_hz, impedance)
         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)
@@ -141,6 +158,7 @@ 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"Dipole Length: {dipole_length:.2f} m\n"
+            f"Input Impedance: {impedance} Ohms\n"
         )
         
     else:
@@ -157,17 +175,4 @@ 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)
-    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],
-        outputs=[output_text, s11_plot, directivity_gain_plot, radiation_pattern_plot, antenna_3d_display]
-    )
-
-demo.launch()
+    impedance = gr.Dropdown([50, 73], label="Input Impedance (Ohms)", value