Upload 10 files

app.py

@@ -39,9 +39,6 @@ def app():
         with st.form("repo_url_form"):
             repo_form.display_form()
 
-        # Check if the API key is valid before proceeding
-        if repo_form.clone_repo_button and not repo_form.is_api_key_valid():
-            st.stop()
 
         repo_url, extensions = repo_form.get_form_data()
 

forms.py

@@ -1,7 +1,7 @@
 import logging
 import os
 
-import openai
+from openai import OpenAI
 import repo
 import streamlit as st
 from streamlit_tree_select import tree_select
@@ -25,15 +25,7 @@ class RepoForm:
         self.repo_url = st.text_input(
             "GitHub Repository URL:", self.default_repo_url
         )
-
-        env_api_key = os.getenv("OPENAI_API_KEY", "")
-        self.api_key = st.text_input(
-            "OpenAI API Key:",
-            env_api_key,
-            placeholder="Paste your API key here",
-        )
-        openai.api_key = self.api_key
-
+        
         self.extensions = st.multiselect(
             "File extensions to analyze",
             options=self.options,

query.py

@@ -1,12 +1,14 @@
 import logging
 from textwrap import dedent
 from typing import Iterable
-
-import openai
+import os
+from dotenv import load_dotenv
+from openai import OpenAI
 import streamlit as st
 import tiktoken
 
-
+load_dotenv()
+client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 def analyze_code_files(code_files: list[str]) -> Iterable[dict[str, str]]:
     """Analyze the selected code files and return recommendations."""
     return (analyze_code_file(code_file) for code_file in code_files)
@@ -134,7 +136,7 @@ Your review:"""
 
     logging.info("Sending request to OpenAI API for code analysis")
     logging.info("Max response tokens: %d", tokens_for_response)
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-3.5-turbo",
         messages=messages,
         max_tokens=tokens_for_response,
@@ -144,6 +146,6 @@ Your review:"""
     logging.info("Received response from OpenAI API")
 
     # Get the assistant's response from the API response
-    assistant_response = response.choices[0].message["content"]
+    assistant_response = response.choices[0].message.content
 
     return assistant_response.strip()