Fix summary truncation: Add intelligent summary generation for impact analysis and policy queries - no more cut-off summaries

groq_websocket_handler.py

@@ -365,11 +365,14 @@ class GroqWebSocketHandler:
                 # Generate charts for impact analysis queries
                 charts = await self._generate_charts_if_needed(query, response_text)
                 
+                # Create intelligent summary based on query type
+                summary = self._create_intelligent_summary(query, response_text)
+                
                 # For text clients, send structured response
                 await self.send_message(session_id, {
                     "type": "streaming_response",
                     "clause_text": response_text,
-                    "summary": response_text[:200] + "..." if len(response_text) > 200 else response_text,
+                    "summary": summary,
                     "role_checklist": [],
                     "source_title": "Government Document Assistant",
                     "clause_id": f"response_{int(time.time())}",
@@ -653,5 +656,65 @@ class GroqWebSocketHandler:
             logger.error(f"❌ Chart generation error: {e}")
             return []
 
+    def _create_intelligent_summary(self, query: str, response_text: str) -> str:
+        """Create intelligent summary based on query type and content"""
+        try:
+            query_lower = query.lower()
+            
+            # For impact analysis queries, create detailed summary
+            if any(keyword in query_lower for keyword in ['impact', 'effect', 'analyze', 'comparison', 'scenario']):
+                # Extract key points from response for impact analysis
+                lines = response_text.split('\n')
+                key_points = []
+                
+                for line in lines:
+                    line = line.strip()
+                    if any(indicator in line.lower() for indicator in ['conclusion', 'comparison', 'impact', 'analysis', 'result']):
+                        if len(line) > 20 and not line.startswith('|'):  # Avoid table rows
+                            key_points.append(line)
+                            if len(key_points) >= 3:  # Limit to top 3 key points
+                                break
+                
+                if key_points:
+                    summary = ' '.join(key_points)
+                    # Ensure summary is not too long but comprehensive
+                    if len(summary) > 500:
+                        summary = summary[:500] + "..."
+                    return summary
+            
+            # For policy overview queries, extract the main policy information
+            elif any(keyword in query_lower for keyword in ['policies', 'rules', 'schemes', 'overview']):
+                # Look for policy definitions and key features
+                lines = response_text.split('\n')
+                policy_info = []
+                
+                for line in lines:
+                    line = line.strip()
+                    if (line.startswith('**') or 'policy' in line.lower() or 'scheme' in line.lower()) and len(line) > 20:
+                        policy_info.append(line.replace('**', '').strip())
+                        if len(policy_info) >= 2:
+                            break
+                
+                if policy_info:
+                    summary = ' '.join(policy_info)
+                    if len(summary) > 400:
+                        summary = summary[:400] + "..."
+                    return summary
+            
+            # Default: Use first paragraph or first 300 characters
+            paragraphs = response_text.split('\n\n')
+            if len(paragraphs) > 0:
+                first_paragraph = paragraphs[0].strip()
+                if len(first_paragraph) > 300:
+                    return first_paragraph[:300] + "..."
+                return first_paragraph
+            
+            # Fallback to character limit
+            return response_text[:300] + "..." if len(response_text) > 300 else response_text
+            
+        except Exception as e:
+            logger.error(f"❌ Summary generation error: {e}")
+            return response_text[:200] + "..." if len(response_text) > 200 else response_text
+
 # Global instance
 groq_websocket_handler = GroqWebSocketHandler()