Create app.py

app.py

@@ -0,0 +1,145 @@
+import json
+import gradio as gr
+
+from llm_output_parser import parse_json, parse_jsons, parse_xml
+
+
+def parse_single_json(text):
+    """Parse a single JSON object from text."""
+    try:
+        result = parse_json(text)
+        return json.dumps(result, indent=2)
+    except (ValueError, TypeError) as e:
+        return f"Error parsing JSON: {str(e)}"
+
+
+def parse_multiple_jsons(text):
+    """Parse multiple JSON objects from text."""
+    try:
+        results = parse_jsons(text)
+        formatted_results = []
+
+        for i, result in enumerate(results):
+            formatted_results.append(f"JSON {i+1}:\n{json.dumps(result, indent=2)}")
+
+        return "\n\n".join(formatted_results)
+    except (ValueError, TypeError) as e:
+        return f"Error parsing JSONs: {str(e)}"
+
+
+def parse_xml_to_json(text):
+    """Parse XML from text and convert to JSON format."""
+    try:
+        result = parse_xml(text)
+        return json.dumps(result, indent=2)
+    except (ValueError, TypeError) as e:
+        return f"Error parsing XML: {str(e)}"
+
+
+def process_text(text, parser_type):
+    """Process text based on selected parser type."""
+    if not text.strip():
+        return "Please enter some text to parse."
+
+    if parser_type == "Single JSON":
+        return parse_single_json(text)
+    elif parser_type == "Multiple JSONs":
+        return parse_multiple_jsons(text)
+    elif parser_type == "XML":
+        return parse_xml_to_json(text)
+    else:
+        return "Invalid parser type selected."
+
+
+# Example texts for the interface
+example_json = """
+```json
+{
+  "name": "John Doe",
+  "age": 30,
+  "isEmployed": true,
+  "address": {
+    "street": "123 Main St",
+    "city": "Anytown"
+  }
+}
+```
+"""
+
+example_multiple_jsons = """
+Here are some JSON objects:
+
+```json
+{"id": 1, "name": "Product A"}
+```
+
+And another one:
+
+```json
+{"id": 2, "name": "Product B"}
+```
+"""
+
+example_xml = """
+```xml
+<?xml version="1.0" encoding="UTF-8"?>
+<root>
+  <person id="1">
+    <name>John Doe</name>
+    <age>30</age>
+    <address>
+      <street>123 Main St</street>
+      <city>Anytown</city>
+    </address>
+  </person>
+</root>
+```
+"""
+
+# Create Gradio interface
+with gr.Blocks(title="LLM Output Parser") as demo:
+    gr.Markdown("# LLM Output Parser")
+    gr.Markdown("Extract structured data from text containing JSON or XML")
+
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(
+                label="Input Text",
+                placeholder="Paste text containing JSON or XML here...",
+                lines=15,
+            )
+            parser_type = gr.Radio(
+                choices=["Single JSON", "Multiple JSONs", "XML"],
+                label="Parser Type",
+                value="Single JSON",
+            )
+            parse_button = gr.Button("Parse", variant="primary")
+
+        with gr.Column():
+            output_text = gr.Textbox(label="Parsed Result", lines=15)
+
+    # Examples
+    with gr.Accordion("Example Inputs", open=False):
+        gr.Examples(
+            examples=[
+                [example_json, "Single JSON"],
+                [example_multiple_jsons, "Multiple JSONs"],
+                [example_xml, "XML"],
+            ],
+            inputs=[input_text, parser_type],
+        )
+
+    # Set up event handler
+    parse_button.click(
+        fn=process_text, inputs=[input_text, parser_type], outputs=output_text
+    )
+
+    gr.Markdown(
+        "## How to use\n"
+        "1. Paste text containing JSON or XML\n"
+        "2. Select the parser type\n"
+        "3. Click 'Parse' to extract structured data"
+    )
+
+if __name__ == "__main__":
+    demo.launch()