Update functions.py

functions.py

@@ -15,7 +15,8 @@ from streamlit import secrets
 # Constants
 GITHUB_API = "https://api.github.com"
 # No default TOKEN assignment here, we will handle it dynamically
-
+# Replace with your actual OpenAI API key
+openai.api_key = secrets['GITHUB_API_TOKEN']
 # Function to update headers with a new token
 def get_headers(updated_token=None):
     # Use the provided token or fallback to the existing secret token
@@ -78,16 +79,16 @@ def extract_commit_summary(commit_data):
         'message': commit_details.get('message', ''),
         'author': author_info.get('name', ''),
         'date': author_info.get('date', ''),  # Include commit date
-        'file_patches': [
-            {'filename': file.get('filename', ''), 'patch': file.get('patch', '')}
-            for file in files_info if 'patch' in file
-        ],
-        'stats': {
-            'additions': commit_details.get('stats', {}).get('additions', 0),
-            'deletions': commit_details.get('stats', {}).get('deletions', 0),
-            'total': commit_details.get('stats', {}).get('total', 0)
-        }
     }
+        # 'file_patches': [
+        #     {'filename': file.get('filename', ''), 'patch': file.get('patch', '')}
+        #     for file in files_info if 'patch' in file
+        # ],
+        # 'stats': {
+        #     'additions': commit_details.get('stats', {}).get('additions', 0),
+        #     'deletions': commit_details.get('stats', {}).get('deletions', 0),
+        #     'total': commit_details.get('stats', {}).get('total', 0)
+        # }
 
     return commit_summary
 
@@ -166,3 +167,33 @@ def get_contributor_commits(repo_name, contributor, period,branches, updated_tok
     summarized_commits = [extract_commit_summary(commit) for commit in all_commits]
     return summarized_commits
 
+def chat_complete(message, model="gpt-3.5-turbo"):
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=[
+            {"role": "user", "content": str(message)}
+        ]
+    )
+    return response
+
+def openai_get_AI_summary(commit_info):
+    prompt_prefix = "Summarize this git commit given the following context in JSON:\n"
+    prompt_suffix = "\n---\nPut the summary in the following JSON schema: {...}"
+    
+    # Construct the message with commit_info
+    message = prompt_prefix + json.dumps(commit_info, indent=2) + prompt_suffix
+
+    # Fetch the summary from OpenAI
+    response = chat_complete(message)
+    return response.choices[0].message['content'] if response.choices else "No summary available"
+
+def process_commits_with_openai_summaries(commits):
+    summaries = []
+
+    for commit in tqdm(commits, desc='Generating AI summaries'):
+        # Assuming each commit includes 'commit_files_stats' with detailed info
+        summary = openai_get_AI_summary(str(commit))
+        # commit['ai_summary'] = summary
+        summaries.append(summary)
+
+    return summaries