Update app.py

app.py

@@ -11,11 +11,11 @@ API_URL = "https://api.groq.com/openai/v1/chat/completions"  # Updated API URL
 
 def analyze_file(uploaded_file):
     try:
-        # Load the file into a pandas DataFrame from the uploaded file (in memory)
+        # Load the file into a pandas DataFrame
         if uploaded_file.name.endswith('.csv'):
-            df = pd.read_csv(uploaded_file)  # Read directly from memory
+            df = pd.read_csv(uploaded_file)
         elif uploaded_file.name.endswith('.xlsx'):
-            df = pd.read_excel(uploaded_file)  # Read directly from memory
+            df = pd.read_excel(uploaded_file)
         else:
             return "Error: The uploaded file is neither CSV nor Excel."
 
@@ -28,10 +28,14 @@ def analyze_file(uploaded_file):
 
         # Check if required columns are present
         if 'Gain (dB)' in df.columns and 'Frequency (GHz)' in df.columns:
+            # Convert pandas columns to numpy arrays before performing operations
+            gain_values = np.array(df['Gain (dB)'])
+            freq_values = np.array(df['Frequency (GHz)'])
+
             # Perform basic data analysis
-            mean_gain = df['Gain (dB)'].mean()
-            median_gain = df['Gain (dB)'].median()
-            std_dev_gain = df['Gain (dB)'].std()
+            mean_gain = np.mean(gain_values)
+            median_gain = np.median(gain_values)
+            std_dev_gain = np.std(gain_values)
 
             # Display analysis results
             st.write(f"Mean Gain: {mean_gain}")
@@ -40,7 +44,7 @@ def analyze_file(uploaded_file):
 
             # Plotting the data
             fig, ax = plt.subplots()
-            ax.plot(df['Frequency (GHz)'], df['Gain (dB)'], label='Gain (dB)', color='blue')
+            ax.plot(freq_values, gain_values, label='Gain (dB)', color='blue')
             ax.set_xlabel('Frequency (GHz)')
             ax.set_ylabel('Gain (dB)')
             ax.set_title('Gain vs Frequency')