Update cust_types.py

cust_types.py

@@ -1,16 +1,54 @@
 from typing import List, Optional
-from agent import (
-    Code,
-    Prompt,
-    AppType,
-    File,
-    Space,
-    Tutorial,
-    WebApp,
-    GradioApp,
-    StreamlitApp,
-    ReactApp,
-)
+
+# Type definitions (moved from agent/__init__.py)
+class Code:
+    def __init__(self, code: str, language: str):
+        self.code = code
+        self.language = language
+
+class Prompt:
+    def __init__(self, prompt: str):
+        self.prompt = prompt
+
+class AppType:
+    WEB_APP = "WebApp"
+    GRADIO_APP = "GradioApp"
+    STREAMLIT_APP = "StreamlitApp"
+    REACT_APP = "ReactApp"
+
+class File:
+    def __init__(self, name: str, content: str, language: str):
+        self.name = name
+        self.content = content
+        self.language = language
+
+class Space:
+    def __init__(self, space: str):
+        self.space = space
+
+class Tutorial:
+    def __init__(self, tutorial: str):
+        self.tutorial = tutorial
+
+class WebApp:
+    def __init__(self, code: str, language: str):
+        self.code = code
+        self.language = language
+
+class GradioApp:
+    def __init__(self, code: str, language: str):
+        self.code = code
+        self.language = language
+
+class StreamlitApp:
+    def __init__(self, code: str, language: str):
+        self.code = code
+        self.language = language
+
+class ReactApp:
+    def __init__(self, code: str, language: str):
+        self.code = code
+        self.language = language
 
 def createLlamaPrompt(
     app_type: AppType,