initial commit

app.py

@@ -0,0 +1,34 @@
+import streamlit as st
+from handler import *
+
+def main():
+  """ Creates a Streamlit page for code analysis """
+  prompt=None
+  code=None
+  # Title and description
+  st.set_page_config(page_title="CodeBatuk: Code Review Tool", page_icon="")
+  st.title("Code Batuk")
+  st.subheader("A code Analysis Tool")
+  
+  col1, col2 = st.columns([3,1])
+  with col1:
+    # Text input fields
+    code = st.text_area(label="Enter your code here:", height=200)
+  #prompt = st.text_area(label="Custom prompt (optional):")
+  with col2:
+    # Radio buttons for analysis type
+    analysis_type = st.radio("Choose analysis type:", 
+                                ("Code review", "Code refinement", "Unit tests"))
+
+  # Submit button
+  if st.button("Submit"):
+    # Process the code and prompt based on analysis_type (implementation omitted for brevity)
+    if analysis_type == "Code review":
+      res=code_review(code, prompt)
+      #st.text_area(label="Result",value=res, height=300)
+      st.markdown(res)
+    st.success(f"Code analysis for {analysis_type} submitted successfully!")
+
+if __name__ == "__main__":
+  main()
+#git clone https://huggingface.co/spaces/imkhan107/codebatuk

gemini.py

@@ -0,0 +1,33 @@
+import google.generativeai as genai
+import os
+from dotenv import load_dotenv
+from typing import Optional
+
+class GeminiModel:
+    """
+    This class is used to interact with the google llm models for text generation.
+
+    Args:
+        model: The name of the model to be used. Defaults to 'text-bison@002'.
+        max_output_tokens: The maximum number of tokens to generate. Defaults to 1024.
+        top_P: The probability of generating the next token. Defaults to 1.0.
+        temperature: The temperature of the model. Defaults to 0.0.
+        top_k: The number of top tokens to consider. Defaults to 5.
+    """
+
+    model=None
+    def __init__(self,
+                 model: Optional[str] = 'gemini-pro',
+                 max_output_tokens: Optional[int] = 1024,
+                 top_p: Optional[float] = 1.0,
+                 temperature: Optional[float] = 0.0,
+                 top_k: Optional[int] = 5
+                 ):
+        load_dotenv() 
+        genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+        self.model=genai.GenerativeModel(model)
+        
+    def execute(self,prompt):
+        print(prompt)
+        response=self.model.generate_content(prompt)
+        return response

handler.py

@@ -0,0 +1,14 @@
+from gemini import GeminiModel
+from prompts import *
+
+
+def code_review(code,c_prompt=None):
+    
+    if c_prompt is not None and len(c_prompt) > 30:
+        prompt=custom_prompt(code,c_prompt)
+    else:
+        prompt=default_gemini_prompt(code)
+    model=GeminiModel()    
+    res= model.execute(prompt)
+
+    return res.text

prompts.py

@@ -0,0 +1,90 @@
+TASK="As an experienced software developer responsible for code review and feedback, review the following python code by considering the best practices given in the guidelines below and provide your Observations and Suggestions."
+INSTRUCTIONS="""Take a deep breath before providing the feedback and thoroughly understand the code logic.
+Avoid verbosity and repetition in your feedback, focussing on essential points and offering valuable suggestions.
+Add your own observations and suggestions as needed, and only include comments that are neccessary for clarity and improvement.
+If you have specific code suggestions, ensure they are correct and aligned with the best practices.  
+"""
+GUIDELINES = f""" Use Intention-Revealing Names.
+Pick one word per concept
+Use Solution/Problem Domain Names
+Classes should be small
+Functions should be small
+Do one Thing
+Declare constants as applicable
+Don't Repeat Yourself (Avoid Duplication)
+Explain yourself in code
+Make sure the code formatting is applied
+Use Exceptions rather than Return codes
+Don't return Null
+Favor the use of standard exceptions
+Don't ignore exceptions
+Check parameters for validity
+Return empty arrays or collections, not nulls
+Minimize the accessibility of classes and members
+In public classes, use accessor methods, not public fields
+Minimize the scope of local variables
+Refer to objects by their interfaces
+Adhere to generally accepted naming conventions
+Avoid finalizers
+Synchronize access to shared mutable data
+Valid unit test cases exist
+Objects accessed by multiple threads are accessed only through a lock, or synchronized methods.
+Race conditions have been handled
+Locks are acquired and released in the right order to prevent deadlocks, even in error handling code.
+All methods are commented in clear language
+"""
+
+STEPS="""Follow the steps in the same order:
+1. Understand the code logic and structure.
+2. Notedown the observations based on the guodelines given above.
+3. Give suggestions for improving the code.
+4. Give an example of Final improved code. 
+"""
+
+def default_gemini_prompt(code, guidelines=None):
+    if guidelines is None:
+        guidelines=GUIDELINES
+    prompt=f"""<Task>: {TASK}
+    <Instructions>: {INSTRUCTIONS}
+
+    <Guidelines>: {guidelines}
+
+    <Steps to follow>: {STEPS}
+
+    <code>: {code}
+    """
+
+    return prompt
+
+
+def prompt2(code):
+    prompt=f"""
+<task>: As an experienced software developer responsible for code review and feedback, review the following code by considering the best practices given in the guidelines below and provide your Observations and Suggestions
+Code: {code}
+
+The following custom guidelines should be considered during the review:
+
+{GUIDELINES}
+
+Starting from the beginning of the code, analyze each line following a chain of thoughts. Consider the purpose of the code, its readability, efficiency, and adherence to the listed custom guidelines. 
+
+For each line, provide feedback that includes:
+
+* Whether the code achieves its intended purpose.
+* If there are any readability issues or areas for improvement.
+* Suggestions for optimization or alternative approaches (if applicable).
+* If there are any violations of the custom guidelines.
+
+Break down your review into a step-by-step process, explaining your reasoning at each point. Maintain a clear and concise tone throughout the review.
+Add your own observations and suggestions as needed, and only include comments that are neccessary for clarity and improvement.
+If you have specific code suggestions, ensure they are correct and aligned with the best practices
+
+**Overall Recommendation:**
+
+Based on your analysis, provide an overall recommendation for the code. This could be approval with minor suggestions, approval with moderate changes, or rejection with a clear explanation for required improvements.
+"""
+    
+    return prompt
+
+def custom_prompt(c_prompt,code):
+    return f'{c_prompt}\n<Code>: {code}'

requirements.txt

@@ -0,0 +1,4 @@
+streamlit
+google-generativeai
+python-dotenv
+langchain