Update summarizer/summarizer.py

summarizer/summarizer.py

@@ -6,36 +6,27 @@ class Summarizer:
     def __init__(self):
         self.api_url = "https://api-inference.huggingface.co/models/Aadityaramrame/carecompanion-summarizer"
         self.api_token = os.getenv("API_KEY")
-
         if not self.api_token:
             raise ValueError("Hugging Face API Key not found in environment variables.")
-
         self.headers = {
             "Authorization": f"Bearer {self.api_token}",
             "Content-Type": "application/json"
         }
         self.translator = TextTranslator()
-
     def clean_text(self, text: str) -> str:
         """Remove unnecessary spaces and line breaks."""
         return ' '.join(text.replace('\n', ' ').split())
-
     def format_summary(self, summary: str) -> str:
         """Apply simple formatting rules to the model output."""
         summary = summary.strip()
-
         if summary and not summary[0].isupper():
             summary = summary[0].upper() + summary[1:]
-
         if "expected to recover within" in summary and not summary.endswith("days."):
             summary = summary.rstrip('. ')
             summary += " 7–10 days."
-
         if "antibiotic" in summary.lower() and "supportive" in summary.lower() and "treatment" not in summary.lower():
             summary += " Treatment includes antibiotics and supportive care."
-
         return summary
-
     def summarize_text(self, text: str, target_lang: str = 'en') -> str:
         """Detect language, translate if needed, summarize, format, and retranslate if needed."""
         try:
@@ -43,35 +34,26 @@ class Summarizer:
             detected_lang = self.translator.detect_language(text)
             if detected_lang != 'en':
                 text = self.translator.translate_to_english(text)
-
             cleaned_text = self.clean_text(text)
-
             payload = {
                 "inputs": f"summarize the clinical case with diagnosis, comorbidities, and treatment plan: {cleaned_text}"
             }
-
             response = requests.post(self.api_url, headers=self.headers, json=payload, timeout=30)
             response.raise_for_status()
-
             response_data = response.json()
-
             # Hugging Face returns list of dicts: [{'generated_text': 'summary'}]
             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.translator.translate_from_english(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)}"