Update app.py

app.py

@@ -73,6 +73,7 @@ if clear_button:
     counter_placeholder.write(f"Next item ...")
 
 # Insert a file uploader that accepts multiple files at a time
+col1, col2 = st.columns([3, 1])
 if uploaded_file is not None:
     # Success message
     st.sidebar.success("File uploaded successfully.")
@@ -80,41 +81,45 @@ if uploaded_file is not None:
     # Can be used wherever a "file-like" object is accepted:
     df = pd.read_csv(uploaded_file)
 
-    # Generate the HTML using Pygwalker
-    pyg_html = pyg.walk(df, return_html=True)
-
-    # Embed the HTML into the Streamlit app
-    components.html(pyg_html, height=1000, scrolling=True)
-
-    # Instantiate a LLM
-    OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
-    llm = OpenAI(api_token=OPENAI_API_KEY)
-    pandas_ai = PandasAI(llm)
-
-    # Container for chat history
-    response_container = st.container()
-    container = st.container()
-    with container:
-        with st.form(key="my_form", clear_on_submit=True):
-            user_input = st.text_area(
-                "Enter your question here:", key="input", height=100
-            )
-            submit_button = st.form_submit_button(label="Send")
-
-        if submit_button:
-            output = pandas_ai(df, prompt=user_input)
-            st.session_state["past"].append(user_input)
-            st.session_state["generated"].append(
-                {"type": "normal", "data": f"{output}"}
-            )
-
-    if st.session_state["generated"]:
-        with response_container:
-            for i in range(len(st.session_state["generated"])):
-                message(st.session_state["past"][i], is_user=True, key=str(i) + "_user")
-                answer = st.session_state["generated"][i]["data"]
-                message(answer)
-                counter_placeholder.write(f"All rights reserved @ Yiqiao Yin")
+    # Col 1:
+    with col1:
+        # Generate the HTML using Pygwalker
+        pyg_html = pyg.walk(df, return_html=True)
+    
+        # Embed the HTML into the Streamlit app
+        components.html(pyg_html, height=1000, scrolling=True)
+
+    # Col 2:
+    with col2:
+        # Instantiate a LLM
+        OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
+        llm = OpenAI(api_token=OPENAI_API_KEY)
+        pandas_ai = PandasAI(llm)
+    
+        # Container for chat history
+        response_container = st.container()
+        container = st.container()
+        with container:
+            with st.form(key="my_form", clear_on_submit=True):
+                user_input = st.text_area(
+                    "Enter your question here:", key="input", height=100
+                )
+                submit_button = st.form_submit_button(label="Send")
+    
+            if submit_button:
+                output = pandas_ai(df, prompt=user_input)
+                st.session_state["past"].append(user_input)
+                st.session_state["generated"].append(
+                    {"type": "normal", "data": f"{output}"}
+                )
+    
+        if st.session_state["generated"]:
+            with response_container:
+                for i in range(len(st.session_state["generated"])):
+                    message(st.session_state["past"][i], is_user=True, key=str(i) + "_user")
+                    answer = st.session_state["generated"][i]["data"]
+                    message(answer)
+                    counter_placeholder.write(f"All rights reserved @ Yiqiao Yin")
 
 else:
     st.warning("Please upload a csv file.")