Rename logx/prompts.mts to logx/prompts.py

logx/prompts.mts

@@ -1,38 +0,0 @@
-from string import Template
-
-def create_prompt(app_type: str, app_name: str, app_description: str, app_features: list[str], app_dependencies: list[str], app_space: str, app_tutorial: str) -> str:
-    """Creates a dynamic prompt based on the provided information."""
-
-    prompt_template = Template("""
-        Create a $app_type app called $app_name that does the following:
-        $app_description
-        The app should have the following features:
-        $app_features
-        The app should use the following dependencies:
-        $app_dependencies
-        The app should be deployed to $app_space.
-        The app should have the following tutorial:
-        $app_tutorial
-    """)
-
-    return prompt_template.substitute(
-        app_type=app_type,
-        app_name=app_name,
-        app_description=app_description,
-        app_features="\n".join(app_features),
-        app_dependencies="\n".join(app_dependencies),
-        app_space=app_space,
-        app_tutorial=app_tutorial,
-    )
-
-# Example usage:
-app_type = "web app"
-app_name = "MyWebApp"
-app_description = "This app displays a welcome message."
-app_features = ["Display a welcome message", "Show the current time"]
-app_dependencies = ["Flask", "Jinja2"]
-app_space = "Heroku"
-app_tutorial = "1. Install dependencies.\n2. Run the app."
-
-prompt = create_prompt(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial)
-print(prompt)

logx/prompts.py

@@ -0,0 +1,340 @@
+from typing import List, Dict, Optional
+from types import (
+    Code,
+    Prompt,
+    AppType,
+    File,
+    Space,
+    Tutorial,
+    App,
+    WebApp,
+    GradioApp,
+    StreamlitApp,
+    ReactApp,
+    Code,
+)
+
+def createLlamaPrompt(
+    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,
+) -> Prompt:
+    """
+    Generates a prompt for a Llama model to create a web application.
+    """
+    prompt = f"""
+    I need you to help me create a {app_type} web application.
+    The application name is: {app_name}
+    The application description is: {app_description}
+    The application features are: {app_features}
+    The application dependencies are: {app_dependencies}
+    The application space is: {app_space}
+    The application tutorial is: {app_tutorial}
+    Please generate the code for the application.
+    """
+    return Prompt(prompt=prompt)
+
+def createSpace(
+    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,
+) -> Space:
+    """
+    Generates a space for a web application.
+    """
+    space = f"""
+    {app_name}
+    {app_description}
+    {app_features}
+    {app_dependencies}
+    {app_space}
+    {app_tutorial}
+    """
+    return Space(space=space)
+
+def isPythonOrGradioAppPrompt(
+    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,
+) -> Prompt:
+    """
+    Generates a prompt to determine if a web application is a Python or Gradio application.
+    """
+    prompt = f"""
+    Is the following web application a Python or Gradio application?
+    {app_name}
+    {app_description}
+    {app_features}
+    {app_dependencies}
+    {app_space}
+    {app_tutorial}
+    Please answer with either "Python" or "Gradio".
+    """
+    return Prompt(prompt=prompt)
+
+def isReactAppPrompt(
+    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,
+) -> Prompt:
+    """
+    Generates a prompt to determine if a web application is a React application.
+    """
+    prompt = f"""
+    Is the following web application a React application?
+    {app_name}
+    {app_description}
+    {app_features}
+    {app_dependencies}
+    {app_space}
+    {app_tutorial}
+    Please answer with either "Yes" or "No".
+    """
+    return Prompt(prompt=prompt)
+
+def isStreamlitAppPrompt(
+    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,
+) -> Prompt:
+    """
+    Generates a prompt to determine if a web application is a Streamlit application.
+    """
+    prompt = f"""
+    Is the following web application a Streamlit application?
+    {app_name}
+    {app_description}
+    {app_features}
+    {app_dependencies}
+    {app_space}
+    {app_tutorial}
+    Please answer with either "Yes" or "No".
+    """
+    return Prompt(prompt=prompt)
+
+def getWebApp(
+    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,
+) -> WebApp:
+    """
+    Generates code for a web application.
+    """
+    code = f"""
+    <!DOCTYPE html>
+    <html lang="en">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>{app_name}</title>
+    </head>
+    <body>
+        <h1>{app_name}</h1>
+        <p>{app_description}</p>
+    </body>
+    </html>
+    """
+    return WebApp(code=code, language="html")
+
+def getGradioApp(
+    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,
+) -> GradioApp:
+    """
+    Generates code for a Gradio application.
+    """
+    code = f"""
+    import gradio as gr
+
+    def greet(name):
+        return f"Hello, {name}!"
+
+    demo = gr.Interface(greet, "text", "text")
+
+    if __name__ == "__main__":
+        demo.launch()
+    """
+    return GradioApp(code=code, language="python")
+
+def getReactApp(
+    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,
+) -> ReactApp:
+    """
+    Generates code for a React application.
+    """
+    code = f"""
+    import React from 'react';
+
+    function App() {{
+        return (
+            <div className="App">
+                <h1>{app_name}</h1>
+                <p>{app_description}</p>
+            </div>
+        );
+    }}
+
+    export default App;
+    """
+    return ReactApp(code=code, language="javascript")
+
+def getStreamlitApp(
+    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,
+) -> StreamlitApp:
+    """
+    Generates code for a Streamlit application.
+    """
+    code = f"""
+    import streamlit as st
+
+    st.title('{app_name}')
+    st.write('{app_description}')
+    """
+    return StreamlitApp(code=code, language="python")
+
+def parseTutorial(
+    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,
+) -> Tutorial:
+    """
+    Parses a tutorial for a web application.
+    """
+    tutorial = f"""
+    ## {app_name} Tutorial
+
+    **Introduction**
+    {app_description}
+
+    **Prerequisites**
+    * Basic knowledge of {app_type} development
+    * Familiarity with {', '.join(app_dependencies)}
+
+    **Steps**
+    1. {app_features[0]}
+    2. {app_features[1]}
+    3. {app_features[2]}
+
+    **Conclusion**
+    Congratulations! You have successfully created a {app_name} application.
+    """
+    return Tutorial(tutorial=tutorial)
+
+def generateFiles(
+    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,
+) -> List[File]:
+    """
+    Generates files for a web application.
+    """
+    files = []
+    if app_type == AppType.WEB_APP:
+        files.append(File(name="index.html", content=getWebApp(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial).code, language="html"))
+    elif app_type == AppType.GRADIO_APP:
+        files.append(File(name="app.py", content=getGradioApp(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial).code, language="python"))
+    elif app_type == AppType.STREAMLIT_APP:
+        files.append(File(name="app.py", content=getStreamlitApp(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial).code, language="python"))
+    elif app_type == AppType.REACT_APP:
+        files.append(File(name="App.js", content=getReactApp(app_type, app_name, app_description, app_features, app_dependencies, app_space, app_tutorial).code, language="javascript"))
+    return files
+
+4. types.py
+
+from enum import Enum
+from typing import List, Dict, Optional
+
+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(Enum):
+    WEB_APP = "web app"
+    GRADIO_APP = "gradio app"
+    STREAMLIT_APP = "streamlit app"
+    REACT_APP = "react app"
+
+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 App:
+    def __init__(self, code: str, language: str):
+        self.code = code
+        self.language = language
+
+class WebApp(App):
+    pass
+
+class GradioApp(App):
+    pass
+
+class StreamlitApp(App):
+    pass
+
+class ReactApp(App):
+    pass