Update app.py

app.py

@@ -399,106 +399,6 @@ def create_interface():
         
     return demo
 
-
-# Add FastAPI integration for smolagents
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-from typing import Optional
-
-# Import smolagents
-from smolagents.agent import Agent
-
-# Create FastAPI app to mount with Gradio
-api = FastAPI()
-
-# Define API models
-class ProductURLRequest(BaseModel):
-    url: str
-    format: Optional[str] = "json"
-
-class ProductDataAgent(Agent):
-    def __init__(self):
-        super().__init__(
-            name="ProductDataAgent",
-            description="Extracts product data from IKEA URLs"
-        )
-    
-    def run(self, url: str):
-        try:
-            extraction_result = extractor.extract_images_from_url(url)
-            return extraction_result.to_dict()
-        except Exception as e:
-            return {
-                "error": str(e),
-                "status": "failed"
-            }
-
-@api.post("/api/extract")
-async def extract_product_data(request: ProductURLRequest):
-    try:
-        agent = ProductDataAgent()
-        result = agent.run(request.url)
-        
-        if request.format == "markdown":
-            # Add markdown formatting (implementation from smolagent_api.py)
-            def format_as_markdown(data: Dict[str, Any]) -> str:
-                """Convert the extracted data to markdown format for nicer display"""
-                result = []
-                
-                # Add request ID
-                result.append(f"### Request ID: {data.get('request_id', 'N/A')}")
-                result.append("")
-                
-                # Add images section
-                result.append("### Images")
-                images = data.get("images", {})
-                if images:
-                    for img_id, img_info in images.items():
-                        result.append(f"- **ID**: {img_id}")
-                        result.append(f"  - **URL**: {img_info.get('url', 'N/A')}")
-                        result.append(f"  - **Type**: {img_info.get('type', 'unknown')}")
-                        result.append(f"  - **Alt**: {img_info.get('alt', 'N/A')}")
-                        result.append("")
-                else:
-                    result.append("No images found.")
-                    result.append("")
-                
-                # Add measurements section
-                result.append("### Measurements")
-                measurements = data.get("measurements", {})
-                if measurements:
-                    for key, value in measurements.items():
-                        result.append(f"- **{key.title()}**: {value}")
-                    result.append("")
-                else:
-                    result.append("No measurements found.")
-                    result.append("")
-                
-                # Add materials section
-                result.append("### Materials")
-                materials = data.get("materials", {})
-                if materials:
-                    for key, value in materials.items():
-                        result.append(f"- **{key.title()}**: {value}")
-                    result.append("")
-                else:
-                    result.append("No materials information found.")
-                    result.append("")
-                
-                return "\n".join(result)
-            result["markdown"] = format_as_markdown(result)
-        
-        return {
-            "status": "success",
-            "data": result
-        }
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
-# Update the launch code to include the FastAPI app
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(server_name="0.0.0.0", 
-                server_port=7860,
-                favicon_path=None,
-                app_kwargs={"app": api})  # Mount FastAPI on Gradio
+    demo.launch()