Update app.py

app.py

@@ -1,61 +1,109 @@
+import openai
 import streamlit as st
 import pandas as pd
 
-# Load the Excel file into a DataFrame (adjust the file path accordingly)
-df = pd.read_excel('Book1.xlsx')
-
-# Clean up the DataFrame (strip any unnecessary whitespaces in column names)
-df.columns = df.columns.str.strip()
-
-# Function to clean routes and remove unwanted characters
-def clean_route(route):
-    # Remove the "_x000D_" carriage return issue and other unwanted characters
-    return route.replace("_x000D_", "").strip()
-
-# Function to search buses by location
-def search_by_location(location):
-    # Filter buses that have the location in their routes
-    buses_with_location = df[df['Routes'].str.contains(location, case=False, na=False)]
-    return buses_with_location
-
-# Function to get the route of a selected bus
-def get_route_by_bus(bus_name):
-    # Get the route for the selected bus
-    bus_route = df[df['Dhaka Local Buses'] == bus_name]['Routes'].values
-    if len(bus_route) > 0:
-        return clean_route(bus_route[0])  # Clean the route before returning
-    else:
-        return "No route found for this bus."
-
-def main():
-    st.title("Dhaka Local Buses and Routes Finder")
-    
-    # Bus Name Search Section
-    st.header("Search Bus Name")
-    
-    # Add a placeholder as the first item in the bus list
-    bus_name_options = ['Select Bus Name from dropdown'] + df['Dhaka Local Buses'].tolist()
-    
-    bus_name = st.selectbox('Select a Bus Name', bus_name_options)
-    
-    if bus_name != 'Select Bus Name from dropdown':
-        route = get_route_by_bus(bus_name)
-        st.write(f"Routes for **{bus_name}**:")
-        st.write(route)
-    else:
-        st.write("Please select a bus name from the dropdown to view its route.")
-    
-    # Location Search Section
-    st.header("Search Location")
-    location = st.text_input('Enter a location (e.g., Gabtoli, Mirpur)', '').strip()
-    if location:
-        buses = search_by_location(location)
-        if not buses.empty:
-            st.write(f"Buses passing through **{location}**:")
-            for idx, row in buses.iterrows():
-                st.write(f"- {row['Dhaka Local Buses']}: {clean_route(row['Routes'])}")  # Clean routes before displaying
+# Prompt user to enter OpenAI API key
+openai.api_key = st.text_input("Enter your OpenAI API Key", type="password")
+
+# Ensure the API key is entered before continuing
+if not openai.api_key:
+    st.warning("Please enter your OpenAI API key to use the app.")
+else:
+    # Load the Excel file into a DataFrame (adjust the file path accordingly)
+    df = pd.read_excel('Book1.xlsx')
+
+    # Clean up the DataFrame (strip any unnecessary whitespaces in column names)
+    df.columns = df.columns.str.strip()
+
+    # Function to clean routes and remove unwanted characters
+    def clean_route(route):
+        return route.replace("_x000D_", "").strip()
+
+    # Function to search buses by location
+    def search_by_location(location):
+        buses_with_location = df[df['Routes'].str.contains(location, case=False, na=False)]
+        return buses_with_location
+
+    # Function to get the route of a selected bus
+    def get_route_by_bus(bus_name):
+        bus_route = df[df['Dhaka Local Buses'] == bus_name]['Routes'].values
+        if len(bus_route) > 0:
+            return clean_route(bus_route[0])  # Clean the route before returning
+        else:
+            return "No route found for this bus."
+
+    # Function to generate creative content using GPT-3.5
+    def generate_creative_content(prompt):
+        try:
+            response = openai.Completion.create(
+                model="gpt-3.5-turbo",  # Or "gpt-3.5-turbo" for more creativity
+                prompt=prompt,
+                max_tokens=150,
+                n=1,
+                stop=None,
+                temperature=0.7  # You can adjust temperature for more creative results
+            )
+            return response.choices[0].text.strip()
+        except Exception as e:
+            return f"Error: {str(e)}"
+
+    # Adding GPT-3.5 functionality to your app
+    def main():
+        st.title("Dhaka Local Buses and Routes Finder with GPT Creativity")
+
+        # Fun Bus-Related Facts or Trivia Section
+        st.header("Ask GPT for Fun Bus Facts or Trivia!")
+        trivia_input = st.text_input("Ask a fun fact or trivia about Dhaka's local buses:")
+        if trivia_input:
+            trivia_response = generate_creative_content(f"Tell me an interesting fact or trivia about Dhaka's local buses.")
+            st.write(trivia_response)
+
+        # Route Optimization Suggestions Section
+        st.header("Route Optimization Suggestions")
+        start_location = st.text_input("Enter your starting location (e.g., Mirpur):")
+        destination_location = st.text_input("Enter your destination location (e.g., Dhaka University):")
+        if start_location and destination_location:
+            route_suggestion_prompt = f"Suggest the best bus route from {start_location} to {destination_location}, considering speed, convenience, and the most popular routes."
+            route_suggestion = generate_creative_content(route_suggestion_prompt)
+            st.write(route_suggestion)
+
+        # Bus-Themed Storytelling Section
+        st.header("Create a Bus-Themed Story!")
+        story_input = st.text_input("Enter a location (e.g., Mirpur to Gabtoli) to create a bus-themed story:")
+        if story_input:
+            story_prompt = f"Create a short story about a passenger taking a bus from {story_input}."
+            bus_story = generate_creative_content(story_prompt)
+            st.write(bus_story)
+
+        # General Questions about Local Buses
+        st.header("Ask General Questions About Dhaka's Local Buses")
+        question_input = st.text_input("Ask a question about the Dhaka bus system (e.g., most common issues faced by commuters):")
+        if question_input:
+            question_answer = generate_creative_content(f"Answer the question: {question_input} about Dhaka's local buses.")
+            st.write(question_answer)
+
+        # Existing bus search functionality (unchanged)
+        st.header("Search Bus Name")
+        bus_name_options = ['Select Bus Name from dropdown'] + df['Dhaka Local Buses'].tolist()
+        bus_name = st.selectbox('Select a Bus Name', bus_name_options)
+        if bus_name != 'Select Bus Name from dropdown':
+            route = get_route_by_bus(bus_name)
+            st.write(f"Routes for **{bus_name}**:")
+            st.write(route)
         else:
-            st.write(f"No buses found for the location '**{location}**'.")
+            st.write("Please select a bus name from the dropdown to view its route.")
+
+        # Location search functionality remains as before
+        st.header("Search Location")
+        location = st.text_input('Enter a location (e.g., Gabtoli, Mirpur)', '').strip()
+        if location:
+            buses = search_by_location(location)
+            if not buses.empty:
+                st.write(f"Buses passing through **{location}**:")
+                for idx, row in buses.iterrows():
+                    st.write(f"- {row['Dhaka Local Buses']}: {clean_route(row['Routes'])}")
+            else:
+                st.write(f"No buses found for the location '**{location}**.")
 
-if __name__ == "__main__":
-    main()
+    if __name__ == "__main__":
+        main()