Update app.py

app.py

@@ -29,17 +29,18 @@ def main():
         # Read data
         data = pd.read_csv(uploaded_file)
         # Check if 'Description' and 'Published' columns exist
-        if 'Description' in data.columns and 'Published' in data.columns:
-            # Apply full translation and summarization based on 'Published' column
-            data['Full Summary'] = data.apply(
-                lambda row: translate_and_summarize(row['Description']) if pd.isna(row['Published']) else "", axis=1
-            )
-            # Apply first paragraph translation and summarization
-            data['First Paragraph Summary'] = data['Description'].apply(translate_and_summarize_first_paragraph)
+        if 'Description' in data.columns:
+            # Apply full translation and summarization
+            data['DescriptionEn'] = data['Description'].apply(translate_and_summarize)
+            # Apply first paragraph translation and summarization for Title
+            data['Title'] = data['Description'].apply(translate_and_summarize_first_paragraph)
             # Display data in a table
-            st.write(data[['ID', 'Title', 'Full Summary', 'First Paragraph Summary']])
+            st.write(data[['ID', 'Title', 'DescriptionEn']])
+            # Provide download link for the updated CSV
+            csv = data.to_csv(index=False)
+            st.download_button(label="Download updated CSV", data=csv, file_name="updated_data.csv", mime="text/csv")
         else:
-            st.error("Uploaded CSV does not contain required 'Description' and 'Published' columns.")
+            st.error("Uploaded CSV does not contain required 'Description' column.")
 
 if __name__ == "__main__":
     main()