Update app.py

app.py

@@ -116,12 +116,11 @@ Suggestions:
 
 Now, review the following code using this approach:
 
-{code_to_analyze}
-
-Provide a detailed review following the structure above, including understanding the code, identifying potential security issues, identifying potential logic vulnerabilities, and offering specific suggestions for improvement. Start your response with 'Code Review:' and end it with 'End of Review.'"
 """
 
-    query = few_shot_prompt.format(code_to_analyze=code_to_analyze)
+    # Concatenate the prompt with the code to analyze
+    query = few_shot_prompt + "\n\n" + code_to_analyze + "\n\nProvide a detailed review following the structure above, including understanding the code, identifying potential security issues, identifying potential logic vulnerabilities, and offering specific suggestions for improvement. Start your response with 'Code Review:' and end it with 'End of Review.'"
+
     full_result = get_completion(query, model, tokenizer)
     
     # Extract only the AI's answer