Update app.py

app.py

@@ -1,125 +1,61 @@
-import openai
 import streamlit as st
 import pandas as pd
 
-# 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 (chat version)
-    def generate_creative_content(prompt):
-        try:
-            response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo",  # Ensure using the chat model
-                messages=[{"role": "system", "content": "You are a helpful assistant."},
-                          {"role": "user", "content": prompt}],
-                max_tokens=150,
-                temperature=0.7  # You can adjust temperature for more creative results
-            )
-            return response['choices'][0]['message']['content'].strip()
-        except Exception as e:
-            return f"Error: {str(e)}"
-
-    # Adding this function to find buses between two locations
-    def get_buses_between_locations(start_location, destination_location):
-        # Filter buses that pass through both the start and destination locations
-        buses_start = df[df['Routes'].str.contains(start_location, case=False, na=False)]
-        buses_destination = df[df['Routes'].str.contains(destination_location, case=False, na=False)]
-
-        # Find buses that are common in both filtered dataframes
-        common_buses = pd.merge(buses_start, buses_destination, on='Dhaka Local Buses')
-
-        return common_buses
-
-    # Main Streamlit function
-    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:
-            # Get buses that pass through both the start and destination locations
-            buses_with_common_routes = get_buses_between_locations(start_location, destination_location)
-
-            if not buses_with_common_routes.empty:
-                st.write(f"Suggested buses from **{start_location}** to **{destination_location}**:")
-                for idx, row in buses_with_common_routes.iterrows():
-                    st.write(f"- **{row['Dhaka Local Buses']}**: {clean_route(row['Routes'])}")
-            else:
-                st.write(f"No buses found passing through both **{start_location}** and **{destination_location}**.")
-
-        # 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)
+# 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."
+
+# Streamlit app
+def main():
+    st.title("Dhaka Local Buses and Routes Finder")
+    
+    # Bus Name Search Section
+    st.header("Search Bus Name")
+    
+    # Add a placeholder as the default option for the selectbox
+    bus_name = st.selectbox(
+        'Select a Bus Name from dropdown', 
+        ['Select Bus Name from dropdown'] + df['Dhaka Local Buses'].tolist()  # Adding the placeholder at the beginning
+    )
+    
+    if bus_name != 'Select Bus Name from dropdown':  # Proceed only if a valid bus is selected
+        route = get_route_by_bus(bus_name)
+        st.write(f"Routes for **{bus_name}**:")
+        st.write(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
         else:
-            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}**.")
+            st.write(f"No buses found for the location '**{location}**'.")
 
-    if __name__ == "__main__":
-        main()
+if __name__ == "__main__":
+    main()