Update cust_types.py

cust_types.py

@@ -8,54 +8,137 @@ class AppType(Enum):
     STREAMLIT_APP = "streamlit_app"
     REACT_APP = "react_app"
 
+import hashlib
+import os
+import time
+from typing import Optional
+
+class Code:
+    def __init__(self, code: str, language: str):
+        self.code = code
+        self.language = language
+
+    def get_code_length(self) -> int:
+        return len(self.code)
+
+    def get_language(self) -> str:
+        return self.language
+
+    def hash_code(self) -> str:
+        return hashlib.sha256(self.code.encode()).hexdigest()
+
 class Prompt:
     def __init__(self, prompt: str):
         self.prompt = prompt
 
+    def get_prompt_length(self) -> int:
+        return len(self.prompt)
+
+    def preview_prompt(self, length: int = 50) -> str:
+        return self.prompt[:length] + ('...' if len(self.prompt) > length else '')
+
+class AppType:
+    def __init__(self, app_type: str):
+        self.app_type = app_type
+
+    def is_web_app(self) -> bool:
+        return self.app_type.lower() == 'web'
+
+    def is_gradio_app(self) -> bool:
+        return self.app_type.lower() == 'gradio'
+
+    def is_streamlit_app(self) -> bool:
+        return self.app_type.lower() == 'streamlit'
+
+    def is_react_app(self) -> bool:
+        return self.app_type.lower() == 'react'
+
+class File:
+    def __init__(self, path: str, content: str):
+        self.path = path
+        self.content = content
+
+    def save(self) -> None:
+        with open(self.path, 'w') as file:
+            file.write(self.content)
+
+    def load(self) -> None:
+        if os.path.exists(self.path):
+            with open(self.path, 'r') as file:
+                self.content = file.read()
+
+    def get_file_size(self) -> int:
+        return len(self.content)
+
 class Space:
     def __init__(self, space: str):
         self.space = space
 
+    def get_space_name(self) -> str:
+        return self.space
+
 class Tutorial:
     def __init__(self, tutorial: str):
         self.tutorial = tutorial
 
-class File:
-    def __init__(self, name: str, content: str, language: str):
-        self.name = name
-        self.content = content
-        self.language = language
+    def get_tutorial_length(self) -> int:
+        return len(self.tutorial)
+
+    def preview_tutorial(self, length: int = 100) -> str:
+        return self.tutorial[:length] + ('...' if len(self.tutorial) > length else '')
 
 class App:
-    def __init__(self, app_type: AppType, app_name: str, app_description: str, app_features: List[str], app_dependencies: List[str], app_space: Optional[Space] = None, app_tutorial: Optional[Tutorial] = None):
+    def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
+        self.code = code
         self.app_type = app_type
-        self.app_name = app_name
-        self.app_description = app_description
-        self.app_features = app_features
-        self.app_dependencies = app_dependencies
-        self.app_space = app_space
-        self.app_tutorial = app_tutorial
+        self.space = space
+        self.tutorial = tutorial
+        self._id = self.generate_id()
+
+    def generate_id(self) -> str:
+        return f"app-{hashlib.md5(str(time.time()).encode()).hexdigest()}"
+
+    def launch(self) -> None:
+        print(f"Launching app {self._id}...")
 
 class WebApp(App):
-    def __init__(self, code: str, language: str, app_type: AppType = AppType.WEB_APP, app_name: str = "WebApp", app_description: str = "A simple web application", app_features: List[str] = ["Basic HTML structure", "Title and description", "Simple styling"], app_dependencies: List[str] = ["HTML", "CSS"], app_space: Optional[Space] = None, app_tutorial: Optional[Tutorial] = None):
-        super().__init__(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial)
-        self.code = code
-        self.language = language
+    def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
+        super().__init__(code, app_type, space, tutorial)
+
+    def launch(self) -> None:
+        super().launch()
+        print("WebApp specific launch sequence...")
 
 class GradioApp(App):
-    def __init__(self, code: str, language: str, app_type: AppType = AppType.GRADIO_APP, app_name: str = "GradioApp", app_description: str = "A simple Gradio application", app_features: List[str] = ["Basic Gradio interface", "Input and output components", "Simple function"], app_dependencies: List[str] = ["Gradio", "Python"], app_space: Optional[Space] = None, app_tutorial: Optional[Tutorial] = None):
-        super().__init__(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial)
-        self.code = code
-        self.language = language
+    def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
+        super().__init__(code, app_type, space, tutorial)
+
+    def launch(self) -> None:
+        super().launch()
+        print("GradioApp specific launch sequence...")
 
 class StreamlitApp(App):
-    def __init__(self, code: str, language: str, app_type: AppType = AppType.STREAMLIT_APP, app_name: str = "StreamlitApp", app_description: str = "A simple Streamlit application", app_features: List[str] = ["Basic Streamlit interface", "Title and description", "Simple layout"], app_dependencies: List[str] = ["Streamlit", "Python"], app_space: Optional[Space] = None, app_tutorial: Optional[Tutorial] = None):
-        super().__init__(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial)
-        self.code = code
-        self.language = language
+    def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
+        super().__init__(code, app_type, space, tutorial)
+
+    def launch(self) -> None:
+        super().launch()
+        print("StreamlitApp specific launch sequence...")
 
 class ReactApp(App):
-    def __init__(self, code: str, language: str, app_type: AppType = AppType.REACT_APP, app_name: str = "ReactApp", app_description: str = "A simple React application", app_features: List[str] = ["Basic React component", "JSX syntax", "State management"], app_dependencies: List[str] = ["React", "JavaScript"], app_space: Optional[Space] = None, app_tutorial: Optional[Tutorial] = None):
-        super().__init__(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial)
-        self.code = code
-        self.language = language
+    def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
+        super().__init__(code, app_type, space, tutorial)
+
+    def launch(self) -> None:
+        super().launch()
+        print("ReactApp specific launch sequence...")
+
+# Example usage
+if __name__ == "__main__":
+    code = Code("example code", "python")
+    app_type = AppType("web")
+    space = Space("example space")
+    tutorial = Tutorial("example tutorial")
+
+    app = WebApp(code, app_type, space, tutorial)
+    app.launch()