Update app.py

app.py

@@ -6,6 +6,7 @@ import cv2
 import pytesseract
 from PIL import Image
 import io
+import numpy as np
 
 # Define the LiteLLMModel
 model = LiteLLMModel(model_id="gpt-4o")
@@ -53,8 +54,8 @@ uploaded_file = st.file_uploader("Upload a web design image", type=["png", "jpg"
 if st.button("Extract and Generate Code"):
     if uploaded_file is not None:
         image_data = uploaded_file.read()
-        components = agent.run("extract_components", image_data=image_data)
-        code = agent.run("generate_code", components=components)
+        components = agent.run(f"extract_components {image_data}")
+        code = agent.run(f"generate_code {components}")
         st.write("Extracted Components:", components)
         st.write("Generated Code:", code)
     else: