Completed Code Whisperer

Author: jeeshumittal7

.env

@@ -1,3 +1,10 @@
 PORT=3001
 MONGODB_URI=mongodb://localhost:27017/codewhisperer
-OPENAI_API_KEY={your_openai_api_key}
+OPENAI_API_KEY={your_openai_api_key}
+BASE_URL=http://localhost:3001
+EMAIL_HOST=smtp.sendgrid.net
+EMAIL_PORT=587
+EMAIL_FROM={your_validated_sending_identity}
+EMAIL_USER={username}
+EMAIL_PASSWORD={password}
+EMAIL_SECURE=false

README.md

@@ -1,33 +1,34 @@
 # CodeWhisperer
 
-CodeWhisperer is a Node.js application leveraging Express and MongoDB, with a Bootstrap-based UI. It facilitates conversation with code repositories by summarizing files and projects using OpenAI's gpt-3.5-turbo-16k and gpt-4-turbo-preview models.
+CodeWhisperer is an innovative Node.js application that allows users to have conversations with their code repositories. By leveraging the power of OpenAI's AI models, it can summarize text files and entire code repositories to facilitate a chat-like interaction.
 
 ## Overview
 
-The app utilizes Express.js for server-side operations and MongoDB as a data storage solution. It uses the Bootstrap framework to create a responsive UI. The application operates by accepting GitHub repository URLs and email addresses, cloning repositories, extracting and summarizing textual content, and storing summaries in MongoDB. Email notifications are sent through Sendgrid upon task completion.
+The application is built using Express for server management, MongoDB for data persistence, and Bootstrap for the front-end interface. It primarily interacts with OpenAI's API to create summaries of repositories and engage users in conversations about their code. The app processes GitHub repositories by cloning them, generating summaries through the gpt-3.5-turbo-16k and gpt-4-turbo-preview AI models, and then communicates the results via email using Sendgrid.
 
 ## Features
 
-- Accept GitHub repository URLs and emails on the main page
-- Clone, summarize, and delete repositories upon processing
-- Interact with OpenAI's API to generate code and project summaries
-- Email users with links to interact with their repository summaries
-- Display project summary and offer a chat-like interface for Q&A powered by OpenAI
+- Accepts GitHub repository URLs and user email addresses for processing
+- Clones repositories and summarizes text files and overall project contents
+- Communicates with OpenAI's API to generate summaries and answer user queries
+- Sends email notifications with links to interact with the repository summary
+- Provides a chat interface for dynamic interaction with the project summary
 
 ## Getting started
 
 ### Requirements
 
-- Node.js
-- MongoDB
-- An OpenAI API key
+- Node.js environment
+- MongoDB instance
+- OpenAI API key
+- Sendgrid credentials for email notifications
 
 ### Quickstart
 
-1. Clone the repository to your local machine.
-2. Install dependencies with `npm install`.
-3. Set up your `.env` file with the necessary environment variables (PORT, MONGODB_URI, and OPENAI_API_KEY).
-4. Run the server using `npm start` or `node server.js`.
+1. Clone the repository to your machine.
+2. Install necessary Node.js packages with `npm install`.
+3. Configure the required environment variables within an `.env` file.
+4. Run the application using `npm start` or `node server.js`.
 
 ### License
 

chatService.js

@@ -0,0 +1,47 @@
+const OpenAI = require('openai');
+const Repository = require('./repository');
+require('dotenv').config();
+
+const openai = new OpenAI(process.env.OPENAI_API_KEY);
+
+async function interactWithRepository(uuid, userMessage) {
+  try {
+    const repository = await Repository.findOne({ uuid: uuid }).exec();
+    if (!repository || !repository.isProcessed) {
+      throw new Error('Repository not found or not processed yet.');
+    }
+
+    // gpt_pilot_debugging_log
+    console.log('Repository file summaries:', repository.fileSummaries);
+
+    const systemMessage = {
+      role: 'system',
+      content: `This is a summary of the project: ${repository.summary}`
+    };
+    const userMessageObj = {
+      role: 'user',
+      content: userMessage
+    };
+
+    const response = await openai.chat.completions.create({
+      model: 'gpt-4-turbo-preview',
+      messages: [systemMessage, ...(repository.fileSummaries || []).map(summary => ({ role: 'system', content: summary })), userMessageObj],
+      max_tokens: 1024,
+      temperature: 0.5
+    });
+
+    if (!response.choices || response.choices.length === 0 || !response.choices[0].message) {
+      throw new Error('Invalid response from OpenAI');
+    }
+
+    return response.choices[0].message.content.trim();
+  } catch (error) {
+    // gpt_pilot_debugging_log
+    console.error('Error in interactWithRepository:', error.message, error.stack);
+    throw error;
+  }
+}
+
+module.exports = {
+  interactWithRepository
+};

