Update app.py

app.py

@@ -106,6 +106,17 @@ class GenerateResponse(BaseModel):
     reasoning_content: str
     generated_text: str
 
+# New model for summarization request
+class SummarizeRequest(BaseModel):
+    replies: List[str]
+    task: str  # expecting "summarisation"
+
+# New model for summarization response
+class SummarizeResponse(BaseModel):
+    individual_summaries: Dict[int, Dict[str, str]]  # {index: {"reasoning": str, "summary": str}}
+    combined_reasoning: str
+    combined_summary: str
+
 # --- Utility Functions ---
 
 def clean_text(text: str) -> str: