Upload 3 files

README.md

@@ -1,12 +1 @@
----
-title: Code Review
-emoji: 🦀
-colorFrom: purple
-colorTo: green
-sdk: gradio
-sdk_version: 5.20.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# code_review_tool

app.py

@@ -0,0 +1,346 @@
+import os
+import gradio as gr
+import psycopg2
+from psycopg2.extras import RealDictCursor
+from dotenv import load_dotenv
+from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+import openai
+from contextlib import contextmanager
+
+load_dotenv()  # Load environment variables
+
+GPT_4O_MINI = "gpt-4o-mini"
+GPT_4 = "chatgpt-4o-latest" 
+O1_MINI = "o1-mini"
+
+# Global cache for prompts
+cached_prompts = {}
+
+@contextmanager
+def openai_session():
+    """Context manager to properly handle OpenAI API sessions"""
+    try:
+        # Initialize client
+        client = openai.OpenAI()
+        yield client
+    finally:
+        # Clean up client resources
+        if hasattr(client, 'close'):
+            client.close()
+
+@with_langtrace_root_span()
+def call_model(prompt, model):
+    print(f"calling {model} with prompt: {prompt[:100]}")
+    with openai_session() as client:
+        try:
+            # Call API
+            response = client.chat.completions.create(
+                model=model,
+                messages=[{"role": "user", "content": prompt}]
+            )
+            # Extract response text
+            result = response.choices[0].message.content
+            return result
+        
+        except Exception as e:
+            return f"Error generating output: {str(e)}"
+
+def get_db_connection():
+    """Establishes and returns a new database connection."""
+    db_params = {
+        'dbname': os.getenv('DB_NAME'),
+        'user': os.getenv('DB_USER'),
+        'password': os.getenv('DB_PASSWORD'),
+        'host': os.getenv('DB_HOST'),
+        'port': os.getenv('DB_PORT')
+    }
+    conn = psycopg2.connect(**db_params)
+    return conn
+
+def load_prompts():
+    """Fetches prompts from the DB and caches them as a dict mapping prompt name to a dict with body and id."""
+    global cached_prompts
+    conn = get_db_connection()
+    cursor = conn.cursor(cursor_factory=RealDictCursor)
+    cursor.execute("""
+        SELECT code_review_prompt_name, code_review_prompt_body, code_review_prompt_id 
+        FROM code_review_prompt
+        ORDER BY code_review_prompt_name;
+    """)
+    rows = cursor.fetchall()
+    # Cache format: { prompt_name: {"body": prompt_body, "id": prompt_id} }
+    cached_prompts = { row["code_review_prompt_name"]: {"body": row["code_review_prompt_body"], "id": row["code_review_prompt_id"]} for row in rows }
+    cursor.close()
+    conn.close()
+    return cached_prompts
+
+# Initially load prompts
+load_prompts()
+
+def update_prompt_text(selected_prompt):
+    """Uses the global cache to return the prompt body for the selected prompt."""
+    return cached_prompts.get(selected_prompt, {}).get("body", "")
+
+def build_final_prompt(custom_prompt, input_code):
+    """
+    Builds the final prompt for code review.
+    """
+    return (
+        f"Please review the following code:\n\n"
+        f"{custom_prompt}\n\n"
+        f"<CODE>\n"
+        f"{input_code}\n"
+    )
+
+def store_submission(prompt_name, custom_prompt, input_code, review_output):
+    """
+    Stores the submission in the database.
+    """
+    prompt_data = cached_prompts.get(prompt_name)
+    prompt_id = prompt_data["id"] if prompt_data else None
+
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        INSERT INTO code_review_submission (code_review_prompt_id, custom_prompt, input_code, ai_review)
+        VALUES (%s, %s, %s, %s);
+    """, (prompt_id, custom_prompt, input_code, review_output))
+    conn.commit()
+    cursor.close()
+    conn.close()
+
+def generate_ai_review(prompt_name, custom_prompt, input_code, model):
+    """
+    Generates an AI review by calling the API, then stores the submission.
+    Uses prompt_name to look up the prompt id.
+    """
+    final_prompt = build_final_prompt(custom_prompt, input_code)
+    review_output = call_model(final_prompt, model)
+    store_submission(prompt_name, custom_prompt, input_code, review_output)
+    return review_output
+
+def call_four(prompt_name, custom_prompt, input_code):
+    return generate_ai_review(prompt_name, custom_prompt, input_code, GPT_4)
+
+def call_o1(prompt_name, custom_prompt, input_code):
+    return generate_ai_review(prompt_name, custom_prompt, input_code, O1_MINI)
+
+def call_four_mini(prompt_name, custom_prompt, input_code):
+    return generate_ai_review(prompt_name, custom_prompt, input_code, GPT_4O_MINI)
+
+def add_new_prompt(prompt_name, prompt_body):
+    """Inserts a new prompt into the database, reloads the cache, and returns updated dropdown choices."""
+    if not prompt_name.strip() or not prompt_body.strip():
+        return "Error: Prompt Name and Body cannot be empty!", None, None
+
+    conn = None
+    try:
+        conn = get_db_connection()
+        cursor = conn.cursor(cursor_factory=RealDictCursor)
+        cursor.execute("""
+            INSERT INTO code_review_prompt (code_review_prompt_name, code_review_prompt_body)
+            VALUES (%s, %s)
+            ON CONFLICT (code_review_prompt_name) DO NOTHING
+            RETURNING code_review_prompt_id;
+        """, (prompt_name.strip(), prompt_body.strip()))
+        inserted = cursor.fetchone()
+        if inserted:
+            conn.commit()
+            message = f"Prompt '{prompt_name}' added successfully!"
+        else:
+            message = f"Prompt '{prompt_name}' already exists!"
+        cursor.close()
+        conn.close()
+        # Reload the prompt cache after insertion
+        load_prompts()
+        new_choices = list(cached_prompts.keys())
+        new_dropdown = gr.Dropdown(choices=new_choices, label="Select Review Type", value=new_choices[0] if new_choices else None)
+        return message, new_dropdown, ""
+    except psycopg2.IntegrityError:
+        conn.rollback()
+        return "Prompt already exists!", None, ""
+    except Exception as e:
+        if conn:
+            conn.rollback()
+        return f"Error: {str(e)}", None, ""
+    finally:
+        if conn:
+            conn.close()
+
+def reload_prompts():
+    """Reloads the prompt cache and returns a new Dropdown with updated choices."""
+    load_prompts()
+    new_choices = list(cached_prompts.keys())
+    new_dropdown = gr.Dropdown(choices=new_choices, label="Select Review Type", value=new_choices[0] if new_choices else None)
+    return new_dropdown
+
+# ---------------------- New Code for Loading Previous Submissions ---------------------- #
+def load_submissions_from_db():
+    """Fetch all submissions ordered by their id in ascending order."""
+    conn = get_db_connection()
+    cursor = conn.cursor(cursor_factory=RealDictCursor)
+    cursor.execute("""
+        SELECT code_review_submission_id, code_review_prompt_id, custom_prompt, input_code, ai_review
+        FROM code_review_submission
+        ORDER BY created_at DESC; """)
+    submissions = cursor.fetchall()
+    cursor.close()
+    conn.close()
+    return submissions
+
+def show_submission(index, submissions):
+    """Returns formatted outputs for the submission at the given index:
+    - Submission details (Markdown)
+    - Input code (for gr.Code)
+    - AI review output (Markdown)
+    """
+    if not submissions:
+        return "No submissions available", "", "No submissions available", index
+    # Clamp index within bounds
+    index = max(0, min(index, len(submissions) - 1))
+    submission = submissions[index]
+    # Left panel Markdown for submission details
+    submission_details = (
+        f"**Submission ID:** {submission['code_review_submission_id']}\n\n"
+        f"**Prompt ID:** {submission['code_review_prompt_id']}\n\n"
+        f"**Custom Prompt:**\n{submission['custom_prompt']}\n\n"
+    )
+    # Input code for gr.Code
+    input_code = submission['input_code']
+    # Right panel Markdown for AI review output
+    ai_review_output = f"**AI Review Output:**\n{submission['ai_review']}\n"
+    return submission_details, input_code, ai_review_output, index
+
+
+def load_submissions():
+    """Loads submissions from the database and resets the index."""
+    submissions = load_submissions_from_db()
+    index = 0
+    details, code, review, index = show_submission(index, submissions)
+    return submissions, index, details, code, review
+
+def next_submission(current_index, submissions):
+    """Increment the index and display the next submission."""
+    if submissions and current_index < len(submissions) - 1:
+        current_index += 1
+    details, code, review, current_index = show_submission(current_index, submissions)
+    return current_index, details, code, review
+
+def previous_submission(current_index, submissions):
+    """Decrement the index and display the previous submission."""
+    if submissions and current_index > 0:
+        current_index -= 1
+    details, code, review, current_index = show_submission(current_index, submissions)
+    return current_index, details, code, review
+
+# ----------------------------------------------------------------------------------------- #
+
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("# Code Review Assistant")
+
+    with gr.Tab("Code Review"):
+        with gr.Row():
+            with gr.Column(scale=1):
+                prompt_dropdown = gr.Dropdown(
+                    choices=list(cached_prompts.keys()),
+                    label="Select Review Type",
+                    value=list(cached_prompts.keys())[0] if cached_prompts else None
+                )
+                with gr.Accordion("Selected Prompt", open=False):
+                    prompt_textbox = gr.Textbox(label="", lines=15, interactive=True)
+        with gr.Row():
+            generate_4o_btn = gr.Button("Generate Code Review 4o")
+            generate_4omini_btn = gr.Button("Generate Code Review 4o-mini")
+            generate_o1_btn = gr.Button("Generate Code Review o1-mini")
+        with gr.Row():
+            input_code = gr.Code(language="python", label="Input Code", lines=15, interactive=True)
+            output_review = gr.Markdown(label="AI Review Output", container=True, show_copy_button=True)
+
+    with gr.Tab("Manage Prompts"):
+        with gr.Row():
+            new_prompt_name = gr.Textbox(label="New Prompt Name", placeholder="Enter a prompt name")
+            new_prompt_body = gr.Textbox(label="New Prompt Body", lines=3, placeholder="Enter the prompt details")
+        add_prompt_btn = gr.Button("Add Prompt")
+        reload_prompt_btn = gr.Button("Reload Prompts")
+        prompt_status = gr.Textbox(label="Status", interactive=False)
+
+    with gr.Tab("Submissions"):
+        with gr.Row():
+            load_submissions_btn = gr.Button("Load Submissions")
+        # Hidden states to store submissions and the current index
+        submissions_state = gr.State([])
+        submission_index_state = gr.State(0)
+        # Create two columns: left column for details and code, right column for AI review output
+        with gr.Row():
+            back_btn = gr.Button("Back")
+            next_btn = gr.Button("Next")
+        with gr.Row():
+            with gr.Column():
+                with gr.Accordion("Prompt Details", open= False):
+                    submission_details_md = gr.Markdown(label="Submission Details")
+                input_code_component = gr.Code(label="Input Code", language="python")
+            with gr.Column():
+                ai_review_md = gr.Markdown(label="AI Review Output", container=True, show_copy_button=True)
+        
+        # Load submissions from DB and initialize state
+        load_submissions_btn.click(
+            fn=load_submissions,
+            inputs=[],
+            outputs=[submissions_state, submission_index_state, submission_details_md, input_code_component, ai_review_md]
+        )
+        # Go to previous submission
+        back_btn.click(
+            fn=previous_submission,
+            inputs=[submission_index_state, submissions_state],
+            outputs=[submission_index_state, submission_details_md, input_code_component, ai_review_md]
+        )
+        # Go to next submission
+        next_btn.click(
+            fn=next_submission,
+            inputs=[submission_index_state, submissions_state],
+            outputs=[submission_index_state, submission_details_md, input_code_component, ai_review_md]
+        )
+
+
+
+    # When dropdown changes, update the prompt text
+    prompt_dropdown.change(
+        fn=update_prompt_text,
+        inputs=[prompt_dropdown],
+        outputs=[prompt_textbox]
+    )
+
+    # On code review generation, pass both the dropdown (prompt name) and the editable prompt text
+    generate_4o_btn.click(
+        fn=call_four,
+        inputs=[prompt_dropdown, prompt_textbox, input_code],
+        outputs=[output_review]
+    )
+    generate_4omini_btn.click(
+        fn=call_four_mini,
+        inputs=[prompt_dropdown, prompt_textbox, input_code],
+        outputs=[output_review]
+    )
+    generate_o1_btn.click(
+        fn=call_o1,
+        inputs=[prompt_dropdown, prompt_textbox, input_code],
+        outputs=[output_review]
+    )
+
+    # On adding a new prompt, update status, replace dropdown with a new one, and clear new prompt body textbox.
+    add_prompt_btn.click(
+        fn=add_new_prompt,
+        inputs=[new_prompt_name, new_prompt_body],
+        outputs=[prompt_status, prompt_dropdown, new_prompt_body]
+    )
+
+    # Reload prompts when the reload button is clicked.
+    reload_prompt_btn.click(
+        fn=reload_prompts,
+        inputs=[],
+        outputs=[prompt_dropdown]
+    )
+
+
+demo.launch()

requirements.txt

@@ -0,0 +1,136 @@
+gradio==5.12.0
+gradio_client==1.5.4
+huggingface-hub==0.27.1
+hyperframe==6.0.1
+openai==1.59.6
+psycopg2-binary==2.9.10
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+pydantic==2.10.5
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.19.1
+pypandoc==1.15
+pyparsing==3.2.1
+python-dateutil==2.9.0.post0
+python-docx==1.1.2
+python-dotenv==1.0.1
+python-multipart==0.0.20
+pytz==2024.2
+PyYAML==6.0.2
+supabase==2.11.0
+supafunc==0.9.0
+aiofiles==23.2.1
+aiohappyeyeballs==2.5.0
+aiohttp==3.11.13
+aiosignal==1.3.2
+annotated-types==0.7.0
+anyio==4.8.0
+attrs==25.1.0
+boto3==1.37.7
+botocore==1.37.7
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+Deprecated==1.2.18
+deprecation==2.1.0
+distro==1.9.0
+fastapi==0.115.11
+ffmpy==0.5.0
+filelock==3.17.0
+frozenlist==1.5.0
+fsspec==2025.2.0
+googleapis-common-protos==1.69.0
+gotrue==2.11.4
+gradio==5.12.0
+gradio_client==1.5.4
+groovy==0.1.2
+grpcio==1.70.0
+h11==0.14.0
+h2==4.1.0
+hpack==4.1.0
+httpcore==1.0.7
+httpx==0.27.2
+huggingface-hub==0.27.1
+hyperframe==6.0.1
+idna==3.10
+importlib_metadata==8.5.0
+Jinja2==3.1.6
+jiter==0.8.2
+jmespath==1.0.1
+langtrace-python-sdk==3.8.1
+lxml==5.3.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+multidict==6.1.0
+numpy==2.2.3
+openai==1.59.6
+opentelemetry-api==1.30.0
+opentelemetry-exporter-otlp-proto-common==1.30.0
+opentelemetry-exporter-otlp-proto-grpc==1.30.0
+opentelemetry-exporter-otlp-proto-http==1.30.0
+opentelemetry-instrumentation==0.51b0
+opentelemetry-instrumentation-sqlalchemy==0.51b0
+opentelemetry-proto==1.30.0
+opentelemetry-sdk==1.30.0
+opentelemetry-semantic-conventions==0.51b0
+orjson==3.10.15
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+postgrest==0.19.3
+propcache==0.3.0
+protobuf==5.29.3
+psycopg2==2.9.10
+psycopg2-binary==2.9.10
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+pydantic==2.10.5
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.19.1
+pypandoc==1.15
+pyparsing==3.2.1
+python-dateutil==2.9.0.post0
+python-docx==1.1.2
+python-dotenv==1.0.1
+python-multipart==0.0.20
+pytz==2024.2
+PyYAML==6.0.2
+realtime==2.4.1
+regex==2024.11.6
+requests==2.32.3
+rich==13.9.4
+ruff==0.9.9
+s3transfer==0.11.4
+safehttpx==0.1.6
+safetensors==0.5.3
+semantic-version==2.10.0
+sentry-sdk==2.22.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+SQLAlchemy==2.0.38
+starlette==0.46.0
+storage3==0.11.3
+StrEnum==0.4.15
+supabase==2.11.0
+supafunc==0.9.0
+tiktoken==0.9.0
+tokenizers==0.21.0
+tomlkit==0.13.2
+tqdm==4.67.1
+trace-attributes==7.2.0
+transformers==4.49.0
+typer==0.15.2
+typing_extensions==4.12.2
+tzdata==2025.1
+ujson==5.10.0
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==14.2
+wrapt==1.17.2
+yarl==1.18.3
+zipp==3.21.0