app.py

import gradio as gr
import os
import subprocess
import shutil
import openai

css_style = """
.gradio-container {
    font-family: "IBM Plex Mono";
}
"""

def process_repository(url, model):

    # Split the URL to get the repo name
    repo_name = url.split('/')[-1]
    if repo_name.endswith('.git'):
        repo_name = repo_name[:-4]

    # Change permissions
    subprocess.run(['chmod', 'u+w', '.'])
    
    # Clone the repo
    subprocess.run(['git', 'clone', url], check=True)

    try:
        # Change directory to the cloned repo
        os.chdir(repo_name)

        # Run your package command
        subprocess.run(['gpt4readability', '.', '--function', 'readme','--include-md', 'false', '--model', model])

        # Open the README.md file and return its contents
        with open('README.md', 'r') as readme_file:
            readme_contents = readme_file.read()

        return readme_contents
    finally:
        # Change back to the original directory
        os.chdir('..')

        # Delete the repo directory
        if os.path.exists(repo_name):
            shutil.rmtree(repo_name)

def generate_repo(url, api_key, model):
    if api_key:
        os.environ['OPENAI_API_KEY'] = api_key.strip()
        # 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 update_md(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))","Please add a valid OpenAI API Key (you can get them [here](https://platform.openai.com/account/api-keys))"

def update_md(output):
    return f"GPT4Readability Generated README.md File:\n\n{output}", f"{output}"

with gr.Blocks(css=css_style) as demo:
    gr.Markdown(f"""
        # Hello from GPT4Readability (v0.1.3)
        *Project by Dennis Loevlie ([@DennisLoevlie](https://twitter.com/DennisLoevlie))*  
        [![License Badge](https://img.shields.io/github/license/loevlie/GPT4Readability)](https://github.com/loevlie/GPT4Readability/blob/main/LICENSE)
        Welcome to GPT4Readability, a tool designed to help you generate README.md files and suggest improvements for your code repositories.
        - You can find the source code at [GPT4Readability](https://github.com/loevlie/GPT4Readability).
        - It's making use of the [langchain](https://github.com/hwchase17/langchain) library.
        ## Here's how to get started:
        1. Please enter your API Key ([Need more information?](https://platform.openai.com/account/api-keys))
        2. Provide the GitHub Repository URL that you'd like to analyze
        3. Select a model (Please note, the gpt-4 API isn't available to all as of July 2023)
        4. Click to 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')
    
    btn = gr.Button("Generate README.md")

    with gr.Tab("Rendered README.md"):
        output_rendered = gr.Markdown()

    with gr.Tab("Plain Text for Copying README.md!"):
        plain_output = gr.Textbox(show_copy_button=True, lines=100, max_lines=1000)

    # Link button click event to outputs
    btn.click(fn=generate_repo, inputs=[url, openai_api_key, model], 
              outputs=[output_rendered, plain_output])
    

demo.launch(share=False)