chore: minimal parser

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-
+from parser import MinimalPromptParser
 
 def get_language_from_extension(extension):
     ext_map = {
@@ -23,16 +23,15 @@ def get_language_from_extension(extension):
 
 api_key = st.text_input("Enter your OpenAI API Key:", type="password")
 
-simple_review = 'You are an harsh code reviewer that enforces diligent and good code practices. You are tasked with reviewing the user code in a funny, spitefull and relevant way. Use markdown.';
+
 def llmcall(input):
+    simple_review = MinimalPromptParser().parse_file('simplereview.md', {'code':input})
+
     from openai import OpenAI
     client = OpenAI(api_key = api_key)
     completion = client.chat.completions.create(
     model="gpt-3.5-turbo",
-    messages=[
-        {"role": "system", "content": simple_review},
-        {"role": "user", "content": input}
-    ]
+    messages=simple_review
     )
 
     return completion.choices[0].message.content

parser.py

@@ -0,0 +1,30 @@
+class MinimalPromptParser:
+    def __init__(self):
+        self.parsed_data = []
+
+    def parse_file(self, filename, user_input):
+        with open(filename, 'r') as file:
+            role = None
+            content = ""
+            for line in file:
+                line = line.strip()
+                if line.startswith("system:") or line.startswith("user:"):
+                    if role is not None:
+                        self.parsed_data.append({"role": role, "content": content})
+                    role = line.split(":")[0]
+                    content = ""
+                else:
+                    content += line + "\n"
+            if role is not None:
+                self.parsed_data.append({"role": role, "content": content})
+        
+        self.replace_placeholders(user_input)
+
+        return self.parsed_data
+
+    def replace_placeholders(self, user_input):
+        for item in self.parsed_data:
+            content = item["content"]
+            for key, value in user_input.items():
+                content = content.replace("{{" + key + "}}", value)
+            item["content"] = content

simplereview.md

@@ -0,0 +1,4 @@
+system:
+You are an harsh code reviewer that enforces diligent and good code practices. You are tasked with reviewing the user code in a funny, spitefull and relevant way. Use markdown.
+user:
+{{code}}