emailService.js

@@ -0,0 +1,40 @@
+const nodemailer = require('nodemailer');
+require('dotenv').config();
+
+const transporter = nodemailer.createTransport({
+  host: process.env.EMAIL_HOST,  // Use the SMTP server host from .env
+  port: parseInt(process.env.EMAIL_PORT, 10), // Use the SMTP server port from .env
+  secure: process.env.EMAIL_SECURE === 'true', // Convert EMAIL_SECURE string to boolean
+  auth: {
+    user: process.env.EMAIL_USER,     // Authentication email address from .env
+    pass: process.env.EMAIL_PASSWORD  // Email password from .env
+  },
+  tls: {
+    rejectUnauthorized: false
+  }
+});
+
+async function sendEmailNotification(email, uuid, repoUrl) {
+  console.log(`Preparing to send email notification to ${email}`); // gpt_pilot_debugging_log
+
+  const mailOptions = {
+    from: process.env.EMAIL_FROM, // Sender address from .env
+    to: email,
+    subject: 'Your repo is ready for a chat!',
+    html: `The repository <a href="${repoUrl}">${repoUrl}</a> has been analyzed. You can chat with it at <a href="${process.env.BASE_URL}/explain/${uuid}">${process.env.BASE_URL}/explain/${uuid}</a>.`
+  };
+
+  console.log(`Sending email to ${email} with subject: "${mailOptions.subject}"`); // gpt_pilot_debugging_log
+
+  try {
+    const result = await transporter.sendMail(mailOptions);
+    console.log(`Email sent to ${email} with response: ${JSON.stringify(result)}`); // gpt_pilot_debugging_log
+  } catch (error) {
+    console.error(`Failed to send email notification: ${error}`, error.stack); // gpt_pilot_debugging_log
+    throw error;
+  }
+}
+
+module.exports = {
+  sendEmailNotification
+};

package.json

@@ -14,11 +14,11 @@
     "body-parser": "^1.20.2",
     "bootstrap": "^5.3.2",
     "dotenv": "^16.4.1",
+    "ejs": "^3.1.9",
     "express": "^4.18.2",
     "fs-extra": "^11.2.0",
     "mongoose": "^8.1.1",
     "nodemailer": "^6.9.9",
-    "nodemailer-sendgrid-transport": "^0.2.0",
     "openai": "^4.26.1",
     "simple-git": "^3.22.0",
     "tmp": "^0.2.1",

processRepository.js

@@ -1,6 +1,7 @@
 const Repository = require('./repository');
 const { cloneAndProcessRepo } = require('./gitHandler');
 const { generateSummary } = require('./openaiService');
+const { sendEmailNotification } = require('./emailService');
 const fs = require('fs-extra');
 const OpenAI = require('openai');
 require('dotenv').config();
