Update app.py

app.py

@@ -77,14 +77,23 @@ def main():
                 width: 500px;
                 height: 50px;
             }}
-            .my-container {{
-                border: 2px solid #d72324;
-                padding: 10px;
+            .stDownloadButton > button{{
+                background-color: #e7e7e7;
+                color:black;
+                font-weight: bold; 
+                width: 150px;
+                height: 35px;
+                float: right;
             }}
             .stButton > button:hover {{
                 background-color: white;
                 color:#d72324;
             }}
+            .my-container {{
+                border: 2px solid #d72324;
+                padding: 10px;
+            }}
+           
            
         </style>
         """,
@@ -95,7 +104,9 @@ def main():
     #https://www.freepnglogos.com/uploads/youtube-play-red-logo-png-transparent-background-6.png
     # st.write("Enter the video details below:")
 
-    # Define a boolean flag variable to track prediction status
+    @st.cache_data 
+    def convert_df(df):
+        return df.to_csv(index=False).encode('utf-8')
   
     # Sidebar menu options
     menu_options = ["Predict", "Trending", "Visualize"]
@@ -167,6 +178,15 @@ def main():
                 df = get_trending_videos(country_code)
                 st.dataframe(df)
                 
+                csv = convert_df(df)
+                st.download_button(
+                    "Download",
+                    csv,
+                    "file.csv",
+                    "text/csv",
+                    key='download-csv'
+                )
+                
             with tab2:
                 if df is not None:
                     # Display video titles
@@ -221,9 +241,7 @@ def main():
                     show_top_titleLength()
                 with col13:
                     show_best_titleLength()
-
-
-                
+                    
 def get_picture_from_url(url):
     try:
         response = requests.get(url)