Update app.py

app.py

@@ -1,65 +1,53 @@
+import streamlit as st
 import pandas as pd
+import plotly.express as px
 import openai
-import streamlit as st
-import matplotlib.pyplot as plt
-
-# Analyze using OpenAI
-def get_openai_insights(api_key, prompt):
-    openai.api_key = api_key
-    response = openai.Completion.create(
-        engine="text-davinci-003",
-        prompt=prompt,
-        max_tokens=500,
-        temperature=0.5
-    )
-    return response["choices"][0]["text"].strip()
-
-# Streamlit app
-def main():
-    st.title("Excel Data Visualization with OpenAI Insights")
-
-    # Input OpenAI API Key
-    api_key = st.text_input("Enter your OpenAI API Key", type="password")
-    if not api_key:
-        st.warning("Please enter your OpenAI API key to proceed.")
-        return
-
-    # File upload
-    excel_file = st.file_uploader("Upload the Excel File", type=["xls", "xlsx"])
-
-    if excel_file:
-        # Load Excel data
-        excel_data = pd.ExcelFile(excel_file)
-        st.sidebar.header("Select a Sheet to Visualize")
-        sheet_name = st.sidebar.selectbox("Sheet Name", excel_data.sheet_names)
-
-        if sheet_name:
-            data = pd.read_excel(excel_data, sheet_name=sheet_name)
-            st.subheader(f"Data from Sheet: {sheet_name}")
-            st.dataframe(data)
-
-            # Option to generate insights using OpenAI
-            st.header("Generate AI Insights")
-            if st.button("Get Insights from OpenAI"):
-                with st.spinner("Generating insights..."):
-                    try:
-                        data_sample = data.head(5).to_csv(index=False)
-                        prompt = f"Analyze the following data and provide key insights:\n\n{data_sample}"
-                        insights = get_openai_insights(api_key, prompt)
-                        st.success("AI Insights Generated!")
-                        st.text_area("AI Insights:", insights, height=200)
-                    except openai.error.OpenAIError as e:
-                        st.error(f"Error with OpenAI API: {e}")
-
-            # Visualize numeric data
-            st.header("Visualize Data")
-            numeric_cols = data.select_dtypes(include="number").columns
-            if numeric_cols.any():
-                col_to_plot = st.selectbox("Select a Column to Plot", numeric_cols)
-                if col_to_plot:
-                    fig, ax = plt.subplots()
-                    data[col_to_plot].plot(kind="bar", ax=ax, title=f"{col_to_plot} Analysis")
-                    st.pyplot(fig)
 
-if __name__ == "__main__":
-    main()
+# Setup OpenAI API
+openai.api_key = st.text_input("Enter your OpenAI API key", type="password")
+
+# App Title
+st.title("Bi-Weekly Report Visualizer")
+
+# Upload Excel File
+uploaded_file = st.file_uploader("Upload your Excel file", type=["xls", "xlsx"])
+
+if uploaded_file is not None:
+    # Load data
+    df = pd.read_excel(uploaded_file)
+
+    # Display raw data
+    st.write("### Raw Data")
+    st.dataframe(df)
+
+    # Extract relevant columns (assume "Week", "Category", and "Hours" columns exist in the sheet)
+    if {"Week", "Category", "Hours"}.issubset(df.columns):
+        # Summarize data for visualization
+        summary = df.groupby(["Week", "Category"]).sum().reset_index()
+
+        # Visualize data as a pie chart for each week
+        st.write("### Weekly Time Allocation")
+        weeks = summary["Week"].unique()
+        for week in weeks:
+            week_data = summary[summary["Week"] == week]
+            fig = px.pie(week_data, values="Hours", names="Category", title=f"Week: {week}")
+            st.plotly_chart(fig)
+
+        # Generate descriptive insights using OpenAI
+        if st.button("Generate Insights"):
+            # Get summary in text form
+            insights_data = summary.to_string(index=False)
+            # OpenAI prompt
+            prompt = f"Provide insights about time allocation trends in the following data:\n\n{insights_data}"
+            try:
+                response = openai.Completion.create(
+                    engine="text-davinci-003",
+                    prompt=prompt,
+                    max_tokens=150
+                )
+                st.write("### Insights")
+                st.text(response.choices[0].text.strip())
+            except Exception as e:
+                st.error(f"Failed to generate insights: {e}")
+    else:
+        st.error("The Excel file doesn't have the required columns: 'Week', 'Category', and 'Hours'.")