mirror of
https://github.com/openai/openai-cookbook
synced 2024-11-08 01:10:29 +00:00
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
|
import streamlit as st
|
||
|
|
import openai
|
||
|
|
|
||
|
|
from database import get_redis_connection, get_redis_results
|
||
|
|
from config import INDEX_NAME, COMPLETIONS_MODEL
|
||
|
|
|
||
|
|
# initialise Redis connection
|
||
|
|
|
||
|
|
client = get_redis_connection()
|
||
|
|
|
||
|
|
### SEARCH APP
|
||
|
|
|
||
|
|
st.set_page_config(
|
||
|
|
page_title="Streamlit Search - Demo",
|
||
|
|
page_icon=":robot:"
|
||
|
|
)
|
||
|
|
|
||
|
|
st.title('Formula 1 Search')
|
||
|
|
st.subheader("Search for any Formula 1 rule questions you have")
|
||
|
|
|
||
|
|
prompt = st.text_input("Enter your search here","", key="input")
|
||
|
|
|
||
|
|
if st.button('Submit', key='generationSubmit'):
|
||
|
|
result_df = get_redis_results(client,prompt,INDEX_NAME)
|
||
|
|
|
||
|
|
# Build a prompt to provide the original query, the result and ask to summarise for the user
|
||
|
|
summary_prompt = '''Summarise this result in a bulleted list to answer the search query a customer has sent.
|
||
|
|
Search query: SEARCH_QUERY_HERE
|
||
|
|
Search result: SEARCH_RESULT_HERE
|
||
|
|
Summary:
|
||
|
|
'''
|
||
|
|
summary_prepped = summary_prompt.replace('SEARCH_QUERY_HERE',prompt).replace('SEARCH_RESULT_HERE',result_df['result'][0])
|
||
|
|
summary = openai.Completion.create(engine=COMPLETIONS_MODEL,prompt=summary_prepped,max_tokens=500)
|
||
|
|
|
||
|
|
# Response provided by GPT-3
|
||
|
|
st.write(summary['choices'][0]['text'])
|
||
|
|
|
||
|
|
# Option to display raw table instead of summary from GPT-3
|
||
|
|
#st.table(result_df)
|