Update app.py

app.py

@@ -21,15 +21,17 @@ def main():
     if uploaded_file is not None:
         # Read data
         data = pd.read_csv(uploaded_file)
-        # Check if 'Description' column exists
-        if 'Description' in data.columns:
-            # Apply translation and summarization
-            data['Summary'] = data['Description'].apply(translate_and_summarize)
+        # Check if 'Description' and 'Published' columns exist
+        if 'Description' in data.columns and 'Published' in data.columns:
+            # Apply translation and summarization based on 'Published' column
+            data['Summary'] = data.apply(
+                lambda row: translate_and_summarize(row['Description']) if pd.isna(row['Published']) else "", axis=1
+            )
             # Display data in a table
             st.write(data[['ID', 'Title', 'Summary']])
 
         else:
-            st.error("Uploaded CSV does not contain 'Description' column.")
+            st.error("Uploaded CSV does not contain required 'Description' and 'Published' columns.")
 
 if __name__ == "__main__":
-    main()
+    main()