feat: update example and little code cleaning

Author: bregydoc

examples/pi.rs

@@ -1,7 +1,7 @@
 use auto_rust::auto_implement;
 
 #[auto_implement]
-/// This function calculates pi with a robust and very fast algorithm.
+/// This function calculates pi with a reliable algorithm.
 fn calculate_pi_with_n_decimals(n: u32) -> f64 {
     todo!()
 }

src/generator.rs

@@ -1,7 +1,5 @@
 use std::error::Error;
 
-use proc_macro::TokenStream;
-
 use crate::api::open_ai_chat_completions;
 
 pub fn generate_body_function_from_head(head: String) -> Result<String, Box<dyn Error>> {
@@ -33,7 +31,7 @@ pub fn generate_body_function_from_head(head: String) -> Result<String, Box<dyn
     //TODO: improve the prompt to eliminate the need for this
     let body_str = body_str.trim_matches('`').to_string();
 
-    let implementation = format!("{} {{{}}}", head, body_str);
+    let implementation = format!("{} {{\n{}\n}}", head, body_str);
 
     Ok(implementation)
 }

src/lib.rs

@@ -1,13 +1,10 @@
-// extern crate proc_macro;
-
 mod api;
 mod data;
 mod generator;
 
 use dotenv::dotenv;
 use proc_macro::TokenStream;
-use syn::{parse, ItemFn, Lit, Meta, MetaNameValue, __private::ToTokens};
-use syn::{Attribute, LitStr};
+use syn::{ItemFn, __private::ToTokens};
 
 use crate::generator::generate_body_function_from_head;
 
@@ -25,9 +22,7 @@ pub fn implement(_item: TokenStream) -> TokenStream {
 
 #[proc_macro_attribute]
 pub fn auto_implement(args: TokenStream, input: TokenStream) -> TokenStream {
-    println!("{:?}", input);
-
-    let ast: ItemFn = syn::parse(input.clone()).expect("Failed to parse input as a function");
+    let ast: ItemFn = syn::parse(input).expect("Failed to parse input as a function");
 
     // Search for the information within the attributes.
 
@@ -39,20 +34,9 @@ pub fn auto_implement(args: TokenStream, input: TokenStream) -> TokenStream {
 
     for attr in ast.attrs {
         let data = attr.to_token_stream().to_string();
-        // if attr.path().is_ident("doc") {
-        // if let Ok(Meta::NameValue(meta_name_value)) = attr.parse_args() {
-        //     let info = meta_name_value.value.to_token_stream().to_string();
-        //     // if info.contains("This function calculates") {
-
-        //     target_info = info;
-        //     break;
-        //     // }
-        // }
-        println!("{}", data);
 
         prompt_input.push_str(&data);
         prompt_input.push('\n');
-        // }
     }
 
     prompt_input.push_str(&fn_header);