[python] support shiny apps

Author: javierluraschi

CHANGELOG.md

@@ -1,6 +1,10 @@
 # Changelog
 
-## 2.8.12
+## 2.8.14
+
+- Support describing app.R files
+
+## 2.8.13
 
 - Fix edit link to stl, obj, gltf and step files 
 

apps/hal9/app.py

@@ -6,6 +6,7 @@
 from tools.hal9 import answer_hal9_questions_description, answer_hal9_questions
 from tools.text_agent import analyze_text_file_description, analyze_text_file
 from tools.streamlit import streamlit_generator, streamlit_generator_description
+from tools.shiny import shiny_generator, shiny_generator_description
 from tools.fastapi import fastapi_generator, fastapi_generator_description
 from tools.other_tools import final_response_description, final_response
 from tools.python_execution import python_execution_description ,python_execution
@@ -14,8 +15,8 @@
 messages = load_messages()
 
 # load tools
-tools_descriptions = [python_execution_description, final_response_description, solve_math_problem_description, answer_generic_question_description, analyze_csv_description, images_management_system_description, answer_hal9_questions_description, analyze_text_file_description, fastapi_generator_description, streamlit_generator_description]
-tools_functions = [python_execution, final_response, solve_math_problem, answer_generic_question, analyze_csv, images_management_system, answer_hal9_questions, analyze_text_file, fastapi_generator, streamlit_generator]
+tools_descriptions = [python_execution_description, final_response_description, solve_math_problem_description, answer_generic_question_description, analyze_csv_description, images_management_system_description, answer_hal9_questions_description, analyze_text_file_description, fastapi_generator_description, streamlit_generator_description, shiny_generator_description]
+tools_functions = [python_execution, final_response, solve_math_problem, answer_generic_question, analyze_csv, images_management_system, answer_hal9_questions, analyze_text_file, fastapi_generator, streamlit_generator, shiny_generator]
 
 if len(messages) < 1:
     messages = insert_message(messages, "system", "You are Hal9, a helpful and highly capable AI assistant. Your primary responsibility is to analyze user questions and select the most appropriate tool to provide precise, relevant, and actionable responses. Always prioritize using the right tool to ensure efficiency and clarity in your answers. If a tool is needed, follow these steps: 1. Identify the best tool for the task. 2. Execute the tool and process its response. 3. If the tool provides a valid result, return it to the user. 4. If the tool fails, do NOT retry with the same tool. Instead, explain the failure and suggest improvements in the prompt or alternative approaches.")

apps/hal9/tools/shiny.py

@@ -0,0 +1,62 @@
+from utils import generate_response, load_messages, insert_message, extract_code_block, save_messages
+import os
+import traceback
+from clients import openai_client
+
+def debug_code(python_code):
+    try:
+        exec(python_code)
+        return "About to show your app...", "", ""
+    except Exception as e:
+        tb = traceback.format_exc()
+        relevant_error_info = tb.splitlines()
+        last_line = relevant_error_info[-1]
+        complete_traceback="\n".join(relevant_error_info)
+        return "App fails to run", last_line, complete_traceback
+    
+def save_shiny_code(code):
+    directory = "./.storage/app.R"
+    # Ensure the target directory exists
+    if not os.path.exists(directory):
+        os.makedirs(directory)
+      
+    # Write the code to the app.R file
+    python_file_path = os.path.join(directory, "app.R")
+    with open(python_file_path, 'w') as file:
+        file.write(code)
+
+def shiny_generator(prompt):
+    # load messages
+    messages = load_messages(file_path="./.storage/.shiny_messages.json")
+
+    if len(messages) < 1:
+        messages = insert_message(messages, "system", f"""This is a R Shiny generator system that automates the creation of R shiny apps based on user prompts. It interprets natural language queries, and the response is an complete python script with the including imports for a interactive R Shiny app, return the code as fenced code block with triple backticks (```) as ```python```""")
+    messages = insert_message(messages, "user", f"Generates an app that fullfills this user request -> {prompt}")
+    model_response = generate_response("openai", "gpt-4-turbo", messages) 
+    response_content = model_response.choices[0].message.content
+    shiny_code = extract_code_block(response_content, "python")
+    
+    save_shiny_code(shiny_code)
+    messages = insert_message(messages, "assistant", shiny_code)
+    save_messages(messages, file_path="./.storage/.shiny_messages.json")
+    return "About to show your app..."
+    
+shiny_generator_description = {
+    "type": "function",
+    "function": {
+        "name": "shiny_generator",
+        "description": "Generates a complete R Shiny app based on user-provided natural language prompts. It automates the creation of interactive applications that requires interactions (This tool do not interact with files uploaded)",
+        "strict": True,
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "A detailed natural language description of the desired R Shiny app, including specific requirements or features to implement.",
+                }
+            },
+            "required": ["prompt"],
+            "additionalProperties": False,
+        },
+    }
+}

python/hal9/describe.py

@@ -1,28 +1,33 @@
 import os
 
+type_checks = {
+  'fastapi': {
+    'files': ['app.py', 'main.py'],
+    'contents': ['import FastAPI', 'import Flask']
+  },
+  'shiny': {
+    'files': ['app.R'],
+    'contents': ['library(shiny)']
+  }
+}
+
 def describe_content(target_path):
   result = {'type': ''}
-  
-  if os.path.isfile(target_path) and target_path.endswith('.py'):
-    python_files = [os.path.basename(target_path)]
-    folder = os.path.dirname(target_path) or '.'
+    
+  if os.path.isfile(target_path):
+      all_files = [target_path]
+  elif os.path.isdir(target_path):
+      all_files = os.listdir(target_path)
   else:
-    folder = target_path
-    python_files = [f for f in os.listdir(folder) if f.endswith('.py')]
-  
-  target_file = None
-  
-  if 'app.py' in python_files:
-    target_file = 'app.py'
-  elif 'main.py' in python_files:
-    target_file = 'main.py'
-  elif len(python_files) == 1:
-    target_file = python_files[0]
-  
-  if target_file:
-    with open(os.path.join(folder, target_file), 'r', encoding='utf-8') as file:
-      content = file.read()
-      if 'import FastAPI' in content or 'import Flask' in content:
-        result['type'] = 'fastapi'
+      return result
 
+  for type_name, criteria in type_checks.items():
+    for target_file in criteria['files']:
+      if target_file in all_files:
+        with open(os.path.join(target_path, target_file), 'r', encoding='utf-8') as file:
+          content = file.read()
+          if any(keyword in content for keyword in criteria['contents']):
+            result['type'] = type_name
+            break
+
   return result

python/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "hal9"
-version = "2.8.13"
+version = "2.8.14"
 description = ""
 authors = ["Javier Luraschi <javier@hal9.ai>"]
 readme = "README.md"