Update app.py

app.py

@@ -0,0 +1,71 @@
+import os
+import pandas as pd
+import numpy as np
+import matplotlib.pyplot as plt
+import gradio as gr
+from groq import Groq
+
+# Set the API key for Groq
+os.environ["GROQ_API_KEY"] = "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"
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+
+def analyze_file(uploaded_file):
+    # Load the file into a pandas DataFrame
+    if uploaded_file.name.endswith('.csv'):
+        df = pd.read_csv(uploaded_file.name)
+    elif uploaded_file.name.endswith('.xlsx'):
+        df = pd.read_excel(uploaded_file.name)
+
+    # Perform basic data analysis
+    if 'Gain (dB)' in df.columns and 'Frequency (GHz)' in df.columns:
+        mean_gain = df['Gain (dB)'].mean()
+        median_gain = df['Gain (dB)'].median()
+        std_dev_gain = df['Gain (dB)'].std()
+        
+        # Plot the data
+        fig, ax = plt.subplots()
+        ax.plot(df['Frequency (GHz)'], df['Gain (dB)'], label='Gain (dB)', color='blue')
+        ax.set_xlabel('Frequency (GHz)')
+        ax.set_ylabel('Gain (dB)')
+        ax.set_title('Gain vs Frequency')
+        ax.legend()
+        plt.close(fig)  # Close to avoid inline display
+        
+        # Send summary to Groq API for analysis
+        data_summary = f"""
+        The dataset contains simulation results for antennas. The frequency range is from 1 GHz to 10 GHz. 
+        - The antenna's gain increases from 5 dB to 30 dB as frequency increases. 
+        - Efficiency is consistently above 90%, with the highest reaching 99%.
+        """
+        
+        chat_completion = client.chat.completions.create(
+            messages=[{
+                "role": "user",
+                "content": f"Analyze this data summary: {data_summary}",
+            }],
+            model="llama-3.3-70b-versatile",  # Ensure this model is supported by Groq
+        )
+        
+        groq_analysis = chat_completion.choices[0].message.content
+        
+        return mean_gain, median_gain, std_dev_gain, fig, groq_analysis
+    else:
+        return "Required columns 'Gain (dB)' or 'Frequency (GHz)' not found in the dataset."
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=analyze_file,
+    inputs=gr.inputs.File(label="Upload CSV or Excel File"),
+    outputs=[
+        gr.outputs.Textbox(label="Mean Gain"),
+        gr.outputs.Textbox(label="Median Gain"),
+        gr.outputs.Textbox(label="Standard Deviation of Gain"),
+        gr.outputs.Image(label="Plot"),
+        gr.outputs.Textbox(label="Groq's Analysis")
+    ],
+    title="Data Analysis and Visualization with Groq API",
+    description="Upload a CSV or Excel file to analyze the antenna data and get insights."
+)
+
+# Launch the interface
+iface.launch()