tanish78 commited on
Commit
c6ae83d
·
verified ·
1 Parent(s): ace8a63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -165,18 +165,17 @@ def visualize_clusters(df):
165
 
166
  def main(file, num_clusters):
167
  try:
168
-
169
  df = pd.read_excel(file)
170
-
171
-
172
  df = preprocess_data(df)
173
  df = cluster_data(df, num_clusters)
174
  visualize_clusters(df)
 
 
 
 
175
 
176
-
177
- return df
178
  except Exception as e:
179
-
180
  return str(e)
181
 
182
  interface = gr.Interface(
@@ -185,7 +184,7 @@ interface = gr.Interface(
185
  gr.File(label="Upload Excel File (.xlsx)"),
186
  gr.Number(value=5, label="Number of Clusters")
187
  ],
188
- outputs=gr.Dataframe(label="Clustered Data"),
189
  title="Unanswered User Queries Clustering",
190
  description="Upload an Excel file (.xlsx)"
191
  )
 
165
 
166
  def main(file, num_clusters):
167
  try:
 
168
  df = pd.read_excel(file)
 
 
169
  df = preprocess_data(df)
170
  df = cluster_data(df, num_clusters)
171
  visualize_clusters(df)
172
+
173
+ output = BytesIO()
174
+ df.to_csv(output, index=False)
175
+ output.seek(0)
176
 
177
+ return output
 
178
  except Exception as e:
 
179
  return str(e)
180
 
181
  interface = gr.Interface(
 
184
  gr.File(label="Upload Excel File (.xlsx)"),
185
  gr.Number(value=5, label="Number of Clusters")
186
  ],
187
+ outputs=gr.File(label="Download Clustered Data as CSV"),
188
  title="Unanswered User Queries Clustering",
189
  description="Upload an Excel file (.xlsx)"
190
  )