add graph data output for api

app.py

@@ -423,8 +423,8 @@ def visualize_graph(graph: GPGraph) -> Optional[str]:
 # Gradio Interface
 # ============================================================================
 
-def process_sentence(sentence: str) -> Tuple[Optional[str], str]:
-    """Process a sentence and return the visualization."""
+def process_sentence(sentence: str) -> Tuple[Optional[str], str, Optional[Dict]]:
+    """Process a sentence and return the visualization and graph data."""
     try:
         # Load model if not already loaded
         model_loader = ModelLoader()
@@ -436,21 +436,30 @@ def process_sentence(sentence: str) -> Tuple[Optional[str], str]:
         graph, error = parse_sentence(sentence, model_loader, use_constrained=use_constrained)
         
         if error:
-            return None, error
+            return None, error, None
         
         # Visualize
         img_path = visualize_graph(graph)
         
+        # Convert graph to dictionary (for JSON output)
+        graph_data = None
+        if graph is not None:
+            try:
+                graph_data = graph.data()
+            except Exception as e:
+                print(f"Warning: Failed to serialize graph to dict: {e}")
+                graph_data = None
+        
         decoding_mode = "constrained" if (use_constrained and model_loader.constrained_decoder) else "simple"
         if img_path:
-            return img_path, f"Graph generated successfully! (Decoding: {decoding_mode})"
+            return img_path, f"Graph generated successfully! (Decoding: {decoding_mode})", graph_data
         else:
-            return None, "Failed to generate visualization."
+            return None, "Failed to generate visualization.", graph_data
             
     except Exception as e:
         import traceback
         error_msg = f"Error: {str(e)}\n{traceback.format_exc()}"
-        return None, error_msg
+        return None, error_msg, None
 
 
 def load_model_on_startup():
@@ -500,6 +509,10 @@ with gr.Blocks(title="Lingua Graph Parser") as demo:
             type="filepath",
             height=600
         )
+        output_json = gr.JSON(
+            label="Graph Data (JSON)",
+            visible=False
+        )
     
     # Load model on startup
     demo.load(
@@ -511,7 +524,7 @@ with gr.Blocks(title="Lingua Graph Parser") as demo:
     parse_btn.click(
         fn=process_sentence,
         inputs=[sentence_input],
-        outputs=[output_image, output_text]
+        outputs=[output_image, output_text, output_json]
     )
     
     # Example sentences
@@ -537,7 +550,6 @@ with gr.Blocks(title="Lingua Graph Parser") as demo:
     - Discourse markers
     - And more...
     
-    **Model**: [rudaoshi/lingua](https://huggingface.co/rudaoshi/lingua)
     """)