imkhan107 commited on
Commit
dcb6098
·
1 Parent(s): c21f3cc

Some refinement

Browse files
Files changed (1) hide show
  1. 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
+