Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions main.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
import streamlit as st
from langchain.chains import LLMChain
import replicate
from langchain_community.llms import Replicate
from langchain_core.prompts import PromptTemplate
from langchain.output_parsers import CommaSeparatedListOutputParser
from langchain_core.output_parsers import CommaSeparatedListOutputParser
import os
from streamlit_searchbox import st_searchbox
from typing import List
Expand Down Expand Up @@ -193,7 +192,8 @@ def create_route(city, landmarks):
input_variables=["landmarks", "city", "end"]
)
chain = prompt | llm
landmarks_string = "\n".join([f"{row['Name']}" for index, row in landmarks.iloc[:5,:].iterrows()])
landmarks_string = "
".join([f"{row['Name']}" for index, row in landmarks.iloc[:5,:].iterrows()])
print(landmarks_string)
part_one = chain.invoke({'city': city, 'landmarks':landmarks_string, 'end': landmarks.iloc[5,:]['Name']})
if len(landmarks)<5:
Expand All @@ -212,7 +212,8 @@ def create_route(city, landmarks):
input_variables=["landmarks", "city", "previous"]
)
chain = prompt | llm
landmarks_string = "\n".join([f"{row['Name']}" for index, row in landmarks.iloc[5:,:].iterrows()])
landmarks_string = "
".join([f"{row['Name']}" for index, row in landmarks.iloc[5:,:].iterrows()])
part_two = chain.invoke({'city': city, 'landmarks':landmarks_string, 'previous': landmarks.iloc[4,:]['Name']})
return part_one + " " + part_two

Expand Down Expand Up @@ -267,4 +268,3 @@ def generate_route(city, stops):
file_name='BikeRouter-route.html',
mime='text/html'
)

10 changes: 5 additions & 5 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -24,10 +24,10 @@ jsonpatch==1.33
jsonpointer==2.4
jsonschema==4.22.0
jsonschema-specifications==2023.12.1
langchain==0.2.0
langchain-community==0.2.0
langchain-core==0.2.0
langchain-text-splitters==0.2.0
langchain==1.2.13
langchain-community==0.4.1
langchain-core==1.2.22
langchain-text-splitters==1.1.1
langsmith==0.1.59
Markdown==3.6
markdown-it-py==3.0.0
Expand All @@ -54,7 +54,7 @@ pytz==2024.1
PyYAML==6.0.1
referencing==0.35.1
replicate==0.26.0
requests==2.31.0
requests==2.33.0
rich==13.7.1
rpds-py==0.18.1
setuptools==69.5.1
Expand Down