Spaces:
Build error
Build error
Keane Moraes
commited on
Commit
·
10d9795
1
Parent(s):
d3aa9b9
patch for pormpter folder saving
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@ import streamlit as st
|
|
| 2 |
from topics import TopicModelling
|
| 3 |
import mdforest
|
| 4 |
import utils
|
|
|
|
| 5 |
|
| 6 |
st.title("Drop the first document")
|
| 7 |
file1 = st.file_uploader("Upload a file", type=["md", "txt"], key="first")
|
|
@@ -14,6 +15,9 @@ results = {}
|
|
| 14 |
embedder = utils.load_model()
|
| 15 |
nlp = utils.load_nlp()
|
| 16 |
|
|
|
|
|
|
|
|
|
|
| 17 |
if file1 is not None and file2 is not None:
|
| 18 |
|
| 19 |
input_text1 = file1.read().decode("utf-8")
|
|
|
|
| 2 |
from topics import TopicModelling
|
| 3 |
import mdforest
|
| 4 |
import utils
|
| 5 |
+
import os
|
| 6 |
|
| 7 |
st.title("Drop the first document")
|
| 8 |
file1 = st.file_uploader("Upload a file", type=["md", "txt"], key="first")
|
|
|
|
| 15 |
embedder = utils.load_model()
|
| 16 |
nlp = utils.load_nlp()
|
| 17 |
|
| 18 |
+
if not os.path.exists("./prompter/"):
|
| 19 |
+
os.mkdir("./prompter/")
|
| 20 |
+
|
| 21 |
if file1 is not None and file2 is not None:
|
| 22 |
|
| 23 |
input_text1 = file1.read().decode("utf-8")
|
utils.py
CHANGED
|
@@ -133,8 +133,8 @@ def generate_insights(topics:dict, name1:str, name2:str, text1:str, text2:str, c
|
|
| 133 |
# print(cluster_id, " ", sentences)
|
| 134 |
final_prompt = PROMPT.replace("{{sentences}}", "\n".join(sentences))
|
| 135 |
|
| 136 |
-
with open(f"prompter/insights_{cluster_id}.prompt", "w") as f:
|
| 137 |
-
|
| 138 |
|
| 139 |
# Generate insights for each cluster
|
| 140 |
response = openai.Completion.create(
|
|
@@ -157,8 +157,8 @@ def generate_insights(topics:dict, name1:str, name2:str, text1:str, text2:str, c
|
|
| 157 |
# print(description)
|
| 158 |
# final_insights.append({"name": name_of_insight, "description": description})
|
| 159 |
|
| 160 |
-
with open(f"prompter/insights_{cluster_id}.prompt", "a") as f:
|
| 161 |
-
|
| 162 |
|
| 163 |
final_insights.append(text)
|
| 164 |
return final_insights
|
|
|
|
| 133 |
# print(cluster_id, " ", sentences)
|
| 134 |
final_prompt = PROMPT.replace("{{sentences}}", "\n".join(sentences))
|
| 135 |
|
| 136 |
+
# with open(f"prompter/insights_{cluster_id}.prompt", "w") as f:
|
| 137 |
+
# f.write(final_prompt)
|
| 138 |
|
| 139 |
# Generate insights for each cluster
|
| 140 |
response = openai.Completion.create(
|
|
|
|
| 157 |
# print(description)
|
| 158 |
# final_insights.append({"name": name_of_insight, "description": description})
|
| 159 |
|
| 160 |
+
# with open(f"prompter/insights_{cluster_id}.prompt", "a") as f:
|
| 161 |
+
# f.write(text)
|
| 162 |
|
| 163 |
final_insights.append(text)
|
| 164 |
return final_insights
|