Fix Attempt for coding module

app.py

@@ -2,7 +2,7 @@ import os
 import uuid
 import shutil
 import traceback
-
+import json
 import gradio as gr
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
@@ -31,12 +31,10 @@ OUTPUT_DIR = "./output"
 os.makedirs(TEMP_DIR, exist_ok=True)
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-
 @api.get("/")
 def health_check():
     return {"status": "ok"}
 
-
 @api.post("/process-wireframe")
 async def process_wireframe_api(image: UploadFile = File(...)):
     file_id = str(uuid.uuid4())
@@ -61,7 +59,14 @@ async def process_wireframe_api(image: UploadFile = File(...)):
         
         json_path = results.get("json_path")
         
-        #Generate styled HTML using CodingModule
+        # Read the JSON file content and return it directly
+        json_content = None
+        if json_path and os.path.exists(json_path):
+            with open(json_path, "r") as f:
+                json_content = json.load(f)
+        
+        # Generate styled HTML using CodingModule
+        html_path = None
         try:
             html_generator = HTMLGenerator(json_path)
             html_filename = f"{file_id}_styled.html"
@@ -71,11 +76,18 @@ async def process_wireframe_api(image: UploadFile = File(...)):
             print(f"Warning: Could not generate styled HTML: {e}")
             html_path = results.get("html_path")
         
+        # Read the HTML file content
+        html_content = None
+        if html_path and os.path.exists(html_path):
+            with open(html_path, "r", encoding="utf-8") as f:
+                html_content = f.read()
+        
         return {
             "success": True,
-            "json_path": json_path,
-            "html_path": html_path,
-            "total_elements": len(results["normalized_elements"])
+            "json_data": json_content,  # Return actual JSON content instead of path
+            "html_content": html_content,  # Return actual HTML content instead of path
+            "total_elements": len(results["normalized_elements"]),
+            "normalized_elements": results.get("normalized_elements", [])  # Include normalized elements
         }
 
     except Exception as e:
@@ -102,18 +114,6 @@ async def serve_output_file(filename: str):
             return FileResponse(file_path)
     return JSONResponse(status_code=404, content={"error": "File not Found"})
 
-# # Legacy endpoint for just for backward compatibility
-# @api.get("/view-html/{file_id}")
-# async def view_html(file_id: str):
-#     html_path = os.path.join(OUTPUT_DIR, f"{file_id}.html")
-#     if os.path.exists(html_path):
-#         return FileResponse(html_path, media_type="text/html")
-#     # Try with _styled suffix
-#     html_path = os.path.join(OUTPUT_DIR, f"{file_id}_styled.html")
-#     if os.path.exists(html_path):
-#         return FileResponse(html_path, media_type="text/html")
-#     return JSONResponse(status_code=404, content={"error": "File not found"})
-
 # -----------------------------------------------------------------------------
 # GRADIO (for Hugging Face UI)
 # -----------------------------------------------------------------------------
@@ -216,14 +216,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Wireframe Layout Normalizer") as d
     gr.Markdown("""
     ---
     ### How to use:
-    1. *Upload* a wireframe image (hand-drawn or digital)
-    2. *Click* "Process Wireframe" and wait for processing
-    3. *Download* the JSON file for structured data
-    4. *Download* the HTML file to see a styled preview
+    1. Upload a wireframe image (hand-drawn or digital)
+    2. Click "Process Wireframe" and wait for processing
+    3. Download the JSON file for structured data
+    4. Download the HTML file to see a styled preview
     
     ### Output Files:
-    - *JSON*: Contains all detected elements with positions, types, and grid data
-    - *HTML*: Beautiful, responsive preview with modern styling
+    - JSON: Contains all detected elements with positions, types, and grid data
+    - HTML: Beautiful, responsive preview with modern styling
     
     ### Technical Details:
     - Model: TensorFlow-based object detection
@@ -231,13 +231,12 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Wireframe Layout Normalizer") as d
     - Supported Elements: Buttons, Checkboxes, Text Fields, Images, Paragraphs, Text, Navbars
     """)
 
-
 # -----------------------------------------------------------------------------
 # ENTRY POINT (THIS IS IMPORTANT)
 # -----------------------------------------------------------------------------
 app = gr.mount_gradio_app(api, demo, path="/")
 
-if __name__ == "__main__":
+if __name__ == "_main_":
     import uvicorn
     # For local testing
     uvicorn.run(app, host="0.0.0.0", port=7860)