app.py

import os
import google.generativeai as genai
import gradio as gr
import smtplib
google_api_key = os.getenv("MY_GOOGLE_AI_STUDIO_KEY")  

# Write code to get gemini models
genai.configure(api_key = google_api_key)
model = genai.GenerativeModel('gemini-2.0-flash-exp')
model_getFunctions = genai.GenerativeModel('gemini-2.0-flash-exp')


from  UtilityFunctions import get_files_in_folder, get_file_types, organize_files, zip_folder, send_reminder_email, add_calendar_invite, send_stock_email, schedule_daily_stock_email
#Get function description
functions = 'function - get_files_in_folder - ' +  get_files_in_folder.__doc__ + '''
function - get_file_types - ''' +  get_file_types.__doc__ + '''
function - organize_files - ''' +  organize_files.__doc__ + '''
function - zip_folder - ''' +  zip_folder.__doc__ + '''
function - send_reminder_email - ''' +  send_reminder_email.__doc__ + ''' 
function - add_calendar_invite - ''' +  add_calendar_invite.__doc__ + '''
function - send_stock_email - ''' +  send_stock_email.__doc__ + '''
function - schedule_daily_stock_email - ''' +  schedule_daily_stock_email.__doc__ 


def process_input(option):
    if not option:
        return "Please select an option", "from the dropbox"
    
    prompt = f'''You are a AI assistant. Your job is to break a task into smaller 
    tasks that could be executed as code. Do not write the code, but just list the tasks as steps.
    Here is your task - {option}'''

    response = model.generate_content(prompt)    
    output1 = response.text

    prompt2 = f'''You are a AI assistant. You are given some steps. 
    Your job is to get the function calls for these steps from the list of functions below. 
    Give me an executable python code that uses these functions. Just make the function calls. 
    Do not write any code within these functions. Do not strictly write any other code apart from the function calls. 
    However you are allowed to save the output of these functions into variables and use them as input parameters to other functions. 
    You shall only use the functions mentioned here and you shall not write your own code. 
    Your code should strictly not use any other functions apart from functions mentioned below. 
    Here are your steps - {response.text} ; and
    here are your functions - {functions}'''

    response2 = model_getFunctions.generate_content(prompt2)

    
    output2 = response2.text
    return output1, output2

def handle_execute(code):
    model_formatCode = genai.GenerativeModel('gemini-2.0-flash-exp')
    if not code:
        print("Empty String - nothing to be executed")
    else:
        # Split into lines, remove first and last, then join back
        lines = code.splitlines()
        cleaned_code = "\n".join(lines[1:-1])
        print(cleaned_code)
        exec(cleaned_code)
        
with gr.Blocks() as demo:
    gr.Markdown("# EPAI Capstone - Agents")
    gr.Markdown('''This capstone demonstrate how agents work internally. This project accepts a user input, 
    breaks them into smaller tasks. Chooses from a predefined set of functions(tools) that correspond to these smaller tasks, 
    and,  finally executes all the functions. \n\n Here are the set of available functions(tools):\n
    def get_files_in_folder(root_dir) \n
    def get_file_types(file_list) \n
    def organize_files(type_dict) ''')
    
    with gr.Column():
        dropdown = gr.Dropdown(choices=['''scans a folder called MyDownlads, identify file types and 
move them into categorized folders (e.g., PDFs, images, code files, etc)''', 
'''scans a folder called MyDownlads,  organizes files into categorized folders
 (e.g., PDFs, images, code files, etc) and create a zip archive of each folder''',
 '''Remind Ramya to submit TSAI assignment via email - sriramya.toleti@gmail.com''',
 '''Send a calender invite to sriramya.toleti@gmail.com tomorrow at 7am to join my yoga class''',
 '''Share the stock price for NVIDIA every day at 5 PM via email with sriramya.toleti@gmail.com''',
 '''Write your own task
 '''], label="Select an option", interactive=True)
        text_input = gr.Textbox(label="Enter you task here", interactive=True, visible=False)        
        input_button = gr.Button("Submit")
    output1 = gr.Textbox(label="Your task broken into smaller text")
    output2 = gr.Textbox(label="Functions corresponding to your task from predefined toolbox")
    execute_button = gr.Button("Execute", visible=False)

    def toggle_textbox(option):
        return gr.update(visible=(option.strip() == "Write your own task"))

    dropdown.change(toggle_textbox, inputs=[dropdown], outputs=[text_input])

    def handle_submit(option, text):
        # If user selected "Write your own task", use the text input instead
        task_input = text.strip() if option.strip() == "Write your own task" else option        
        if not task_input:  # Ensure task_input is not empty
            return "Please enter a task.", "", gr.update(visible=False)
        
        out1, out2 = process_input(task_input)  # Pass the correct task
        return out1, out2, gr.update(visible=True)


    input_button.click(handle_submit, inputs=[dropdown, text_input], 
                       outputs=[output1, output2, execute_button])
    
    execute_button.click(handle_execute, inputs=[output2])


demo.launch()