File size: 2,455 Bytes
b51c72b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
import pandas as pd
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()
|