Modify AST nodes inplace and write to file

TechdocsAPI/app.py

@@ -0,0 +1 @@
+from backend import app

techdocs/utils/functools.py

@@ -41,3 +41,8 @@ def request_inference(config, code_block, max_retries=1):
         config.update({"access_token":get_access_token(data)})
         
         return request_inference(config,config["access_token"], code_block, max_retries=max_retries-1)
+    
+
+    
+
+    

techdocs/utils/parse.py

@@ -7,20 +7,18 @@ from utils.functools import *
 
 
 # Function to extract and print the outermost nested function with line number
-def extract_outermost_function(node,  config, line_number=1,):
-    base_url = "http://localhost:8000"
+def extract_outermost_function(node, config):
+    
     if isinstance(node, ast.FunctionDef):
         function_def = ast.unparse(node)
-        print(f"Function starting at line {line_number}:\n{function_def}")
-        print("=" * 30)
-
-        response = request_inference(config=config,code_block=function_def)
+        response = request_inference(config=config, code_block=function_def)
+        docstr = response['docstr']
+        docstring = ast.Expr(value=ast.Str(s=docstr))
+        node.body.insert(0, docstring)
         
 
     for child in ast.iter_child_nodes(node):
-        if isinstance(child, ast.FunctionDef):
-            line_number = child.lineno
-        extract_outermost_function(child, line_number)
+        extract_outermost_function(child, config)
 
 # Function to traverse directories recursively and extract functions from Python files
 def extract_functions_from_directory(config):
@@ -32,4 +30,7 @@ def extract_functions_from_directory(config):
                 with open(file_path, "r",errors='ignore') as file:
                     content = file.read()
                 parsed = ast.parse(content)
-                extract_outermost_function(parsed, config)
+                extract_outermost_function(parsed, config)
+                docstr_code = ast.unparse(parsed)
+                with open(file_path, "w",errors='ignore') as file:
+                    file.write(docstr_code)

testing/test.py

@@ -0,0 +1,21 @@
+def add(a, b):
+    return a + b
+
+
+
+def multiply(a, b):
+    return a * b
+
+def subtract(a, b):
+    return a - b
+
+def divide(a, b):
+    if b == 0:
+        raise ValueError("Cannot divide by zero")
+    return a / b
+
+
+def func(*args, **kwargs):
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+    return wrapper