Update app.py

app.py

@@ -2,39 +2,58 @@ import gradio as gr
 import json
 
 def extract_bq_codes_from_json(input_data):
-    # Handle both API and UI inputs
+    # Debug: Print raw input
+    print("Raw input received:", type(input_data), input_data)
+    
+    # Handle all possible input formats
     if isinstance(input_data, dict):
-        # API case - extract from 'data' field if present
-        if "data" in input_data:
+        if "data" in input_data:  # Gradio API format
             try:
                 if isinstance(input_data["data"][0], str):
                     data = json.loads(input_data["data"][0])
                 else:
                     data = input_data["data"][0]
-            except:
-                return {"error": "Invalid JSON in data field"}
-        else:
+            except Exception as e:
+                print("API format parse error:", e)
+                return {"error": f"Invalid API format: {str(e)}"}
+        else:  # Raw JSON format
             data = input_data
-    else:
-        # UI case - direct string input
+    elif isinstance(input_data, str):  # Direct text input
         try:
             data = json.loads(input_data)
-        except:
-            return {"error": "Invalid JSON string"}
+        except Exception as e:
+            print("String parse error:", e)
+            return {"error": f"Invalid JSON string: {str(e)}"}
+    else:
+        return {"error": "Input must be JSON string or object"}
     
-    # Extraction logic
-    codes = set()
-    for project in data.get("projecttemplates", []):
-        for detail in project.get("projecttemplatedetails", []):
-            if (bqcode := detail.get("bqcodelibrary", {}).get("bqcode")):
-                codes.add(bqcode)
-    return list(codes)
+    # Extraction logic with defensive programming
+    try:
+        codes = set()
+        for project in data.get("projecttemplates", []):
+            for detail in project.get("projecttemplatedetails", []):
+                if isinstance(detail, dict):
+                    bqcode = detail.get("bqcodelibrary", {}).get("bqcode")
+                    if bqcode:
+                        codes.add(bqcode)
+        return {"success": True, "codes": list(codes)}
+    except Exception as e:
+        return {"error": f"Processing error: {str(e)}", "received_data": data}
+
+# Create interface with explicit examples
+examples = [
+    {'projecttemplates': [{'projecttemplatedetails': [{'bqcodelibrary': {'bqcode': 'BQ123'}}]}]},
+    '{"projecttemplates":[{"projecttemplatedetails":[{"bqcodelibrary":{"bqcode":"BQ456"}}]}]'
+]
 
 iface = gr.Interface(
     fn=extract_bq_codes_from_json,
-    inputs=gr.Textbox(lines=10, label="JSON Input"),
-    outputs=gr.JSON(),
-    title="BQ Code Extractor"
+    inputs=gr.Textbox(lines=10, placeholder="Paste JSON here...", label="JSON Input"),
+    outputs=gr.JSON(label="Results"),
+    examples=examples,
+    title="BQ Code Extractor",
+    description="Extracts BQ codes from JSON. Accepts both raw JSON objects and strings."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch(debug=True)