Update app.py

app.py

@@ -198,6 +198,62 @@ def improve_summary_generation(text, model, tokenizer):
     
     return summary
 
+def post_process_medical_summary(summary):
+    """Special post-processing for medical/scientific summaries"""
+    if not summary:
+        return summary
+    
+    # Fix common medical text issues
+    summary = (summary
+        .replace(" p =", " p=")  # Fix p-value spacing
+        .replace(" n =", " n=")  # Fix sample size spacing
+        .replace("( ", "(")      # Fix parentheses spacing
+        .replace(" )", ")")
+        .replace("vs.", "versus")  # Expand abbreviations
+        .replace("..", ".")      # Fix double periods
+    )
+    
+    # Ensure statistical significance symbols are correct
+    summary = (summary
+        .replace("p < ", "p<")
+        .replace("p > ", "p>")
+        .replace("P < ", "p<")
+        .replace("P > ", "p>")
+    )
+    
+    # Fix number formatting
+    summary = (summary
+        .replace(" +/- ", "±")
+        .replace(" ± ", "±")
+    )
+    
+    # Split into sentences and process each
+    sentences = [s.strip() for s in summary.split('.')]
+    processed_sentences = []
+    
+    for sentence in sentences:
+        if sentence:
+            # Capitalize first letter
+            sentence = sentence[0].upper() + sentence[1:] if sentence else sentence
+            
+            # Fix common medical abbreviations spacing
+            sentence = (sentence
+                .replace(" et al ", " et al. ")
+                .replace("et al.", "et al.")  # Fix double period
+            )
+            
+            processed_sentences.append(sentence)
+    
+    # Join sentences
+    summary = '. '.join(processed_sentences)
+    
+    # Ensure proper ending
+    if summary and not summary.endswith('.'):
+        summary += '.'
+    
+    return summary
+
+
 def post_process_medical_summary(summary):
     """Special post-processing for medical/scientific summaries"""
     if not summary: