Update app.py

app.py

@@ -1,41 +1,49 @@
 import streamlit as st
-import pandas as pd
-import matplotlib.pyplot as plt
-import seaborn as sns
-import numpy as np
-
-
-
-
-# App Title
-st.title("📊 Streamlit Data Analysis Tool")
-
-# Upload CSV File
-uploaded_file = st.file_uploader("Upload a CSV file", type=["csv"])
-
-if uploaded_file:
-    df = pd.read_csv(uploaded_file)
-    st.write("### Preview of Uploaded Data")
-    st.dataframe(df.head())
-
-    # Dropdown for column selection
-    numeric_cols = df.select_dtypes(include=['number']).columns.tolist()
-    if numeric_cols:
-        col_to_analyze = st.selectbox("Select a column to analyze", numeric_cols)
-
-        # Summary statistics
-        st.write("### Summary Statistics")
-        st.write(df[col_to_analyze].describe())
-
-        # Histogram Slider
-        bins = st.slider("Number of bins for histogram", min_value=5, max_value=50, value=20)
-
-        # Plot Histogram
-        fig, ax = plt.subplots()
-        sns.histplot(df[col_to_analyze], bins=bins, kde=True, ax=ax)
-        st.pyplot(fig)
+import google.generativeai as genai
+
+# Set up Gemini API
+genai.configure(api_key="AIzaSyBQqw4RV3XdTFuPud7mkoHNPfxQm5EnOrc")  # Replace with your Gemini API key
+
+# Set up the model
+generation_config = {
+    "temperature": 0.9,  # Controls creativity (0 = strict, 1 = creative)
+    "top_p": 1,
+    "top_k": 1,
+    "max_output_tokens": 2048,  # Maximum length of the response
+}
+
+safety_settings = [
+    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+]
+
+model = genai.GenerativeModel(
+    model_name="gemini-pro",
+    generation_config=generation_config,
+    safety_settings=safety_settings,
+)
+
+# Streamlit App
+st.title("📖 Gemini Story Generator")
+st.write("Enter a prompt, and Gemini AI will generate a story for you!")
+
+# User input
+user_prompt = st.text_area("Enter your story prompt:", "Once upon a time...")
+
+# Generate story button
+if st.button("Generate Story"):
+    if user_prompt.strip() == "":
+        st.warning("Please enter a prompt!")
     else:
-        st.warning("No numeric columns found in the dataset.")
-
-else:
-    st.info("Please upload a CSV file to get started.")
+        try:
+            # Generate response using Gemini AI
+            response = model.generate_content(
+                f"Write a creative and engaging story suitable for elementary school children based on this prompt: {user_prompt}"
+            )
+            st.write("### Here's Your Story:")
+            st.write(response.text)
+            st.success("🌟 The End 🌟")
+        except Exception as e:
+            st.error(f"An error occurred: {e}. Please try again.")