Update app.py

app.py

@@ -166,10 +166,6 @@ def archive_all_data(client):
     except Exception as e:
         return f"An error occurred while archiving data: {str(e)}"
 
-# Streamlit app
-st.title("🌟 Cosmos DB and GitHub Integration")
-
-
 
 # Modify the main app
 def main():
@@ -190,7 +186,10 @@ def main():
     # Login section
     if not st.session_state.logged_in:
         st.subheader("🔐 Login")
-        input_key = st.text_input("Enter your Cosmos DB key", type="password")
+        #input_key = st.text_input("Enter your Cosmos DB key", type="password")
+        input_key=Key
+
+        # Cosmos DB configuration
         if st.button("🚀 Login"):
             if input_key:
                 st.session_state.primary_key = input_key
@@ -247,8 +246,51 @@ def main():
             if st.session_state.selected_container:
                 df = pd.DataFrame(documents)
                 st.dataframe(df)
+                    
+            # GitHub section
+            st.subheader("🐙 GitHub Operations")
+            github_token = os.environ.get("GITHUB")  # Read GitHub token from environment variable
+            source_repo = st.text_input("Source GitHub Repository URL", value="https://github.com/AaronCWacker/AIExamples-8-24-Streamlit")
+            new_repo_name = st.text_input("New Repository Name (for cloning)", value=f"AIExample-Clone-{datetime.now().strftime('%Y%m%d_%H%M%S')}")
             
-            # ... (keep the rest of your existing code for GitHub operations, etc.)
+            col1, col2 = st.columns(2)
+            with col1:
+                if st.button("📥 Clone Repository"):
+                    if github_token and source_repo:
+                        try:
+                            local_path = f"./temp_repo_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
+                            download_github_repo(source_repo, local_path)
+                            zip_filename = f"{new_repo_name}.zip"
+                            create_zip_file(local_path, zip_filename[:-4])
+                            st.markdown(get_base64_download_link(zip_filename, zip_filename), unsafe_allow_html=True)
+                            st.success("Repository cloned successfully!")
+                        except Exception as e:
+                            st.error(f"An error occurred: {str(e)}")
+                        finally:
+                            if os.path.exists(local_path):
+                                shutil.rmtree(local_path)
+                            if os.path.exists(zip_filename):
+                                os.remove(zip_filename)
+                    else:
+                        st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided.")
+        
+            with col2:
+                if st.button("📤 Push to New Repository"):
+                    if github_token and source_repo:
+                        try:
+                            g = Github(github_token)
+                            new_repo = create_repo(g, new_repo_name)
+                            local_path = f"./temp_repo_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
+                            download_github_repo(source_repo, local_path)
+                            push_to_github(local_path, new_repo, github_token)
+                            st.success(f"Repository pushed successfully to {new_repo.html_url}")
+                        except Exception as e:
+                            st.error(f"An error occurred: {str(e)}")
+                        finally:
+                            if os.path.exists(local_path):
+                                shutil.rmtree(local_path)
+                    else:
+                        st.error("Please ensure GitHub token is set in environment variables and source repository URL is provided.")
 
         except exceptions.CosmosHttpResponseError as e:
             st.error(f"Failed to connect to Cosmos DB. HTTP error: {str(e)}. Status code: {e.status_code}")