Update app.py

app.py

@@ -17,22 +17,19 @@ def analyze_csv(file_path):
     results = df['text'].apply(lambda x: sentiment_pipeline(x)[0])
     df['sentiment'] = results.apply(lambda r: r['label'])
     df['score'] = results.apply(lambda r: r['score'])
-    
-    # Return the DataFrame as a CSV string
-    return df.to_csv(index=False)
 
-def gradio_analyze(file_obj):
-    # Get the path of the uploaded file and analyze it
-    file_path = file_obj.name
-    return analyze_csv(file_path)
+    # Save output to a new CSV file
+    output_csv_path = "output.csv"
+    df.to_csv(output_csv_path, index=False)
+    return output_csv_path  # Return path to the new CSV
 
 # Define the Gradio interface
 iface = gr.Interface(
-    fn=gradio_analyze,
+    fn=analyze_csv,
     inputs=gr.File(label="Upload CSV File", file_count="single", type="filepath"),
-    outputs=gr.Textbox(label="CSV with Sentiment Analysis"),
+    outputs=gr.File(label="Download CSV File"),
     title="CSV Sentiment Analysis App",
-    description="Upload a CSV file with a 'text' column. The app will run sentiment analysis on each row and return the CSV with sentiment labels and scores."
+    description="Upload a CSV file with a 'text' column. The app will run sentiment analysis on each row and return a downloadable CSV with sentiment labels and scores."
 )
 
 if __name__ == "__main__":