Update app.py

app.py

@@ -7,7 +7,7 @@ import requests
 
 # Groq API Setup
 API_KEY = "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"
-API_URL = "https://api.groq.com/openai/v1/chat/completions"
+API_URL = "https://api.groq.com/openai/v1/chat/completions"  # Updated API URL
 
 def analyze_file(uploaded_file):
     try:
@@ -19,11 +19,15 @@ def analyze_file(uploaded_file):
         else:
             return "Error: The uploaded file is neither CSV nor Excel."
 
+        # Clean up the column names by stripping any leading/trailing spaces
+        df.columns = df.columns.str.strip()
+
         # Display the column names and first few rows for debugging
         print("Columns in the uploaded file:", df.columns)
         print("Preview of the uploaded data:", df.head())
-        st.write(df.columns)  # This will display column names in the UI for debugging
-        st.write(df.head())   # This will display the first few rows in the UI
+        # Display column names and preview in the Gradio UI for debugging
+        st.write(df.columns)  
+        st.write(df.head())   
 
         # Check if required columns are present
         if 'Gain (dB)' in df.columns and 'Frequency (GHz)' in df.columns:
@@ -31,7 +35,7 @@ def analyze_file(uploaded_file):
             mean_gain = df['Gain (dB)'].mean()
             median_gain = df['Gain (dB)'].median()
             std_dev_gain = df['Gain (dB)'].std()
-            
+
             # Display analysis results
             print(f"Mean Gain: {mean_gain}")
             print(f"Median Gain: {median_gain}")
@@ -52,7 +56,7 @@ def analyze_file(uploaded_file):
             - The antenna's gain increases from 5 dB to 30 dB as frequency increases. 
             - Efficiency is consistently above 90%, with the highest reaching 99%.
             """
-            
+
             headers = {
                 "Authorization": f"Bearer {API_KEY}",
                 "Content-Type": "application/json"
@@ -63,9 +67,8 @@ def analyze_file(uploaded_file):
                 "model": "llama-3.3-70b-versatile"  # Ensure this model is supported by Groq
             }
 
-
             # Send the request to Groq API
-            response = requests.post(API_URL, json=payload, headers=headers, timeout=30)  # 30 seconds timeout
+            response = requests.post(API_URL, json=payload, headers=headers)
             if response.status_code == 200:
                 groq_analysis = response.json()["choices"][0]["message"]["content"]
             else:
@@ -97,4 +100,3 @@ iface = gr.Interface(
 
 # Launch the interface
 iface.launch()
-