Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
| 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 |
)
|