Update summarizer/summarizer.py

summarizer/summarizer.py

@@ -38,30 +38,36 @@ class Summarizer:
             return f"Translation failed: {str(e)}"
     def summarize_text(self, text: str, target_lang: str = 'en') -> str:
     """Detect language, translate if needed, summarize, format, and retranslate if needed."""
-    try:
-        cleaned_text = self.clean_text(text)
+        try:
+            cleaned_text = self.clean_text(text)
         # Translate if not in English
-        if target_lang != 'en':
-            cleaned_text = self.translate_text(cleaned_text, "en")
-        # Summarize using Hugging Face API (simplify the payload)
+            if target_lang != 'en':
+                cleaned_text = self.translate_text(cleaned_text, "en")
+        # Summarize using Hugging Face API
         payload = {
-            "inputs": cleaned_text  # Send just the cleaned text as input
+            "inputs": f"summarize the clinical case with diagnosis, comorbidities, and treatment plan: {cleaned_text}"
         }
         response = requests.post(self.summarizer_api_url, headers=self.headers, json=payload, timeout=30)
         response.raise_for_status()
+
         response_data = response.json()
+
         if isinstance(response_data, list) and "generated_text" in response_data[0]:
             summary = response_data[0]["generated_text"]
         else:
             return "Unexpected response format from Hugging Face API."
+
         formatted_summary = self.format_summary(summary)
+
         # Translate back to target language if needed
         if target_lang != 'en':
             formatted_summary = self.translate_text(formatted_summary, target_lang)
+
         return formatted_summary
+
     except requests.exceptions.Timeout:
         return "Summarization request timed out."
     except requests.exceptions.RequestException as e:
         return f"Summarization request failed: {str(e)}"
     except Exception as e:
-        return f"An error occurred during summarization: {str(e)}"
+        return f"An error occurred during summarization: {str(e)}"