@@ -26,19 +27,22 @@ async function processRepoInBackground(githubUrl, email) {
   try {
     const { processedFiles, tempDirPath: dirPath } = await cloneAndProcessRepo(githubUrl);
     tempDirPath = dirPath;
-    let allSummaries = [];
+    let fileSummariesObject = {};
 
     for (const file of processedFiles) {
       try {
         const content = await fs.readFile(file, 'utf8');
         const summary = await generateSummary(content);
-        allSummaries.push(summary);
+        fileSummariesObject[file] = summary; // Store summary associated with file name
       } catch (fileReadError) {
         console.error('Error reading file:', fileReadError.message, fileReadError.stack); // gpt_pilot_debugging_log
       }
     }
 
-    const combinedSummaries = allSummaries.join(' ');
+    const fileSummariesArray = Object.values(fileSummariesObject); // Convert summaries object to array
+    console.log('File summaries:', fileSummariesArray); // gpt_pilot_debugging_log
+    
+    const combinedSummaries = fileSummariesArray.join(' ');
     const projectSummaryResponse = await openai.chat.completions.create({
       model: 'gpt-4-turbo-preview',
       messages: [{ role: "system", content: "Summarize this project based on the individual file summaries." }, { role: "user", content: combinedSummaries }],
@@ -47,12 +51,28 @@ async function processRepoInBackground(githubUrl, email) {
     });
 
     const projectSummary = projectSummaryResponse.choices[0].message.content.trim();
-    await Repository.findOneAndUpdate({ githubUrl, email }, { summary: projectSummary, isProcessed: true }, { new: true });
+    const updatedRepository = await Repository.findOneAndUpdate(
+      { githubUrl, email }, 
+      { 
+        summary: projectSummary, 
+        isProcessed: true, 
+        fileSummaries: fileSummariesArray // Save the summaries array to the database 
+      }, 
+      { new: true }
+    );
+    console.log(`Repository has been processed and file summaries stored: ${githubUrl}`, fileSummariesArray); // gpt_pilot_debugging_log
+    console.log(`Ready to send email notification to: ${email} for repository: ${githubUrl}`); // gpt_pilot_debugging_log
+
+    try {
+      await sendEmailNotification(email, updatedRepository.uuid, githubUrl);
+      console.log(`Email notification sent to: ${email} for repository: ${githubUrl}`); // gpt_pilot_debugging_log
+    } catch (notificationError) {
+      console.error(`Error sending email notification to ${email}:`, notificationError.message, notificationError.stack); // gpt_pilot_debugging_log
+    }
 
-    console.log(`Repository has been processed: ${githubUrl}`); // gpt_pilot_debugging_log
   } catch (error) {
     console.error('Error during repository background processing:', error.message, error.stack); // gpt_pilot_debugging_log
-    throw error; // This preserves the original behavior of rethrowing the error after logging.
+    throw error;
   } finally {
     if (tempDirPath) {
       try {

public/chat.css

@@ -0,0 +1,70 @@
+.chat-container {
+  border: 1px solid #ddd;
+  padding: 10px;
+  border-radius: 4px;
+  max-height: 400px;
+  overflow-y: auto;
+}
+
+.message {
+  padding: 10px;
+  margin: 5px;
+  border-radius: 4px;
+  position: relative;
+}
+
+.user-question {
+  background-color: #007bff;
+  color: white;
+  text-align: left;
+  max-width: 80%;
+  word-wrap: break-word;
+}
+
+.ai-answer {
+  background-color: #f1f1f1;
+  border: 2px solid #007bff;
+  text-align: left;
+  max-width: 80%;
+  word-wrap: break-word;
+  margin-left: auto; /* Align to the right */
+}
+
+/* Style adjustments for a better visual distinction between user and AI messages */
+.user-question:after {
+  content: '';
+  position: absolute;
+  top: 0;
+  right: -10px;
+  width: 0;
+  height: 0;
+  border: 10px solid transparent;
+  border-left-color: #007bff;
+  border-right: 0;
+  border-bottom: 0;
+  margin-top: 5px;
+}
+
+.ai-answer:before {
+  content: '';
+  position: absolute;
+  top: 0;
+  left: -20px; /* Adjust left position to align the triangle properly */
+  width: 0;
+  height: 0;
+  border: 10px solid transparent;
+  border-right-color: #f1f1f1;
+  border-left: 0;
+  border-bottom: 0;
+  margin-top: 5px;
+}
+
+/* Additional styles to make the chat container visually appealing */
+.form-group {
+  margin-bottom: 10px;
+}
+
+/* Scroll to bottom for the latest messages */
+.chat-container {
+  scroll-behavior: smooth;
+}

repository.js

@@ -6,27 +6,31 @@ const repositorySchema = new mongoose.Schema({
   uuid: { type: String, default: function genUUID() { return uuid.v4(); }, unique: true },
   email: { type: String, required: true },
   summary: { type: String, default: '' },
+  fileSummaries: [{ type: String }], // Added field for file summaries
   isProcessed: { type: Boolean, default: false }
 }, { timestamps: true });
 
 repositorySchema.post('save', function(error, doc, next) {
   if (error.name === 'MongoError' && error.code === 11000) {
+    console.error('MongoDB duplicate key error:', error.message, error.stack); // gpt_pilot_error_log
     next(new Error('There was a duplicate key error'));
   } else {
     next(error);
   }
 });
 
-repositorySchema.post('save', function(error, doc, next) {
-  console.log('Repository saved:', doc);
+repositorySchema.post('save', function(doc, next) {
+  console.log('Repository saved:', doc.uuid); // Adjusted for clarity in log
   next();
 });
 
 repositorySchema.post('validate', function(error, doc, next) {
-  console.error('Validation error:', error);
+  if (error) {
+    console.error('Validation error:', error.message, error.stack); // gpt_pilot_error_log
+  }
   next(error);
 });
 
 const Repository = mongoose.model('Repository', repositorySchema);
 
-module.exports = Repository;
+module.exports = Repository;