Update app.py

app.py

@@ -14,9 +14,11 @@ from urllib.parse import urlparse
 import warnings
 
 # Configure matplotlib for better font handling
-plt.rcParams['font.family'] = ['DejaVu Sans', 'Arial', 'Liberation Sans']
+plt.rcParams['font.family'] = ['DejaVu Sans']
 plt.rcParams['font.size'] = 10
+plt.rcParams['font.weight'] = 'normal'
 warnings.filterwarnings('ignore', category=UserWarning)
+warnings.filterwarnings('ignore', message='.*Font family.*not found.*')
 
 def clean_text_for_display(text):
     """Clean text to remove characters that might cause font issues."""
@@ -263,8 +265,28 @@ def build_knowledge_graph(entities_data):
         
         # Convert to PIL Image
         fig.canvas.draw()
-        img_array = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
-        img_array = img_array.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+        
+        # Handle different matplotlib versions
+        try:
+            # Try newer method first
+            img_array = np.frombuffer(fig.canvas.buffer_rgba(), dtype=np.uint8)
+            img_array = img_array.reshape(fig.canvas.get_width_height()[::-1] + (4,))
+            # Convert RGBA to RGB
+            img_array = img_array[:, :, :3]
+        except AttributeError:
+            try:
+                # Fallback to older method
+                img_array = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+                img_array = img_array.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+            except AttributeError:
+                # Final fallback - save to buffer
+                buf = io.BytesIO()
+                fig.savefig(buf, format='png', bbox_inches='tight')
+                buf.seek(0)
+                from PIL import Image
+                pil_image = Image.open(buf).convert('RGB')
+                plt.close(fig)
+                return pil_image
         
         from PIL import Image
         pil_image = Image.fromarray(img_array)
@@ -280,9 +302,28 @@ def build_knowledge_graph(entities_data):
         ax.set_title("Knowledge Graph Error")
         ax.axis('off')
         
-        fig.canvas.draw()
-        img_array = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
-        img_array = img_array.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+        # Handle different matplotlib versions for error image
+        try:
+            # Try newer method first
+            fig.canvas.draw()
+            img_array = np.frombuffer(fig.canvas.buffer_rgba(), dtype=np.uint8)
+            img_array = img_array.reshape(fig.canvas.get_width_height()[::-1] + (4,))
+            img_array = img_array[:, :, :3]  # Convert RGBA to RGB
+        except AttributeError:
+            try:
+                # Fallback to older method
+                fig.canvas.draw()
+                img_array = np.frombuffer(fig.canvas.tostring_rgb(), dtype=np.uint8)
+                img_array = img_array.reshape(fig.canvas.get_width_height()[::-1] + (3,))
+            except AttributeError:
+                # Final fallback - save to buffer
+                buf = io.BytesIO()
+                fig.savefig(buf, format='png', bbox_inches='tight')
+                buf.seek(0)
+                from PIL import Image
+                pil_image = Image.open(buf).convert('RGB')
+                plt.close(fig)
+                return pil_image
         
         from PIL import Image
         pil_image = Image.fromarray(img_array)
@@ -421,4 +462,4 @@ if __name__ == "__main__":
                 show_error=True
             )
         except Exception as e2:
-            print(f"Complete failure: {e2}")
+            print(f"Complete failure: {e2}")