Update app.py

app.py

@@ -476,45 +476,106 @@ def format_results(analysis_result):
         </div>
         ''' if analysis_result["verification_needed"] else ''}
         
-        <div class="entity-section">
-            <h4>👤 People Detected</h4>
-            <ul>{''.join(f'<li>{person}</li>' for person in analysis_result['entities']['people']) or '<li>None detected</li>'}</ul>
-        </div>
-        
-        <div class="entity-section">
-            <h4>🏢 Organizations</h4>
-            <ul>{''.join(f'<li>{org}</li>' for org in analysis_result['entities']['organizations']) or '<li>None detected</li>'}</ul>
-        </div>
-        
-        <div class="entity-section">
-            <h4>📍 Locations</h4>
-            <ul>{''.join(f'<li>{loc}</li>' for loc in analysis_result['entities']['locations']) or '<li>None detected</li>'}</ul>
-        </div>
-        
-        <div class="entity-section">
-            <h4>🕒 Temporal References</h4>
-            <ul>{''.join(f'<li>{time}</li>' for time in analysis_result['entities']['temporal']) or '<li>None detected</li>'}</ul>
-        </div>
-        
-        <div class="entity-section">
-            <h4># Hashtags</h4>
-            <ul>{''.join(f'<li>{tag}</li>' for tag in analysis_result['entities']['hashtags']) or '<li>None detected</li>'}</ul>
+        <div class="grid grid-cols-2 gap-4">
+            <div class="space-y-4">
+                <div class="entity-section">
+                    <h4>👤 People Detected</h4>
+                    <ul>{''.join(f'<li>{person}</li>' for person in analysis_result['entities']['people']) or '<li>None detected</li>'}</ul>
+                </div>
+                
+                <div class="entity-section">
+                    <h4>🏢 Organizations</h4>
+                    <ul>{''.join(f'<li>{org}</li>' for org in analysis_result['entities']['organizations']) or '<li>None detected</li>'}</ul>
+                </div>
+                
+                <div class="entity-section">
+                    <h4>📍 Locations</h4>
+                    <ul>{''.join(f'<li>{loc}</li>' for loc in analysis_result['entities']['locations']) or '<li>None detected</li>'}</ul>
+                </div>
+            </div>
+            
+            <div class="space-y-4">
+                <div class="entity-section">
+                    <h4>🕒 Temporal References</h4>
+                    <ul>{''.join(f'<li>{time}</li>' for time in analysis_result['entities']['temporal']) or '<li>None detected</li>'}</ul>
+                </div>
+                
+                <div class="entity-section">
+                    <h4># Hashtags</h4>
+                    <ul>{''.join(f'<li>{tag}</li>' for tag in analysis_result['entities']['hashtags']) or '<li>None detected</li>'}</ul>
+                </div>
+                
+                {f'''
+                <div class="entity-section">
+                    <h4>🔗 Entity Relationships</h4>
+                    <ul>
+                        {''.join(f"""
+                        <li class="mb-2">
+                            <strong>{rel['source']}</strong> →
+                            <span class="text-blue-600">{rel['type'].replace('_to_', ' → ')}</span> →
+                            <strong>{rel['target']}</strong>
+                            <br/>
+                            <small class="text-gray-600">Confidence: {int(rel['confidence'] * 100)}%</small>
+                        </li>
+                        """ for rel in analysis_result['entity_relationships'])}
+                    </ul>
+                </div>
+                ''' if analysis_result.get('entity_relationships') else ''}
+            </div>
         </div>
         
         {f'''
-        <div class="alert-success">
+        <div class="alert-success mt-4">
             ✅ <strong>Event Validated:</strong> The extracted information meets confidence thresholds.
         </div>
         ''' if not analysis_result["verification_needed"] else ''}
         
         {f'''
         <div class="related-events">
-            <h4>Related Events</h4>
+            <h4>🔄 Related Events</h4>
             <ul>
-                {''.join(f'<li>{event["text"]} ({event["timestamp"]}) - Confidence: {int(event["confidence"] * 100)}%</li>' for event in analysis_result['related_events'])}
+                {''.join(f"""
+                <li class="mb-2">
+                    <div class="flex justify-between items-center">
+                        <div>{event["text"]}</div>
+                        <div class="text-sm text-gray-600">
+                            {event["timestamp"]} | 
+                            Confidence: {int(event["confidence"] * 100)}%
+                            {f' | Shared Entities: {event["shared_entities"]}' if event.get("shared_entities") else ''}
+                        </div>
+                    </div>
+                </li>
+                """ for event in analysis_result['related_events'])}
             </ul>
         </div>
         ''' if analysis_result.get('related_events') else ''}
+        
+        <div class="entity-stats mt-4 p-4 bg-gray-50 rounded-lg">
+            <h4 class="mb-2">📊 Analysis Metrics</h4>
+            <div class="grid grid-cols-3 gap-4 text-sm">
+                <div>
+                    <strong>Confidence Breakdown:</strong>
+                    <ul class="mt-1">
+                        <li>Base Confidence: {int(analysis_result['confidence'] * 70)}%</li>
+                        <li>Entity Boost: {int((analysis_result['confidence'] - 0.7 if analysis_result['confidence'] > 0.7 else 0) * 100)}%</li>
+                    </ul>
+                </div>
+                <div>
+                    <strong>Entity Coverage:</strong>
+                    <ul class="mt-1">
+                        <li>Types Detected: {len([t for t in ['people', 'organizations', 'locations', 'temporal', 'hashtags'] if analysis_result['entities'].get(t)])}</li>
+                        <li>Total Entities: {sum(len(e) for e in analysis_result['entities'].values() if isinstance(e, list))}</li>
+                    </ul>
+                </div>
+                <div>
+                    <strong>Relationships:</strong>
+                    <ul class="mt-1">
+                        <li>Direct: {len(analysis_result.get('entity_relationships', []))}</li>
+                        <li>Related Events: {len(analysis_result.get('related_events', []))}</li>
+                    </ul>
+                </div>
+            </div>
+        </div>
     </div>
     """
     return html