Add sandwich defense for hard challenge

app.py

@@ -1,3 +1,4 @@
+from typing import Literal, Optional
 import huggingface_hub
 import gradio as gr
 import pandas as pd
@@ -179,16 +180,19 @@ def rows_to_text(rows: list[list[str]]) -> str:
     return text
 
 
-def get_summary(rows: list[list[str]]) -> str:
+def get_summary(rows: list[list[str]], difficulty: Optional[Literal["playground", "easy", "hard"]] = None) -> str:
     text = rows_to_text(rows)
     prompt = text
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": prompt},
+    ]
+    if difficulty == "hard":
+        messages.append({"role": "user", "content":"Summarize the feedback"})
     client = openai.Client(api_key=openai_api_key)
     completion = client.chat.completions.create(
         model="gpt-4o-mini",
-        messages=[
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": prompt},
-        ],
+        messages=messages,
     )
     summary = completion.choices[0].message.content
     return summary
@@ -207,7 +211,7 @@ def summarize_feedback(rows: list[list[str]], env: Env, username: str) -> str:
     if env == Env.CHALLENGE_HARD:
         summary = "-"
     else:
-        summary = get_summary(rows)
+        summary = get_summary(rows, "hard")
 
     add_review_db(env.value, user, feedback, summary)