Update app.py

app.py

@@ -99,11 +99,15 @@ def get_images_in_directory(directory):
                 image_files.append(os.path.join(root, file))
     return image_files
 
-def interact_with_agent(file_input, additional_notes):
+def interact_with_agent(file_input, file_input_2, additional_notes):
     shutil.rmtree("./figures")
     os.makedirs("./figures")
-
+    file_1 = pd.read_csv(file_input)
+    file_2 = pd.read_csv(file_input_2)
+    print (file_1.head())
+    print (file_2.head())
     data_file = pd.read_csv(file_input)
+    
     data_structure_notes = f"""- Description (output of .describe()):
     {data_file.describe()}
     - Columns with dtypes:
@@ -146,10 +150,11 @@ with gr.Blocks(
         secondary_hue=gr.themes.colors.blue,
     )
 ) as demo:
-    gr.Markdown("""# Agentville Autonomous Data Exploration 📈🧠
+    gr.Markdown("""# Agentville Autonomous Data Exploration 📈🧠 (Research preview)
 
 Drop a `.csv` file below, add notes to describe this data if needed, and **Agents powered by Gemini and Llama-3.1-70B will analyze the file content and does the analysis for you!**""")
     file_input = gr.File(label="Your file to analyze")
+    file_input_2 = gr.File(label="Your file to analyze")
     text_input = gr.Textbox(
         label="Additional notes to guide the analysis"
     )
@@ -164,11 +169,11 @@ Drop a `.csv` file below, add notes to describe this data if needed, and **Agent
     )
     gr.Examples(
         examples=[["./example/churn.csv", example_notes]],
-        inputs=[file_input, text_input],
+        inputs=[file_input, file_input_2, text_input],
         cache_examples=False
     )
 
-    submit.click(interact_with_agent, [file_input, text_input], [chatbot])
+    submit.click(interact_with_agent, [file_input, file_input_2,text_input], [chatbot])
 
 if __name__ == "__main__":
     demo.launch()