from typing import List, Dict, Optional
from cust_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"""
{app_name}
{app_name}
{app_description}
"""
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 (
{app_name}
{app_description}
);
}}
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