Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 import json
 from graphviz import Digraph
-import os
-from typing import Dict, Any
+import io
+import base64
 
 def generate_concept_map(json_input: str) -> str:
     """
-    Generates a concept map from structured JSON
+    Generate concept map from JSON and return as base64 image
     
     Args:
         json_input (str): JSON describing the concept map structure.
@@ -24,19 +24,14 @@ def generate_concept_map(json_input: str) -> str:
             }
     
     Returns:
-        str: Image URL (compatible with Hugging Face Spaces)
-    
-    Raises:
-        ValueError: If JSON doesn't meet required format
+        str: Base64 data URL of the generated concept map
     """
     try:
-        # Validate input
         if not json_input.strip():
             return "Error: Empty input"
             
         data = json.loads(json_input)
         
-        # Validate basic structure
         if 'central_node' not in data or 'nodes' not in data:
             raise ValueError("Missing required fields: central_node or nodes")
 
@@ -119,15 +114,13 @@ def generate_concept_map(json_input: str) -> str:
                     fontsize='8'
                 )
 
-        # Save image
-        filename = f"/tmp/concept_map_{hash(json_input)}.gv"
-        dot.render(filename, format='png', cleanup=True)
-        
-        # Return Hugging Face compatible URL
-        return f"/file={filename}.png"
+        # Convert to base64 image
+        img_data = dot.pipe(format='png')
+        img_base64 = base64.b64encode(img_data).decode()
+        return f"data:image/png;base64,{img_base64}"
 
     except json.JSONDecodeError:
-        return "Error: Invalid JSON"
+        return "Error: Invalid JSON format"
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -141,16 +134,15 @@ if __name__ == "__main__":
         ),
         outputs=gr.Textbox(
             label="Image URL",
-            placeholder="Concept map URL will be generated here"
+            placeholder="Base64 image URL will appear here"
         ),
         title="Concept Map Generator",
-        description="Create concept maps from JSON (Claude compatible)"
+        description="Create concept maps from JSON for web display"
     )
     
-    # Hugging Face Spaces configuration
     demo.launch(
         mcp_server=True,
         share=False,
-        server_port=int(os.getenv('PORT', 7860)),
+        server_port=7860,
         server_name="0.0.0.0"
     )