Update app.py

app.py

@@ -20,26 +20,37 @@ async def extract_data(content, schema_text, progress=gr.Progress()):
         result = await system.extract_structured_data(content, schema)
         
         extracted_data = json.dumps(result["data"], indent=2)
-        confidence = f"{result['overall_confidence']:.1%}"
         metadata = result["extraction_metadata"]
         
-        analysis = f"""Complexity Tier: {metadata['complexity_tier']}
+        total_expected = len(schema.get('properties', {}))
+        extracted_count = len([k for k, v in result["data"].items() if v is not None and v != ""])
+        completeness = extracted_count / total_expected if total_expected > 0 else 0
+        
+        analysis = f"""Fields Extracted: {extracted_count}/{total_expected} ({completeness:.1%})
+Complexity Tier: {metadata['complexity_tier']}
 Processing Stages: {metadata['stages_executed']}
-Estimated Cost: ${metadata['estimated_cost']:.3f}
 Processing Time: {metadata['actual_processing_time']:.2f}s
 Schema Compliance: {metadata['schema_compliance']:.1%}"""
         
-        if result["review_flags"]:
-            analysis += f"\nReview Flags: {', '.join(result['review_flags'])}"
-            analysis += f"\nReview Time: {metadata['recommended_review_time']} minutes"
+        status_flags = result.get("review_flags", [])
+        
+        if completeness >= 0.8 and not any(flag in ["incomplete_extraction", "low_quality", "schema_violations"] for flag in status_flags):
+            status = "Success"
+        elif completeness >= 0.5:
+            status = "Partial Success"
+        else:
+            status = "Incomplete"
+        
+        if status_flags:
+            analysis += f"\nIssues: {', '.join(status_flags)}"
         
         progress(1.0, desc="Complete")
-        return extracted_data, confidence, analysis, "Success"
+        return extracted_data, analysis, status
         
     except json.JSONDecodeError as e:
-        return "", "0%", f"Invalid JSON Schema: {str(e)}", "Schema Error"
+        return "", f"Invalid JSON Schema: {str(e)}", "Schema Error"
     except Exception as e:
-        return "", "0%", f"Extraction Error: {str(e)}", "Error"
+        return "", f"Extraction Error: {str(e)}", "Error"
 
 def extract_wrapper(content, schema_text):
     return asyncio.run(extract_data(content, schema_text))
@@ -424,7 +435,6 @@ with gr.Blocks(title="Unstructured to Structured Converter", theme=gr.themes.Def
             
         with gr.Column(scale=1):
             gr.Markdown("### Results")
-            confidence_output = gr.Textbox(label="Confidence")
             metadata_output = gr.Textbox(label="Analysis", lines=8)
             status_output = gr.Textbox(label="Status")
     
@@ -451,12 +461,12 @@ with gr.Blocks(title="Unstructured to Structured Converter", theme=gr.themes.Def
     extract_btn.click(
         fn=extract_wrapper,
         inputs=[content_input, schema_input],
-        outputs=[output_json, confidence_output, metadata_output, status_output]
+        outputs=[output_json, metadata_output, status_output]
     )
     
     clear_btn.click(
         lambda: ("", "", "", "", ""),
-        outputs=[content_input, schema_input, output_json, confidence_output, metadata_output]
+        outputs=[content_input, schema_input, output_json, metadata_output, status_output]
     )
 
 if __name__ == "__main__":