Adding model selection

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import subprocess
 import shutil
+import openai
 
 css_style = """
 .gradio-container {
@@ -9,7 +10,7 @@ css_style = """
 }
 """
 
-def process_repository(url):
+def process_repository(url, model):
 
     
     # Split the URL to get the repo name
@@ -25,7 +26,7 @@ def process_repository(url):
         os.chdir(repo_name)
 
         # Run your package command
-        subprocess.run(['gpt4readability', '.', '--function', 'readme'])
+        subprocess.run(['gpt4readability', '.', '--function', 'readme', '--model', model])
 
         # Open the README.md file and return its contents
         with open('README.md', 'r') as readme_file:
@@ -40,10 +41,20 @@ def process_repository(url):
         if os.path.exists(repo_name):
             shutil.rmtree(repo_name)
 
-def generate_repo(url, api_key):
+def generate_repo(url, api_key, model):
     if api_key:
         os.environ['OPENAI_API_KEY'] = api_key.strip()
-        return process_repository(url)
+        if model == 'gpt-4':
+            try:
+                response = openai.Completion.create(
+                    model="gpt-4",  # or whatever the exact model ID is
+                    prompt="test",
+                    max_tokens=5
+                )
+                print("Access to GPT-4 confirmed!")
+            except:
+                return "The API key either does not have access to GPT-4 or is not valid."
+        return process_repository(url, model)
     else:
         return "Please add a valid OpenAI API Key (you can get them [here](https://platform.openai.com/account/api-keys))"
 
@@ -58,15 +69,17 @@ with gr.Blocks(css=css_style) as demo:
         * [langchain](https://github.com/hwchase17/langchain) is the main library this tool utilizes.
         1. Enter API Key ([What is that?](https://platform.openai.com/account/api-keys))
         2. Paste in a GitHub Repository URL
-        3. Generate a README or suggestions markdown file
+        3. Choose a model (The gpt-4 API as of July 2023 is not available to everyone)
+        4. Generate a README or suggestions markdown file
         """)
 
     openai_api_key = gr.Textbox(
             label="OpenAI API Key", placeholder="sk-...", type="password")
     url = gr.Textbox(label="GitHub Repository URL")
+    model = gr.Dropdown(["gpt-3.5-turbo", "gpt-4"], type="value", label='Model Type')
     output = gr.Markdown(label="README.md")
     btn = gr.Button("Generate README.md")
-    btn.click(fn=generate_repo, inputs=[url,openai_api_key], outputs=[output], api_name="Generate README.md")
+    btn.click(fn=generate_repo, inputs=[url, openai_api_key, model], outputs=[output], api_name="Generate README.md")
     
 
 demo.queue(concurrency_count=20)