Update app.py

app.py

@@ -37,10 +37,8 @@ else:
         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}
-                ],
+                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
             )
@@ -48,7 +46,18 @@ else:
         except Exception as e:
             return f"Error: {str(e)}"
 
-    # Adding GPT-3.5 functionality to your app
+    # 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")
 
@@ -64,9 +73,15 @@ else:
         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)
+            # 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!")