Update app.py
Browse files
app.py
CHANGED
|
@@ -7,35 +7,24 @@ import pandas as pd
|
|
| 7 |
import numpy as np
|
| 8 |
import os
|
| 9 |
import time
|
|
|
|
|
|
|
| 10 |
from datetime import datetime
|
|
|
|
| 11 |
|
| 12 |
from utils import create_user_id
|
| 13 |
|
| 14 |
-
from azure.storage.fileshare import ShareServiceClient
|
| 15 |
|
| 16 |
|
| 17 |
-
import re
|
| 18 |
-
import json
|
| 19 |
-
print("1")
|
| 20 |
# ClimateQ&A imports
|
| 21 |
from climateqa.engine.llm import get_llm
|
| 22 |
-
print("2")
|
| 23 |
-
# from climateqa.chains import load_qa_chain_with_docs,load_qa_chain_with_text
|
| 24 |
-
# from climateqa.chains import load_reformulation_chain
|
| 25 |
from climateqa.engine.rag import make_rag_chain
|
| 26 |
-
print("3")
|
| 27 |
from climateqa.engine.vectorstore import get_pinecone_vectorstore
|
| 28 |
-
print("4")
|
| 29 |
from climateqa.engine.retriever import ClimateQARetriever
|
| 30 |
-
print("5")
|
| 31 |
from climateqa.engine.embeddings import get_embeddings_function
|
| 32 |
-
print("6")
|
| 33 |
from climateqa.engine.prompts import audience_prompts
|
| 34 |
-
print("7")
|
| 35 |
from climateqa.sample_questions import QUESTIONS
|
| 36 |
-
print("8")
|
| 37 |
from climateqa.constants import POSSIBLE_REPORTS
|
| 38 |
-
print("9")
|
| 39 |
from climateqa.utils import get_image_from_azure_blob_storage
|
| 40 |
|
| 41 |
# Load environment variables in local mode
|
|
@@ -52,8 +41,6 @@ theme = gr.themes.Base(
|
|
| 52 |
font=[gr.themes.GoogleFont("Poppins"), "ui-sans-serif", "system-ui", "sans-serif"],
|
| 53 |
)
|
| 54 |
|
| 55 |
-
print("1")
|
| 56 |
-
|
| 57 |
|
| 58 |
|
| 59 |
init_prompt = ""
|
|
@@ -97,13 +84,6 @@ def parse_output_llm_with_sources(output):
|
|
| 97 |
return content_parts
|
| 98 |
|
| 99 |
|
| 100 |
-
|
| 101 |
-
# Create embeddings function and LLM
|
| 102 |
-
print("1")
|
| 103 |
-
embeddings_function = get_embeddings_function()
|
| 104 |
-
print("1")
|
| 105 |
-
|
| 106 |
-
|
| 107 |
# Create vectorstore and retriever
|
| 108 |
vectorstore = get_pinecone_vectorstore(embeddings_function)
|
| 109 |
|
|
|
|
| 7 |
import numpy as np
|
| 8 |
import os
|
| 9 |
import time
|
| 10 |
+
import re
|
| 11 |
+
import json
|
| 12 |
from datetime import datetime
|
| 13 |
+
from azure.storage.fileshare import ShareServiceClient
|
| 14 |
|
| 15 |
from utils import create_user_id
|
| 16 |
|
|
|
|
| 17 |
|
| 18 |
|
|
|
|
|
|
|
|
|
|
| 19 |
# ClimateQ&A imports
|
| 20 |
from climateqa.engine.llm import get_llm
|
|
|
|
|
|
|
|
|
|
| 21 |
from climateqa.engine.rag import make_rag_chain
|
|
|
|
| 22 |
from climateqa.engine.vectorstore import get_pinecone_vectorstore
|
|
|
|
| 23 |
from climateqa.engine.retriever import ClimateQARetriever
|
|
|
|
| 24 |
from climateqa.engine.embeddings import get_embeddings_function
|
|
|
|
| 25 |
from climateqa.engine.prompts import audience_prompts
|
|
|
|
| 26 |
from climateqa.sample_questions import QUESTIONS
|
|
|
|
| 27 |
from climateqa.constants import POSSIBLE_REPORTS
|
|
|
|
| 28 |
from climateqa.utils import get_image_from_azure_blob_storage
|
| 29 |
|
| 30 |
# Load environment variables in local mode
|
|
|
|
| 41 |
font=[gr.themes.GoogleFont("Poppins"), "ui-sans-serif", "system-ui", "sans-serif"],
|
| 42 |
)
|
| 43 |
|
|
|
|
|
|
|
| 44 |
|
| 45 |
|
| 46 |
init_prompt = ""
|
|
|
|
| 84 |
return content_parts
|
| 85 |
|
| 86 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
# Create vectorstore and retriever
|
| 88 |
vectorstore = get_pinecone_vectorstore(embeddings_function)
|
| 89 |
|