Owen Wang commited on
Commit
86f6d54
·
1 Parent(s): 471307c

fix description and make more similar to original

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -5,13 +5,19 @@ import openai
5
  import pinecone
6
  import json
7
  import re
 
8
 
9
  PINECONE_API_KEY = st.secrets["PINECONE_API_KEY"]
 
10
  OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
 
 
 
 
11
 
12
  class Metadata(TypedDict):
13
  title: str
14
- description
15
  slides: str
16
 
17
  # Initialize Pinecone and OpenAI
@@ -59,9 +65,9 @@ def fetch_lesson(query: str):
59
  # OpenAI prompt generation function
60
  def query_openai(prompt) -> str:
61
  response = openai.Completion.create(
62
- engine="text-davinci-003",
63
  prompt=prompt,
64
- max_tokens=1024,
65
  n=1,
66
  stop=None,
67
  temperature=0.7,
@@ -95,8 +101,9 @@ def format_metadata(metadata) -> List[str]:
95
  return f"Title: {metadata['title']}\n\nDescription: {metadata['description']}"
96
 
97
  # Streamlit UI
 
98
  st.title("Personalized Learning Curriculum Generator")
99
- user_input = st.text_input("Enter what you want to learn:")
100
  submit_button = st.button("Generate response")
101
  status = st.empty()
102
 
 
5
  import pinecone
6
  import json
7
  import re
8
+ import os
9
 
10
  PINECONE_API_KEY = st.secrets["PINECONE_API_KEY"]
11
+ # Set OpenAI API key from Streamlit Secrets
12
  OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
13
+ # Set maximum token length
14
+ MAX_TOKENS = 1024
15
+ # Set OpenAI model
16
+ MODEL = "text-davinci-003"
17
 
18
  class Metadata(TypedDict):
19
  title: str
20
+ description: str
21
  slides: str
22
 
23
  # Initialize Pinecone and OpenAI
 
65
  # OpenAI prompt generation function
66
  def query_openai(prompt) -> str:
67
  response = openai.Completion.create(
68
+ engine=MODEL,
69
  prompt=prompt,
70
+ max_tokens=MAX_TOKENS,
71
  n=1,
72
  stop=None,
73
  temperature=0.7,
 
101
  return f"Title: {metadata['title']}\n\nDescription: {metadata['description']}"
102
 
103
  # Streamlit UI
104
+ st.set_page_config(layout="centered")
105
  st.title("Personalized Learning Curriculum Generator")
106
+ user_input = st.text_area("Enter what you want to learn:", height=200)
107
  submit_button = st.button("Generate response")
108
  status = st.empty()
109