Some refinement
Browse files- handler.py +8 -3
handler.py
CHANGED
|
@@ -3,12 +3,17 @@ from prompts import *
|
|
| 3 |
|
| 4 |
|
| 5 |
def code_review(code,c_prompt=None):
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
if c_prompt is not None and len(c_prompt) > 30:
|
| 8 |
-
prompt=custom_prompt(code,c_prompt)
|
| 9 |
else:
|
| 10 |
-
prompt=default_gemini_prompt(code)
|
|
|
|
|
|
|
| 11 |
model=GeminiModel()
|
| 12 |
res= model.execute(prompt)
|
| 13 |
-
|
| 14 |
return res.text
|
|
|
|
|
|
| 3 |
|
| 4 |
|
| 5 |
def code_review(code,c_prompt=None):
|
| 6 |
+
|
| 7 |
+
if code is None or len(code)< 5 or code.isspace():
|
| 8 |
+
return "No Code Provided"
|
| 9 |
|
| 10 |
if c_prompt is not None and len(c_prompt) > 30:
|
| 11 |
+
prompt=custom_prompt(code.strip(),c_prompt.strip())
|
| 12 |
else:
|
| 13 |
+
prompt=default_gemini_prompt(code.strip())
|
| 14 |
+
|
| 15 |
+
|
| 16 |
model=GeminiModel()
|
| 17 |
res= model.execute(prompt)
|
|
|
|
| 18 |
return res.text
|
| 19 |
+
|