Update app.py

app.py

@@ -4,6 +4,8 @@ from dotenv import load_dotenv
 import os
 import requests
 import hashlib
+from git import Repo, exc
+
 from crewai import Agent, Task, Crew
 from langchain.tools import tool
 
@@ -16,6 +18,41 @@ GITHUB_ACCESS_TOKEN = userdata.get('GITHUB_ACCESS_TOKEN')
 # Initialize the OpenAI client
 openai_client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
 
+def check_for_updates(repo_path, remote_name='origin', branch='main'):
+    """
+    Check for updates in the specified Git repository using GitPython.
+    Input: repo_path - Path to the local git repository.
+    Output: A string indicating if updates are available or not.
+    """
+    try:
+        # Open the existing repository
+        repo = Repo(repo_path)
+        if repo.bare:
+            raise Exception("Repository is bare.")
+
+        # Fetch updates from the remote repository
+        remote = repo.remotes[remote_name]
+        fetch_info = remote.fetch()
+
+        # Compare local and remote branches
+        local_commit = repo.heads[branch].commit
+        remote_commit = repo.refs[f'refs/remotes/{remote_name}/{branch}'].commit
+
+        if local_commit.hexsha != remote_commit.hexsha:
+            return "Updates are available."
+        else:
+            return "No updates available."
+            
+    except exc.InvalidGitRepositoryError:
+        return "Invalid Git repository."
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Example usage
+# repo_path = "/path/to/your/streamlit-repo"
+# print(check_for_updates(repo_path))
+        
+
 # Define the custom tool for GitHub interaction
 @tool
 def github_streamlit_expert() -> str:
@@ -96,10 +133,16 @@ if st.button("Ask"):
 for chat in chat_history:
     st.text(chat)
     
-# Manual update button
-if st.button('Check for Streamlit Updates Now'):
-    result = crew.kickoff()  # Trigger the Crew to perform the update task
-    st.write(result)
-else:
-    st.write('Click the button above to check for updates.')
+# Button to check for updates in the Streamlit GitHub repository
+if st.button('Check for Streamlit GitHub Updates'):
+    repo_path = "/path/to/your/streamlit-repo"  # Path to your local Streamlit repository
+    update_status = check_for_updates(repo_path)
+    st.write(update_status)
+
+    if "Updates are available" in update_status:
+        pass
+        # Optionally, trigger further actions like updating the local repo or analyzing changes
+        # For example:
+        # response = analyze_new_changes(repo_path)
+        # st.write(response)