Update custom_types.py

custom_types.py

@@ -1,72 +1,134 @@
-from custom_types import SomeClass  # Updated import
+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
 
+    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, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
         self.code = code
         self.app_type = app_type
         self.space = space
         self.tutorial = tutorial
-        self._id = None
+        self._id = self.generate_id()
 
-    def generate_id(self):
-        # Simple id generation for demonstration purposes
-        return f"app-{id(self)}"
+    def generate_id(self) -> str:
+        return f"app-{hashlib.md5(str(time.time()).encode()).hexdigest()}"
 
-    def launch(self):
-        # Simulate launching the app
+    def launch(self) -> None:
         print(f"Launching app {self._id}...")
 
 class WebApp(App):
     def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
         super().__init__(code, app_type, space, tutorial)
-        self._id = self.generate_id()
+
+    def launch(self) -> None:
+        super().launch()
+        print("WebApp specific launch sequence...")
 
 class GradioApp(App):
     def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
         super().__init__(code, app_type, space, tutorial)
-        self._id = self.generate_id()
+
+    def launch(self) -> None:
+        super().launch()
+        print("GradioApp specific launch sequence...")
 
 class StreamlitApp(App):
     def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
         super().__init__(code, app_type, space, tutorial)
-        self._id = self.generate_id()
+
+    def launch(self) -> None:
+        super().launch()
+        print("StreamlitApp specific launch sequence...")
 
 class ReactApp(App):
     def __init__(self, code: Code, app_type: AppType, space: Space, tutorial: Tutorial):
         super().__init__(code, app_type, space, tutorial)
-        self._id = self.generate_id()
+
+    def launch(self) -> None:
+        super().launch()
+        print("ReactApp specific launch sequence...")
 
 # Example usage
-code = Code("example code", "python")
-app_type = AppType("web")
-space = Space("example space")
-tutorial = Tutorial("example tutorial")
+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()
+    app = WebApp(code, app_type, space, tutorial)
+    app.launch()