Updated app 1.2.0

.gitignore

@@ -3,4 +3,5 @@ __pycache__/app.cpython-38.pyc
 __pycache__/app.cpython-39.pyc
 __pycache__/utils.cpython-38.pyc
 
-notebooks/
+notebooks/
+*.pyc

app.py

@@ -1,163 +1,187 @@
 import gradio as gr
-from haystack.document_stores import FAISSDocumentStore
-from haystack.nodes import EmbeddingRetriever
-import openai
 import pandas as pd
+import numpy as np
 import os
+from datetime import datetime
+
 from utils import (
     make_pairs,
     set_openai_api_key,
     create_user_id,
     to_completion,
 )
-import numpy as np
-from datetime import datetime
+
 from azure.storage.fileshare import ShareServiceClient
 
+# Langchain
+from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.schema import AIMessage, HumanMessage
+from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
+
+# ClimateQ&A imports
+from climateqa.llm import get_llm
+from climateqa.chains import load_climateqa_chain
+from climateqa.vectorstore import get_pinecone_vectorstore
+from climateqa.retriever import ClimateQARetriever
+from climateqa.prompts import audience_prompts
+
+# Load environment variables in local mode
 try:
     from dotenv import load_dotenv
     load_dotenv()
 except:
     pass
 
-
+# Set up Gradio Theme
 theme = gr.themes.Soft(
     primary_hue="sky",
     font=[gr.themes.GoogleFont("Poppins"), "ui-sans-serif", "system-ui", "sans-serif"],
 )
 
-init_prompt = (
-    "You are ClimateQA, an AI Assistant by Ekimetrics. "
-    "You are given a question and extracted parts of the IPCC and IPBES reports."
-    "Provide a clear and structured answer based on the context provided. "
-    "When relevant, use bullet points and lists to structure your answers."
-)
-sources_prompt = (
-    "When relevant, use facts and numbers from the following documents in your answer. "
-    "Whenever you use information from a document, reference it at the end of the sentence (ex: [doc 2]). "
-    "You don't have to use all documents, only if it makes sense in the conversation. "
-    "If no relevant information to answer the question is present in the documents, "
-    "just say you don't have enough information to answer."
-)
-
-
-def get_reformulation_prompt(query: str) -> str:
-    return f"""Reformulate the following user message to be a short standalone question in English, in the context of an educational discussion about climate change.
----
-query: La technologie nous sauvera-t-elle ?
-standalone question: Can technology help humanity mitigate the effects of climate change?
-language: French
----
-query: what are our reserves in fossil fuel?
-standalone question: What are the current reserves of fossil fuels and how long will they last?
-language: English
----
-query: what are the main causes of climate change?
-standalone question: What are the main causes of climate change in the last century?
-language: English
----
-query: {query}
-standalone question:"""
-
+init_prompt = ""
 
 system_template = {
     "role": "system",
     "content": init_prompt,
 }
 
-openai.api_type = "azure"
-openai.api_key = os.environ["api_key"]
-openai.api_base = os.environ["ressource_endpoint"]
-openai.api_version = "2023-06-01-preview"
-
-retriever = EmbeddingRetriever(
-    document_store=FAISSDocumentStore.load(
-        index_path="./climateqa_v3.faiss",
-        config_path="./climateqa_v3.json",
-    ),
-    embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
-    model_format="sentence_transformers",
-    progress_bar=False,
-)
 
-# retrieve_giec = EmbeddingRetriever(
-#     document_store=FAISSDocumentStore.load(
-#         index_path="./documents/climate_gpt_v2_only_giec.faiss",
-#         config_path="./documents/climate_gpt_v2_only_giec.json",
-#     ),
-#     embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
-#     model_format="sentence_transformers",
-# )
-
-credential = {
-    "account_key": os.environ["account_key"],
-    "account_name": os.environ["account_name"],
-}
+# credential = {
+#     "account_key": os.environ["account_key"],
+#     "account_name": os.environ["account_name"],
+# }
+
+# account_url = os.environ["account_url"]
+# file_share_name = "climategpt"
+# service = ShareServiceClient(account_url=account_url, credential=credential)
+# share_client = service.get_share_client(file_share_name)
 
-account_url = os.environ["account_url"]
-file_share_name = "climategpt"
-service = ShareServiceClient(account_url=account_url, credential=credential)
-share_client = service.get_share_client(file_share_name)
 user_id = create_user_id(10)
 
 
+#---------------------------------------------------------------------------
+# ClimateQ&A core functions
+#---------------------------------------------------------------------------
 
-def filter_sources(df,k_summary = 3,k_total = 10,source = "ipcc"):
-    assert source in ["ipcc","ipbes","all"]
+# Create embeddings function and LLM
+embeddings_function = HuggingFaceEmbeddings(model_name = "sentence-transformers/multi-qa-mpnet-base-dot-v1")
+llm = get_llm(max_tokens = 1024,temperature = 0.0,verbose = True,streaming = False,
+    callbacks=[StreamingStdOutCallbackHandler()],            
+)
+
+# Create vectorstore and retriever
+vectorstore = get_pinecone_vectorstore(embeddings_function)
+retriever = ClimateQARetriever(vectorstore=vectorstore,sources = ["IPCC"],k_summary = 3,k_total = 10)
+chain = load_climateqa_chain(retriever,llm)
+
+
+#---------------------------------------------------------------------------
+# ClimateQ&A Streaming
+# From https://github.com/gradio-app/gradio/issues/5345
+#---------------------------------------------------------------------------
+
+# from langchain.callbacks.base import BaseCallbackHandler
+# from queue import Queue, Empty
+# from threading import Thread
+# from collections.abc import Generator
+
+# class QueueCallback(BaseCallbackHandler):
+#     """Callback handler for streaming LLM responses to a queue."""
+
+#     def __init__(self, q):
+#         self.q = q
+
+#     def on_llm_new_token(self, token: str, **kwargs: any) -> None:
+#         self.q.put(token)
+
+#     def on_llm_end(self, *args, **kwargs: any) -> None:
+#         return self.q.empty()
+    
+
+# def stream(input_text) -> Generator:
+#     # Create a Queue
+#     q = Queue()
+#     job_done = object()
+
+#     llm = get_llm(max_tokens = 1024,temperature = 0.0,verbose = True,streaming = True,
+#         callbacks=[QueueCallback(q)],            
+#     )
+
+#     chain = load_climateqa_chain(retriever,llm)
+
+#     # Create a funciton to call - this will run in a thread
+#     def task():
+#         answer = chain({"query":input_text,"audience":"expert climate scientist"})
+#         q.put(job_done)
+
+#     # Create a thread and start the function
+#     t = Thread(target=task)
+#     t.start()
+
+#     content = ""
+
+#     # Get each new token from the queue and yield for our generator
+#     while True:
+#         try:
+#             next_token = q.get(True, timeout=1)
+#             if next_token is job_done:
+#                 break
+#             content += next_token
+#             yield next_token, content
+#         except Empty:
+#             continue
+
+
+def answer_user(message,history):
+    return message, history + [[message, None]]
+
+
+def answer_bot(message,history):
+    print("YO",message,history)
+    # history_langchain_format = []
+    # for human, ai in history:
+    #     history_langchain_format.append(HumanMessage(content=human))
+    #     history_langchain_format.append(AIMessage(content=ai))
+    # history_langchain_format.append(HumanMessage(content=message)
+    # for next_token, content in stream(message):
+    #     yield(content)
+    output = chain({"query":message,"audience":"expert climate scientist"})
+    question = output["question"]
+    sources = output["source_documents"]
+
+    if len(sources) > 0:
+        sources_text = []
+        for i, d in enumerate(sources, 1):
+            sources_text.append(make_html_source(d,i))
+        sources_text = "\n\n".join([f"Query used for retrieval:\n{question}"] + sources_text)
+
+        history[-1][1] = output["answer"]
+        return "",history,sources_text
 
-    # Filter by source
-    if source == "ipcc":
-        df = df.loc[df["source"]=="IPCC"]
-    elif source == "ipbes":
-        df = df.loc[df["source"]=="IPBES"]
-    else:
-        pass
-
-    # Separate summaries and full reports
-    df_summaries = df.loc[df["report_type"].isin(["SPM","TS"])]
-    df_full = df.loc[~df["report_type"].isin(["SPM","TS"])]
-
-    # Find passages from summaries dataset
-    passages_summaries = df_summaries.head(k_summary)
-
-    # Find passages from full reports dataset
-    passages_fullreports = df_full.head(k_total - len(passages_summaries))
-
-    # Concatenate passages
-    passages = pd.concat([passages_summaries,passages_fullreports],axis = 0,ignore_index = True)
-    return passages
-
-
-def retrieve_with_summaries(query,retriever,k_summary = 3,k_total = 10,source = "ipcc",max_k = 100,threshold = 0.555,as_dict = True):
-    assert max_k > k_total
-    docs = retriever.retrieve(query,top_k = max_k)
-    docs = [{**x.meta,"score":x.score,"content":x.content} for x in docs if x.score > threshold]
-    if len(docs) == 0:
-        return []
-    res = pd.DataFrame(docs)
-    passages_df = filter_sources(res,k_summary,k_total,source)
-    if as_dict:
-        contents = passages_df["content"].tolist()
-        meta = passages_df.drop(columns = ["content"]).to_dict(orient = "records")
-        passages = []
-        for i in range(len(contents)):
-            passages.append({"content":contents[i],"meta":meta[i]})
-        return passages
     else:
-        return passages_df
+        sources_text = "⚠️ No relevant passages found in the climate science reports (IPCC and IPBES)"
+        complete_response = "**⚠️ No relevant passages found in the climate science reports (IPCC and IPBES), you may want to ask a more specific question (specifying your question on climate issues).**"
+        history[-1][1] = complete_response
+        return "",history, sources_text
+
+
+#---------------------------------------------------------------------------
+# ClimateQ&A core functions
+#---------------------------------------------------------------------------
 
 
 def make_html_source(source,i):
-    meta = source['meta']
+    meta = source.metadata
+    content = source.page_content.split(":",1)[1].strip()
     return f"""
 <div class="card">
     <div class="card-content">
-        <h2>Doc {i} - {meta['short_name']} - Page {meta['page_number']}</h2>
-        <p>{source['content']}</p>
+        <h2>Doc {i} - {meta['short_name']} - Page {int(meta['page_number'])}</h2>
+        <p>{content}</p>
     </div>
     <div class="card-footer">
         <span>{meta['name']}</span>
-        <a href="{meta['url']}#page={meta['page_number']}" target="_blank" class="pdf-link">
+        <a href="{meta['url']}#page={int(meta['page_number'])}" target="_blank" class="pdf-link">
             <span role="img" aria-label="Open PDF">🔗</span>
         </a>
     </div>
@@ -166,106 +190,106 @@ def make_html_source(source,i):
 
 
 
-def chat(
-    user_id: str,
-    query: str,
-    history: list = [system_template],
-    report_type: str = "IPCC",
-    threshold: float = 0.555,
-) -> tuple:
-    """retrieve relevant documents in the document store then query gpt-turbo
-
-    Args:
-        query (str): user message.
-        history (list, optional): history of the conversation. Defaults to [system_template].
-        report_type (str, optional): should be "All available" or "IPCC only". Defaults to "All available".
-        threshold (float, optional): similarity threshold, don't increase more than 0.568. Defaults to 0.56.
-
-    Yields:
-        tuple: chat gradio format, chat openai format, sources used.
-    """
-
-    if report_type not in ["IPCC","IPBES"]: report_type = "all"
-    print("Searching in ",report_type," reports")
-    # if report_type == "All available":
-    #     retriever = retrieve_all
-    # elif report_type == "IPCC only":
-    #     retriever = retrieve_giec
-    # else:
-    #     raise Exception("report_type arg should be in (All available, IPCC only)")
-
-    reformulated_query = openai.Completion.create(
-        engine="EkiGPT",
-        prompt=get_reformulation_prompt(query),
-        temperature=0,
-        max_tokens=128,
-        stop=["\n---\n", "<|im_end|>"],
-    )
-    reformulated_query = reformulated_query["choices"][0]["text"]
-    reformulated_query, language = reformulated_query.split("\n")
-    language = language.split(":")[1].strip()
-
-
-    sources = retrieve_with_summaries(reformulated_query,retriever,k_total = 10,k_summary = 3,as_dict = True,source = report_type.lower(),threshold = threshold)
-    response_retriever = {
-      "language":language,
-      "reformulated_query":reformulated_query,
-      "query":query,
-      "sources":sources,
-    }
-
-    # docs = [d for d in retriever.retrieve(query=reformulated_query, top_k=10) if d.score > threshold]
-    messages = history + [{"role": "user", "content": query}]
-
-    if len(sources) > 0:
-        docs_string = []
-        docs_html = []
-        for i, d in enumerate(sources, 1):
-            docs_string.append(f"📃 Doc {i}: {d['meta']['short_name']} page {d['meta']['page_number']}\n{d['content']}")
-            docs_html.append(make_html_source(d,i))
-        docs_string = "\n\n".join([f"Query used for retrieval:\n{reformulated_query}"] + docs_string)
-        docs_html = "\n\n".join([f"Query used for retrieval:\n{reformulated_query}"] + docs_html)
-        messages.append({"role": "system", "content": f"{sources_prompt}\n\n{docs_string}\n\nAnswer in {language}:"})
-
-
-        response = openai.Completion.create(
-            engine="EkiGPT",
-            prompt=to_completion(messages),
-            temperature=0,  # deterministic
-            stream=True,
-            max_tokens=1024,
-        )
-
-        complete_response = ""
-        messages.pop()
-
-        messages.append({"role": "assistant", "content": complete_response})
-        timestamp = str(datetime.now().timestamp())
-        file = user_id[0] + timestamp + ".json"
-        logs = {
-            "user_id": user_id[0],
-            "prompt": query,
-            "retrived": sources,
-            "report_type": report_type,
-            "prompt_eng": messages[0],
-            "answer": messages[-1]["content"],
-            "time": timestamp,
-        }
-        log_on_azure(file, logs, share_client)
-
-        for chunk in response:
-            if (chunk_message := chunk["choices"][0].get("text")) and chunk_message != "<|im_end|>":
-                complete_response += chunk_message
-                messages[-1]["content"] = complete_response
-                gradio_format = make_pairs([a["content"] for a in messages[1:]])
-                yield gradio_format, messages, docs_html
-
-    else:
-        docs_string = "⚠️ No relevant passages found in the climate science reports (IPCC and IPBES)"
-        complete_response = "**⚠️ No relevant passages found in the climate science reports (IPCC and IPBES), you may want to ask a more specific question (specifying your question on climate issues).**"
-        messages.append({"role": "assistant", "content": complete_response})
-        gradio_format = make_pairs([a["content"] for a in messages[1:]])
-        yield gradio_format, messages, docs_string
+# def chat(
+#     user_id: str,
+#     query: str,
+#     history: list = [system_template],
+#     report_type: str = "IPCC",
+#     threshold: float = 0.555,
+# ) -> tuple:
+#     """retrieve relevant documents in the document store then query gpt-turbo
+
+#     Args:
+#         query (str): user message.
+#         history (list, optional): history of the conversation. Defaults to [system_template].
+#         report_type (str, optional): should be "All available" or "IPCC only". Defaults to "All available".
+#         threshold (float, optional): similarity threshold, don't increase more than 0.568. Defaults to 0.56.
+
+#     Yields:
+#         tuple: chat gradio format, chat openai format, sources used.
+#     """
+
+#     if report_type not in ["IPCC","IPBES"]: report_type = "all"
+#     print("Searching in ",report_type," reports")
+#     # if report_type == "All available":
+#     #     retriever = retrieve_all
+#     # elif report_type == "IPCC only":
+#     #     retriever = retrieve_giec
+#     # else:
+#     #     raise Exception("report_type arg should be in (All available, IPCC only)")
+
+#     reformulated_query = openai.Completion.create(
+#         engine="EkiGPT",
+#         prompt=get_reformulation_prompt(query),
+#         temperature=0,
+#         max_tokens=128,
+#         stop=["\n---\n", "<|im_end|>"],
+#     )
+#     reformulated_query = reformulated_query["choices"][0]["text"]
+#     reformulated_query, language = reformulated_query.split("\n")
+#     language = language.split(":")[1].strip()
+
+
+#     sources = retrieve_with_summaries(reformulated_query,retriever,k_total = 10,k_summary = 3,as_dict = True,source = report_type.lower(),threshold = threshold)
+#     response_retriever = {
+#       "language":language,
+#       "reformulated_query":reformulated_query,
+#       "query":query,
+#       "sources":sources,
+#     }
+
+#     # docs = [d for d in retriever.retrieve(query=reformulated_query, top_k=10) if d.score > threshold]
+#     messages = history + [{"role": "user", "content": query}]
+
+#     if len(sources) > 0:
+#         docs_string = []
+#         docs_html = []
+#         for i, d in enumerate(sources, 1):
+#             docs_string.append(f"📃 Doc {i}: {d['meta']['short_name']} page {d['meta']['page_number']}\n{d['content']}")
+#             docs_html.append(make_html_source(d,i))
+#         docs_string = "\n\n".join([f"Query used for retrieval:\n{reformulated_query}"] + docs_string)
+#         docs_html = "\n\n".join([f"Query used for retrieval:\n{reformulated_query}"] + docs_html)
+#         messages.append({"role": "system", "content": f"{sources_prompt}\n\n{docs_string}\n\nAnswer in {language}:"})
+
+
+#         response = openai.Completion.create(
+#             engine="EkiGPT",
+#             prompt=to_completion(messages),
+#             temperature=0,  # deterministic
+#             stream=True,
+#             max_tokens=1024,
+#         )
+
+#         complete_response = ""
+#         messages.pop()
+
+#         messages.append({"role": "assistant", "content": complete_response})
+#         timestamp = str(datetime.now().timestamp())
+#         file = user_id[0] + timestamp + ".json"
+#         logs = {
+#             "user_id": user_id[0],
+#             "prompt": query,
+#             "retrived": sources,
+#             "report_type": report_type,
+#             "prompt_eng": messages[0],
+#             "answer": messages[-1]["content"],
+#             "time": timestamp,
+#         }
+#         log_on_azure(file, logs, share_client)
+
+#         for chunk in response:
+#             if (chunk_message := chunk["choices"][0].get("text")) and chunk_message != "<|im_end|>":
+#                 complete_response += chunk_message
+#                 messages[-1]["content"] = complete_response
+#                 gradio_format = make_pairs([a["content"] for a in messages[1:]])
+#                 yield gradio_format, messages, docs_html
+
+#     else:
+#         docs_string = "⚠️ No relevant passages found in the climate science reports (IPCC and IPBES)"
+#         complete_response = "**⚠️ No relevant passages found in the climate science reports (IPCC and IPBES), you may want to ask a more specific question (specifying your question on climate issues).**"
+#         messages.append({"role": "assistant", "content": complete_response})
+#         gradio_format = make_pairs([a["content"] for a in messages[1:]])
+#         yield gradio_format, messages, docs_string
 
 
 def save_feedback(feed: str, user_id):
@@ -290,162 +314,194 @@ def log_on_azure(file, logs, share_client):
     file_client.upload_file(str(logs))
 
 
-with gr.Blocks(title="🌍 Climate Q&A", css="style.css", theme=theme) as demo:
-    user_id_state = gr.State([user_id])
 
-    # Gradio
-    gr.Markdown("<h1><center>Climate Q&A 🌍</center></h1>")
-    gr.Markdown("<h4><center>Ask climate-related questions to the IPCC reports</center></h4>")
-    gr.Markdown("<h2 style='color:red'><center>WARNING - We have a small temporary bug on HF platform, you can use the new v1.0 version in the meantime https://climateqa.com</center></h4>")
-
-
-    with gr.Row():
-        with gr.Column(scale=2):
-            chatbot = gr.Chatbot(elem_id="chatbot", label="ClimateQ&A chatbot",show_label = False)
-            state = gr.State([system_template])
-
-            with gr.Row():
-                ask = gr.Textbox(
-                    show_label=False,
-                    placeholder="Ask here your climate-related question and press enter",
-                ).style(container=False)
-                ask_examples_hidden = gr.Textbox(elem_id="hidden-message")
-
-            examples_questions = gr.Examples(
-                [
-                    "Is climate change caused by humans?",
-                    "What evidence do we have of climate change?",
-                    "What are the impacts of climate change?",
-                    "Can climate change be reversed?",
-                    "What is the difference between climate change and global warming?",
-                    "What can individuals do to address climate change?",
-                    "What are the main causes of climate change?",
-                    "What is the Paris Agreement and why is it important?",
-                    "Which industries have the highest GHG emissions?",
-                    "Is climate change a hoax created by the government or environmental organizations?",
-                    "What is the relationship between climate change and biodiversity loss?",
-                    "What is the link between gender equality and climate change?",
-                    "Is the impact of climate change really as severe as it is claimed to be?",
-                    "What is the impact of rising sea levels?",
-                    "What are the different greenhouse gases (GHG)?",
-                    "What is the warming power of methane?",
-                    "What is the jet stream?",
-                    "What is the breakdown of carbon sinks?",
-                    "How do the GHGs work ? Why does temperature increase ?",
-                    "What is the impact of global warming on ocean currents?",
-                    "How much warming is possible in 2050?",
-                    "What is the impact of climate change in Africa?",
-                    "Will climate change accelerate diseases and epidemics like COVID?",
-                    "What are the economic impacts of climate change?",
-                    "How much is the cost of inaction ?",
-                    "What is the relationship between climate change and poverty?",
-                    "What are the most effective strategies and technologies for reducing greenhouse gas (GHG) emissions?",
-                    "Is economic growth possible? What do you think about degrowth?",
-                    "Will technology save us?",
-                    "Is climate change a natural phenomenon ?",
-                    "Is climate change really happening or is it just a natural fluctuation in Earth's temperature?",
-                    "Is the scientific consensus on climate change really as strong as it is claimed to be?",
-                ],
-                [ask_examples_hidden],
-                examples_per_page=15,
-            )
-
-        with gr.Column(scale=1, variant="panel"):
-            gr.Markdown("### Sources")
-            sources_textbox = gr.Markdown(show_label=False)
-
-    dropdown_sources = gr.inputs.Dropdown(
-        ["IPCC", "IPBES","IPCC and IPBES"],
-        default="IPCC",
-        label="Select reports",
-    )
-    ask.submit(
-        fn=chat,
-        inputs=[
-            user_id_state,
-            ask,
-            state,
-            dropdown_sources
-
-        ],
-        outputs=[chatbot, state, sources_textbox],
-    )
-    ask.submit(reset_textbox, [], [ask])
-
-    ask_examples_hidden.change(
-        fn=chat,
-        inputs=[
-            user_id_state,
-            ask_examples_hidden,
-            state,
-            dropdown_sources
-        ],
-        outputs=[chatbot, state, sources_textbox],
-    )
 
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown(
-                """
-<p><b>Climate change and environmental disruptions have become some of the most pressing challenges facing our planet today</b>. As global temperatures rise and ecosystems suffer, it is essential for individuals to understand the gravity of the situation in order to make informed decisions and advocate for appropriate policy changes.</p>
-<p>However, comprehending the vast and complex scientific information can be daunting, as the scientific consensus references, such as <b>the Intergovernmental Panel on Climate Change (IPCC) reports, span thousands of pages</b>. To bridge this gap and make climate science more accessible, we introduce <b>ClimateQ&A as a tool to distill expert-level knowledge into easily digestible insights about climate science.</b></p>
-<div class="tip-box">
-<div class="tip-box-title">
-    <span class="light-bulb" role="img" aria-label="Light Bulb">💡</span>
-    How does ClimateQ&A work?
-</div>
-ClimateQ&A harnesses modern OCR techniques to parse and preprocess IPCC reports. By leveraging state-of-the-art question-answering algorithms, <i>ClimateQ&A is able to sift through the extensive collection of climate scientific reports and identify relevant passages in response to user inquiries</i>. Furthermore, the integration of the ChatGPT API allows ClimateQ&A to present complex data in a user-friendly manner, summarizing key points and facilitating communication of climate science to a wider audience.
-</div>
 
-<div class="warning-box">
-Version 0.2-beta - This tool is under active development
-</div>
+# --------------------------------------------------------------------
+# Gradio
+# --------------------------------------------------------------------
 
 
-"""
-            )
-
-        with gr.Column(scale=1):
-            gr.Markdown("![](https://i.postimg.cc/fLvsvMzM/Untitled-design-5.png)")
-            gr.Markdown("*Source : IPCC AR6 - Synthesis Report of the IPCC 6th assessment report (AR6)*")
-
-    gr.Markdown("## How to use ClimateQ&A")
-    with gr.Row():
-        with gr.Column(scale=1):
-            gr.Markdown(
-                """
-    ### 💪 Getting started
-    - In the chatbot section, simply type your climate-related question, and ClimateQ&A will provide an answer with references to relevant IPCC reports.
-        - ClimateQ&A retrieves specific passages from the IPCC reports to help answer your question accurately.
-        - Source information, including page numbers and passages, is displayed on the right side of the screen for easy verification.
-        - Feel free to ask follow-up questions within the chatbot for a more in-depth understanding.
-    - ClimateQ&A integrates multiple sources (IPCC and IPBES, … ) to cover various aspects of environmental science, such as climate change and biodiversity. See all sources used below.
-    """
-            )
-        with gr.Column(scale=1):
-            gr.Markdown(
-                """
-    ### ⚠️ Limitations
+
+
+
+with gr.Blocks(title="🌍 Climate Q&A", css="style.css", theme=theme) as demo:
+    # user_id_state = gr.State([user_id])
+
+    # Gradio
+    gr.Markdown("<h1><center>Climate Q&A 🌍</center></h1>")
+    gr.Markdown("<h4><center>Ask climate-related questions to the IPCC and IPBES reports using AI</center></h4>")
+
+    with gr.Tab("💬 Chatbot"):
+
+        with gr.Row():
+            with gr.Column(scale=2):
+                # state = gr.State([system_template])
+                bot = gr.Chatbot(height=400)
+
+                with gr.Row():
+                    with gr.Column(scale = 7):
+                        textbox=gr.Textbox(placeholder="Ask me a question about climate change or biodiversity in any language, and press Enter",show_label=False)
+                    with gr.Column(scale = 1):
+                        submit_button = gr.Button("Submit")
+                
+                examples_hidden = gr.Textbox(elem_id="hidden-message")
+
+                examples_questions = gr.Examples(
+                    [
+                        "Is climate change caused by humans?",
+                        "What evidence do we have of climate change?",
+                        "What are the impacts of climate change?",
+                        "Can climate change be reversed?",
+                        "What is the difference between climate change and global warming?",
+                        "What can individuals do to address climate change?",
+                        "What are the main causes of climate change?",
+                        "What is the Paris Agreement and why is it important?",
+                        "Which industries have the highest GHG emissions?",
+                        "Is climate change a hoax created by the government or environmental organizations?",
+                        "What is the relationship between climate change and biodiversity loss?",
+                        "What is the link between gender equality and climate change?",
+                        "Is the impact of climate change really as severe as it is claimed to be?",
+                        "What is the impact of rising sea levels?",
+                        "What are the different greenhouse gases (GHG)?",
+                        "What is the warming power of methane?",
+                        "What is the jet stream?",
+                        "What is the breakdown of carbon sinks?",
+                        "How do the GHGs work ? Why does temperature increase ?",
+                        "What is the impact of global warming on ocean currents?",
+                        "How much warming is possible in 2050?",
+                        "What is the impact of climate change in Africa?",
+                        "Will climate change accelerate diseases and epidemics like COVID?",
+                        "What are the economic impacts of climate change?",
+                        "How much is the cost of inaction ?",
+                        "What is the relationship between climate change and poverty?",
+                        "What are the most effective strategies and technologies for reducing greenhouse gas (GHG) emissions?",
+                        "Is economic growth possible? What do you think about degrowth?",
+                        "Will technology save us?",
+                        "Is climate change a natural phenomenon ?",
+                        "Is climate change really happening or is it just a natural fluctuation in Earth's temperature?",
+                        "Is the scientific consensus on climate change really as strong as it is claimed to be?",
+                    ],
+                    [examples_hidden],
+                    examples_per_page=10,
+                )
+
+            with gr.Column(scale=1, variant="panel"):
+
+                dropdown_sources = gr.CheckboxGroup(
+                    ["IPCC", "IPBES"],
+                    label="Select reports",
+                    value = ["IPCC"],
+                )
+
+                dropdown_audience = gr.Dropdown(
+                    ["Children","Adult","Experts"],
+                    label="Select audience",
+                    value="Experts",
+                )
+
+                gr.Markdown("### Sources")
+                sources_textbox = gr.Markdown(show_label=False)
+
+            # textbox.submit(predict_climateqa,[textbox,bot],[None,bot,sources_textbox])
+
+            textbox.submit(answer_user, [textbox, bot], [textbox, bot], queue=False).then(
+                    answer_bot, [textbox,bot], [textbox,bot,sources_textbox]
+                )
+            examples_hidden.change(answer_user, [examples_hidden, bot], [textbox, bot], queue=False).then(
+                    answer_bot, [textbox,bot], [textbox,bot,sources_textbox]
+                )
+        
+            submit_button.click(answer_user, [textbox, bot], [textbox, bot], queue=False).then(
+                    answer_bot, [textbox,bot], [textbox,bot,sources_textbox]
+                )
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+#---------------------------------------------------------------------------------------
+# OTHER TABS
+#---------------------------------------------------------------------------------------
+
+
+    with gr.Tab("ℹ️ About ClimateQ&A"):
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown(
+                    """
+    <p><b>Climate change and environmental disruptions have become some of the most pressing challenges facing our planet today</b>. As global temperatures rise and ecosystems suffer, it is essential for individuals to understand the gravity of the situation in order to make informed decisions and advocate for appropriate policy changes.</p>
+    <p>However, comprehending the vast and complex scientific information can be daunting, as the scientific consensus references, such as <b>the Intergovernmental Panel on Climate Change (IPCC) reports, span thousands of pages</b>. To bridge this gap and make climate science more accessible, we introduce <b>ClimateQ&A as a tool to distill expert-level knowledge into easily digestible insights about climate science.</b></p>
+    <div class="tip-box">
+    <div class="tip-box-title">
+        <span class="light-bulb" role="img" aria-label="Light Bulb">💡</span>
+        How does ClimateQ&A work?
+    </div>
+    ClimateQ&A harnesses modern OCR techniques to parse and preprocess IPCC reports. By leveraging state-of-the-art question-answering algorithms, <i>ClimateQ&A is able to sift through the extensive collection of climate scientific reports and identify relevant passages in response to user inquiries</i>. Furthermore, the integration of the ChatGPT API allows ClimateQ&A to present complex data in a user-friendly manner, summarizing key points and facilitating communication of climate science to a wider audience.
+    </div>
+
     <div class="warning-box">
-    <ul>
-        <li>Please note that, like any AI, the model may occasionally generate an inaccurate or imprecise answer. Always refer to the provided sources to verify the validity of the information given. If you find any issues with the response, kindly provide feedback to help improve the system.</li>
-        <li>ClimateQ&A is specifically designed for climate-related inquiries. If you ask a non-environmental question, the chatbot will politely remind you that its focus is on climate and environmental issues.</li>
+    Version 0.2-beta - This tool is under active development
     </div>
+
+
     """
-            )
+                )
+
+            with gr.Column(scale=1):
+                gr.Markdown("![](https://i.postimg.cc/fLvsvMzM/Untitled-design-5.png)")
+                gr.Markdown("*Source : IPCC AR6 - Synthesis Report of the IPCC 6th assessment report (AR6)*")
+
+        gr.Markdown("## How to use ClimateQ&A")
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown(
+                    """
+        ### 💪 Getting started
+        - In the chatbot section, simply type your climate-related question, and ClimateQ&A will provide an answer with references to relevant IPCC reports.
+            - ClimateQ&A retrieves specific passages from the IPCC reports to help answer your question accurately.
+            - Source information, including page numbers and passages, is displayed on the right side of the screen for easy verification.
+            - Feel free to ask follow-up questions within the chatbot for a more in-depth understanding.
+            - You can ask question in any language, ClimateQ&A is multi-lingual !
+        - ClimateQ&A integrates multiple sources (IPCC and IPBES, … ) to cover various aspects of environmental science, such as climate change and biodiversity. See all sources used below.
+        """
+                )
+            with gr.Column(scale=1):
+                gr.Markdown(
+                    """
+        ### ⚠️ Limitations
+        <div class="warning-box">
+        <ul>
+            <li>Please note that, like any AI, the model may occasionally generate an inaccurate or imprecise answer. Always refer to the provided sources to verify the validity of the information given. If you find any issues with the response, kindly provide feedback to help improve the system.</li>
+            <li>ClimateQ&A is specifically designed for climate-related inquiries. If you ask a non-environmental question, the chatbot will politely remind you that its focus is on climate and environmental issues.</li>
+        </div>
+        """
+                )
+
+
+    with gr.Tab("📧 Contact, feedback and feature requests"):
+        gr.Markdown(
+            """
 
-    gr.Markdown("## 🙏 Feedback and feature requests")
-    gr.Markdown(
+        🤞 For any question or press request, contact Théo Alves Da Costa at <b>[email protected]</b>
+
+        - ClimateQ&A welcomes community contributions. To participate, head over to the Community Tab and create a "New Discussion" to ask questions and share your insights.
+        - Provide feedback through email, letting us know which insights you found accurate, useful, or not. Your input will help us improve the platform.
+        - Only a few sources (see below) are integrated (all IPCC, IPBES), if you are a climate science researcher and net to sift through another report, please let us know.
+        
+        *This tool has been developed by the R&D lab at **Ekimetrics** (Jean Lelong, Nina Achache, Gabriel Olympie, Nicolas Chesneau, Natalia De la Calzada, Théo Alves Da Costa)*
         """
-    ### Beta test
-    - ClimateQ&A welcomes community contributions. To participate, head over to the Community Tab and create a "New Discussion" to ask questions and share your insights.
-    - Provide feedback through email, letting us know which insights you found accurate, useful, or not. Your input will help us improve the platform.
-    - Only a few sources (see below) are integrated (all IPCC, IPBES), if you are a climate science researcher and net to sift through another report, please let us know.
-    
-    If you need us to ask another climate science report or ask any question, contact us at <b>[email protected]</b>
-    """
-    )
+        )
     # with gr.Row():
     #     with gr.Column(scale=1):
     #         gr.Markdown("### Feedbacks")
@@ -475,50 +531,49 @@ Version 0.2-beta - This tool is under active development
     # openai_api_key_textbox.change(set_openai_api_key, inputs=[openai_api_key_textbox])
     # openai_api_key_textbox.submit(set_openai_api_key, inputs=[openai_api_key_textbox])
 
-    gr.Markdown(
-        """
-
-
-## 📚 Sources
-| Source | Report | URL | Number of pages | Release date |
-| --- | --- | --- | --- | --- |
-IPCC | Summary for Policymakers. In: Climate Change 2021: The Physical Science Basis. Contribution of the WGI to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg1/downloads/report/IPCC_AR6_WGI_SPM.pdf | 32 | 2021
-IPCC | Full Report. In: Climate Change 2021: The Physical Science Basis. Contribution of the WGI to the AR6 of the IPCC. | https://report.ipcc.ch/ar6/wg1/IPCC_AR6_WGI_FullReport.pdf | 2409 | 2021
-IPCC | Technical Summary. In: Climate Change 2021: The Physical Science Basis. Contribution of the WGI to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg1/downloads/report/IPCC_AR6_WGI_TS.pdf | 112 | 2021
-IPCC | Summary for Policymakers. In: Climate Change 2022: Impacts, Adaptation and Vulnerability. Contribution of the WGII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg2/downloads/report/IPCC_AR6_WGII_SummaryForPolicymakers.pdf | 34 | 2022
-IPCC | Technical Summary. In: Climate Change 2022: Impacts, Adaptation and Vulnerability. Contribution of the WGII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg2/downloads/report/IPCC_AR6_WGII_TechnicalSummary.pdf | 84 | 2022
-IPCC | Full Report. In: Climate Change 2022: Impacts, Adaptation and Vulnerability. Contribution of the WGII to the AR6 of the IPCC. | https://report.ipcc.ch/ar6/wg2/IPCC_AR6_WGII_FullReport.pdf | 3068 | 2022
-IPCC | Summary for Policymakers. In: Climate Change 2022: Mitigation of Climate Change. Contribution of the WGIII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg3/downloads/report/IPCC_AR6_WGIII_SummaryForPolicymakers.pdf | 50 | 2022
-IPCC | Technical Summary. In: Climate Change 2022: Mitigation of Climate Change. Contribution of the WGIII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg3/downloads/report/IPCC_AR6_WGIII_TechnicalSummary.pdf | 102 | 2022
-IPCC | Full Report. In: Climate Change 2022: Mitigation of Climate Change. Contribution of the WGIII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg3/downloads/report/IPCC_AR6_WGIII_FullReport.pdf | 2258 | 2022
-IPCC | Summary for Policymakers. In: Global Warming of 1.5°C. An IPCC Special Report on the impacts of global warming of 1.5°C above pre-industrial levels and related global greenhouse gas emission pathways, in the context of strengthening the global response to the threat of climate change, sustainable development, and efforts to eradicate poverty. | https://www.ipcc.ch/site/assets/uploads/sites/2/2022/06/SPM_version_report_LR.pdf | 24 | 2018
-IPCC | Summary for Policymakers. In: Climate Change and Land: an IPCC special report on climate change, desertification, land degradation, sustainable land management, food security, and greenhouse gas fluxes in terrestrial ecosystems. | https://www.ipcc.ch/site/assets/uploads/sites/4/2022/11/SRCCL_SPM.pdf | 36 | 2019
-IPCC | Summary for Policymakers. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/01_SROCC_SPM_FINAL.pdf | 36 | 2019
-IPCC | Technical Summary. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/02_SROCC_TS_FINAL.pdf | 34 | 2019
-IPCC | Chapter 1 - Framing and Context of the Report. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/03_SROCC_Ch01_FINAL.pdf | 60 | 2019
-IPCC | Chapter 2 - High Mountain Areas. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/04_SROCC_Ch02_FINAL.pdf | 72 | 2019
-IPCC | Chapter 3 - Polar Regions. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/05_SROCC_Ch03_FINAL.pdf | 118 | 2019
-IPCC | Chapter 4 - Sea Level Rise and Implications for Low-Lying Islands, Coasts and Communities. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/06_SROCC_Ch04_FINAL.pdf | 126 | 2019
-IPCC | Chapter 5 -  Changing Ocean, Marine Ecosystems, and Dependent Communities. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/07_SROCC_Ch05_FINAL.pdf | 142 | 2019
-IPCC | Chapter 6 - Extremes, Abrupt Changes and Managing Risk. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/08_SROCC_Ch06_FINAL.pdf | 68 | 2019
-IPCC | Cross-Chapter Box 9: Integrative Cross-Chapter Box on Low-Lying Islands and Coasts. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2019/11/11_SROCC_CCB9-LLIC_FINAL.pdf | 18 | 2019
-IPCC | Annex I: Glossary [Weyer, N.M. (ed.)]. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/10_SROCC_AnnexI-Glossary_FINAL.pdf | 28 | 2019
-IPBES | Full Report. Global assessment report on biodiversity and ecosystem services of the IPBES. | https://zenodo.org/record/6417333/files/202206_IPBES%20GLOBAL%20REPORT_FULL_DIGITAL_MARCH%202022.pdf | 1148 | 2019
-IPBES | Summary for Policymakers. Global assessment report on biodiversity and ecosystem services of the IPBES (Version 1). | https://zenodo.org/record/3553579/files/ipbes_global_assessment_report_summary_for_policymakers.pdf | 60 | 2019
-IPBES | Full Report. Thematic assessment of the sustainable use of wild species of the IPBES. | https://zenodo.org/record/7755805/files/IPBES_ASSESSMENT_SUWS_FULL_REPORT.pdf | 1008 | 2022
-IPBES | Summary for Policymakers. Summary for policymakers of the thematic assessment of the sustainable use of wild species of the IPBES. | https://zenodo.org/record/7411847/files/EN_SPM_SUSTAINABLE%20USE%20OF%20WILD%20SPECIES.pdf | 44 | 2022
-IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for Africa. | https://zenodo.org/record/3236178/files/ipbes_assessment_report_africa_EN.pdf | 494 | 2018
-IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for Africa. | https://zenodo.org/record/3236189/files/ipbes_assessment_spm_africa_EN.pdf | 52 | 2018
-IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for the Americas. | https://zenodo.org/record/3236253/files/ipbes_assessment_report_americas_EN.pdf | 660 | 2018
-IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for the Americas. | https://zenodo.org/record/3236292/files/ipbes_assessment_spm_americas_EN.pdf | 44 | 2018
-IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for Asia and the Pacific. | https://zenodo.org/record/3237374/files/ipbes_assessment_report_ap_EN.pdf | 616 | 2018
-IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for Asia and the Pacific. | https://zenodo.org/record/3237383/files/ipbes_assessment_spm_ap_EN.pdf | 44 | 2018
-IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for Europe and Central Asia. | https://zenodo.org/record/3237429/files/ipbes_assessment_report_eca_EN.pdf | 894 | 2018
-IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for Europe and Central Asia. | https://zenodo.org/record/3237468/files/ipbes_assessment_spm_eca_EN.pdf | 52 | 2018
-IPBES | Full Report. Assessment Report on Land Degradation and Restoration. | https://zenodo.org/record/3237393/files/ipbes_assessment_report_ldra_EN.pdf | 748 | 2018
-IPBES | Summary for Policymakers. Assessment Report on Land Degradation and Restoration. | https://zenodo.org/record/3237393/files/ipbes_assessment_report_ldra_EN.pdf | 48 | 2018
-
-## 🛢️ Carbon Footprint
+    with gr.Tab("📚 Sources"):
+        gr.Markdown("""
+    | Source | Report | URL | Number of pages | Release date |
+    | --- | --- | --- | --- | --- |
+    IPCC | Summary for Policymakers. In: Climate Change 2021: The Physical Science Basis. Contribution of the WGI to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg1/downloads/report/IPCC_AR6_WGI_SPM.pdf | 32 | 2021
+    IPCC | Full Report. In: Climate Change 2021: The Physical Science Basis. Contribution of the WGI to the AR6 of the IPCC. | https://report.ipcc.ch/ar6/wg1/IPCC_AR6_WGI_FullReport.pdf | 2409 | 2021
+    IPCC | Technical Summary. In: Climate Change 2021: The Physical Science Basis. Contribution of the WGI to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg1/downloads/report/IPCC_AR6_WGI_TS.pdf | 112 | 2021
+    IPCC | Summary for Policymakers. In: Climate Change 2022: Impacts, Adaptation and Vulnerability. Contribution of the WGII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg2/downloads/report/IPCC_AR6_WGII_SummaryForPolicymakers.pdf | 34 | 2022
+    IPCC | Technical Summary. In: Climate Change 2022: Impacts, Adaptation and Vulnerability. Contribution of the WGII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg2/downloads/report/IPCC_AR6_WGII_TechnicalSummary.pdf | 84 | 2022
+    IPCC | Full Report. In: Climate Change 2022: Impacts, Adaptation and Vulnerability. Contribution of the WGII to the AR6 of the IPCC. | https://report.ipcc.ch/ar6/wg2/IPCC_AR6_WGII_FullReport.pdf | 3068 | 2022
+    IPCC | Summary for Policymakers. In: Climate Change 2022: Mitigation of Climate Change. Contribution of the WGIII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg3/downloads/report/IPCC_AR6_WGIII_SummaryForPolicymakers.pdf | 50 | 2022
+    IPCC | Technical Summary. In: Climate Change 2022: Mitigation of Climate Change. Contribution of the WGIII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg3/downloads/report/IPCC_AR6_WGIII_TechnicalSummary.pdf | 102 | 2022
+    IPCC | Full Report. In: Climate Change 2022: Mitigation of Climate Change. Contribution of the WGIII to the AR6 of the IPCC. | https://www.ipcc.ch/report/ar6/wg3/downloads/report/IPCC_AR6_WGIII_FullReport.pdf | 2258 | 2022
+    IPCC | Summary for Policymakers. In: Global Warming of 1.5°C. An IPCC Special Report on the impacts of global warming of 1.5°C above pre-industrial levels and related global greenhouse gas emission pathways, in the context of strengthening the global response to the threat of climate change, sustainable development, and efforts to eradicate poverty. | https://www.ipcc.ch/site/assets/uploads/sites/2/2022/06/SPM_version_report_LR.pdf | 24 | 2018
+    IPCC | Summary for Policymakers. In: Climate Change and Land: an IPCC special report on climate change, desertification, land degradation, sustainable land management, food security, and greenhouse gas fluxes in terrestrial ecosystems. | https://www.ipcc.ch/site/assets/uploads/sites/4/2022/11/SRCCL_SPM.pdf | 36 | 2019
+    IPCC | Summary for Policymakers. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/01_SROCC_SPM_FINAL.pdf | 36 | 2019
+    IPCC | Technical Summary. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/02_SROCC_TS_FINAL.pdf | 34 | 2019
+    IPCC | Chapter 1 - Framing and Context of the Report. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/03_SROCC_Ch01_FINAL.pdf | 60 | 2019
+    IPCC | Chapter 2 - High Mountain Areas. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/04_SROCC_Ch02_FINAL.pdf | 72 | 2019
+    IPCC | Chapter 3 - Polar Regions. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/05_SROCC_Ch03_FINAL.pdf | 118 | 2019
+    IPCC | Chapter 4 - Sea Level Rise and Implications for Low-Lying Islands, Coasts and Communities. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/06_SROCC_Ch04_FINAL.pdf | 126 | 2019
+    IPCC | Chapter 5 -  Changing Ocean, Marine Ecosystems, and Dependent Communities. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/07_SROCC_Ch05_FINAL.pdf | 142 | 2019
+    IPCC | Chapter 6 - Extremes, Abrupt Changes and Managing Risk. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/08_SROCC_Ch06_FINAL.pdf | 68 | 2019
+    IPCC | Cross-Chapter Box 9: Integrative Cross-Chapter Box on Low-Lying Islands and Coasts. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2019/11/11_SROCC_CCB9-LLIC_FINAL.pdf | 18 | 2019
+    IPCC | Annex I: Glossary [Weyer, N.M. (ed.)]. In: IPCC Special Report on the Ocean and Cryosphere in a Changing Climate. | https://www.ipcc.ch/site/assets/uploads/sites/3/2022/03/10_SROCC_AnnexI-Glossary_FINAL.pdf | 28 | 2019
+    IPBES | Full Report. Global assessment report on biodiversity and ecosystem services of the IPBES. | https://zenodo.org/record/6417333/files/202206_IPBES%20GLOBAL%20REPORT_FULL_DIGITAL_MARCH%202022.pdf | 1148 | 2019
+    IPBES | Summary for Policymakers. Global assessment report on biodiversity and ecosystem services of the IPBES (Version 1). | https://zenodo.org/record/3553579/files/ipbes_global_assessment_report_summary_for_policymakers.pdf | 60 | 2019
+    IPBES | Full Report. Thematic assessment of the sustainable use of wild species of the IPBES. | https://zenodo.org/record/7755805/files/IPBES_ASSESSMENT_SUWS_FULL_REPORT.pdf | 1008 | 2022
+    IPBES | Summary for Policymakers. Summary for policymakers of the thematic assessment of the sustainable use of wild species of the IPBES. | https://zenodo.org/record/7411847/files/EN_SPM_SUSTAINABLE%20USE%20OF%20WILD%20SPECIES.pdf | 44 | 2022
+    IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for Africa. | https://zenodo.org/record/3236178/files/ipbes_assessment_report_africa_EN.pdf | 494 | 2018
+    IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for Africa. | https://zenodo.org/record/3236189/files/ipbes_assessment_spm_africa_EN.pdf | 52 | 2018
+    IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for the Americas. | https://zenodo.org/record/3236253/files/ipbes_assessment_report_americas_EN.pdf | 660 | 2018
+    IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for the Americas. | https://zenodo.org/record/3236292/files/ipbes_assessment_spm_americas_EN.pdf | 44 | 2018
+    IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for Asia and the Pacific. | https://zenodo.org/record/3237374/files/ipbes_assessment_report_ap_EN.pdf | 616 | 2018
+    IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for Asia and the Pacific. | https://zenodo.org/record/3237383/files/ipbes_assessment_spm_ap_EN.pdf | 44 | 2018
+    IPBES | Full Report. Regional Assessment Report on Biodiversity and Ecosystem Services for Europe and Central Asia. | https://zenodo.org/record/3237429/files/ipbes_assessment_report_eca_EN.pdf | 894 | 2018
+    IPBES | Summary for Policymakers. Regional Assessment Report on Biodiversity and Ecosystem Services for Europe and Central Asia. | https://zenodo.org/record/3237468/files/ipbes_assessment_spm_eca_EN.pdf | 52 | 2018
+    IPBES | Full Report. Assessment Report on Land Degradation and Restoration. | https://zenodo.org/record/3237393/files/ipbes_assessment_report_ldra_EN.pdf | 748 | 2018
+    IPBES | Summary for Policymakers. Assessment Report on Land Degradation and Restoration. | https://zenodo.org/record/3237393/files/ipbes_assessment_report_ldra_EN.pdf | 48 | 2018
+""")
+
+    with gr.Tab("🛢️ Carbon Footprint"):
+        gr.Markdown("""
 
 Carbon emissions were measured during the development and inference process using CodeCarbon [https://github.com/mlco2/codecarbon](https://github.com/mlco2/codecarbon)
 
@@ -531,24 +586,10 @@ Carbon emissions were measured during the development and inference process usin
 
 Carbon Emissions are **relatively low but not negligible** compared to other usages: one question asked to ClimateQ&A is around 0.482gCO2e - equivalent to 2.2m by car (https://datagir.ademe.fr/apps/impact-co2/)  
 Or around 2 to 4 times more than a typical Google search. 
-
-## 📧 Contact 
-This tool has been developed by the R&D lab at **Ekimetrics** (Jean Lelong, Nina Achache, Gabriel Olympie, Nicolas Chesneau, Natalia De la Calzada, Théo Alves Da Costa)
-
-If you have any questions or feature requests, please feel free to reach us out at <b>[email protected]</b>.
-
-## 💻 Developers
-For developers, the methodology used is detailed below : 
-
-- Extract individual paragraphs from scientific reports (e.g., IPCC, IPBES) using OCR techniques and open sources algorithms
-- Use Haystack to compute semantically representative embeddings for each paragraph using a sentence transformers model (https://huggingface.co/sentence-transformers/multi-qa-mpnet-base-dot-v1). 
-- Store all the embeddings in a FAISS Flat index. 
-- Reformulate each user query to be as specific as possible and compute its embedding. 
-- Retrieve up to 10 semantically closest paragraphs (using dot product similarity) from all available scientific reports. 
-- Provide these paragraphs as context for GPT-Turbo's answer in a system message. 
 """
     )
 
     demo.queue(concurrency_count=16)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()

climateqa/chains.py

@@ -0,0 +1,68 @@
+# https://python.langchain.com/docs/modules/chains/how_to/custom_chain
+# Including reformulation of the question in the chain
+import json
+
+from langchain import PromptTemplate, LLMChain
+from langchain.chains import RetrievalQAWithSourcesChain
+from langchain.chains import TransformChain, SequentialChain
+from langchain.chains.qa_with_sources import load_qa_with_sources_chain
+
+from climateqa.prompts import answer_prompt, reformulation_prompt,audience_prompts
+
+
+def load_reformulation_chain(llm):
+
+    prompt = PromptTemplate(
+        template = reformulation_prompt,
+        input_variables=["query"],
+    )
+    reformulation_chain = LLMChain(llm = llm,prompt = prompt,output_key="json")
+
+    # Parse the output
+    def parse_output(output):
+        query = output["query"]
+        json_output = json.loads(output["json"])
+        question = json_output.get("question", query)
+        language = json_output.get("language", "English")
+        return {
+            "question": question,
+            "language": language,
+        }
+    
+    transform_chain = TransformChain(
+        input_variables=["json"], output_variables=["question","language"], transform=parse_output
+    )
+
+    reformulation_chain = SequentialChain(chains = [reformulation_chain,transform_chain],input_variables=["query"],output_variables=["question","language"])
+    return reformulation_chain
+
+
+
+def load_answer_chain(retriever,llm):
+    prompt = PromptTemplate(template=answer_prompt, input_variables=["summaries", "question","audience","language"])
+    qa_chain = load_qa_with_sources_chain(llm, chain_type="stuff",prompt = prompt)
+
+    # This could be improved by providing a document prompt to avoid modifying page_content in the docs
+    # See here https://github.com/langchain-ai/langchain/issues/3523
+
+    answer_chain = RetrievalQAWithSourcesChain(
+        combine_documents_chain = qa_chain,
+        retriever=retriever,
+        return_source_documents = True,
+    )
+    return answer_chain
+
+
+def load_climateqa_chain(retriever,llm):
+
+    reformulation_chain = load_reformulation_chain(llm)
+    answer_chain = load_answer_chain(retriever,llm)
+
+    climateqa_chain = SequentialChain(
+        chains = [reformulation_chain,answer_chain],
+        input_variables=["query","audience"],
+        output_variables=["answer","question","language","source_documents"],
+        return_all = True,
+    )
+    return climateqa_chain
+

climateqa/chat.py

@@ -0,0 +1,39 @@
+# LANGCHAIN IMPORTS
+from langchain import PromptTemplate, LLMChain
+from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.chains import RetrievalQAWithSourcesChain
+from langchain.chains.qa_with_sources import load_qa_with_sources_chain
+
+
+# CLIMATEQA
+from climateqa.retriever import ClimateQARetriever
+from climateqa.vectorstore import get_pinecone_vectorstore
+from climateqa.chains import load_climateqa_chain
+
+
+class ClimateQA:
+    def __init__(self,hf_embedding_model = "sentence-transformers/multi-qa-mpnet-base-dot-v1",
+                 show_progress_bar = False,batch_size = 1,max_tokens = 1024,**kwargs):
+    
+        self.llm = self.get_llm(max_tokens = max_tokens,**kwargs)
+        self.embeddings_function = HuggingFaceEmbeddings(
+            model_name=hf_embedding_model,
+            encode_kwargs={"show_progress_bar":show_progress_bar,"batch_size":batch_size}
+        )
+
+
+
+    def get_vectorstore(self):
+        pass
+
+
+    def reformulate(self):
+        pass
+
+
+    def retrieve(self):
+        pass
+
+
+    def ask(self):
+        pass

climateqa/llm.py

@@ -0,0 +1,23 @@
+from langchain.chat_models import AzureChatOpenAI
+
+# LOAD ENVIRONMENT VARIABLES
+from dotenv import load_dotenv
+import os
+load_dotenv()
+
+
+def get_llm(max_tokens = 1024,temperature = 0.0,verbose = True,streaming = False, **kwargs):
+
+    llm = AzureChatOpenAI(
+        openai_api_base=os.environ["AZURE_OPENAI_API_BASE_URL"],
+        openai_api_version=os.environ["AZURE_OPENAI_API_VERSION"],
+        deployment_name=os.environ["AZURE_OPENAI_API_DEPLOYMENT_NAME"],
+        openai_api_key=os.environ["AZURE_OPENAI_API_KEY"],
+        openai_api_type = "azure",
+        max_tokens = max_tokens,
+        temperature = temperature,
+        verbose = verbose,
+        streaming = streaming,
+        **kwargs,
+    )
+    return llm

climateqa/prompts.py

@@ -0,0 +1,57 @@
+
+# If the message is not relevant to climate change (like "How are you", "I am 18 years old" or "When was built the eiffel tower"), return N/A
+
+reformulation_prompt = """
+Reformulate the following user message to be a short standalone question in English, in the context of an educational discussion about climate change.
+---
+query: La technologie nous sauvera-t-elle ?
+question: Can technology help humanity mitigate the effects of climate change?
+language: French
+---
+query: what are our reserves in fossil fuel?
+question: What are the current reserves of fossil fuels and how long will they last?
+language: English
+---
+query: what are the main causes of climate change?
+question: What are the main causes of climate change in the last century?
+language: English
+---
+
+Output the result as json with two keys "question" and "language"
+query: {query}
+answer:"""
+
+system_prompt = """
+You are ClimateQ&A, an AI Assistant created by Ekimetrics, you will act as a climate scientist and answer questions about climate change and biodiversity. 
+You are given a question and extracted passages of the IPCC and/or IPBES reports. Provide a clear and structured answer based on the passages provided, the context and the guidelines.
+"""
+
+
+answer_prompt = """
+You are ClimateQ&A, an AI Assistant created by Ekimetrics. You are given a question and extracted passages of the IPCC and/or IPBES reports. Provide a clear and structured answer based on the passages provided, the context and the guidelines.
+
+Guidelines:
+- If the passages have useful facts or numbers, use them in your answer.
+- When you use information from a passage, mention where it came from by using [Doc i] at the end of the sentence. i stands for the number of the document.
+- Do not use the sentence 'Doc i says ...' to say where information came from.
+- If the same thing is said in more than one document, you can mention all of them like this: [Doc i, Doc j, Doc k]
+- Do not just summarize each passage one by one. Group your summaries to highlight the key parts in the explanation.
+- If it makes sense, use bullet points and lists to make your answers easier to understand.
+- You do not need to use every passage. Only use the ones that help answer the question.
+- If the documents do not have the information needed to answer the question, just say you do not have enough information.
+
+-----------------------
+Passages:
+{summaries}
+
+-----------------------
+Question: {question} - Explained to {audience}
+Answer in {language} with the passages citations:
+"""
+
+
+audience_prompts = {
+    "children": "6 year old children that don't know anything about science and climate change and need metaphors to learn",
+    "general": "the general public who know the basics in science and climate change and want to learn more about it without technical terms. Still use references to passages.",
+    "experts": "expert and climate scientists that are not afraid of technical terms",
+}

climateqa/retriever.py

@@ -0,0 +1,149 @@
+# https://github.com/langchain-ai/langchain/issues/8623
+
+import pandas as pd
+
+from langchain.schema.retriever import BaseRetriever, Document
+from langchain.vectorstores.base import VectorStoreRetriever
+from langchain.vectorstores import VectorStore
+from langchain.callbacks.manager import CallbackManagerForRetrieverRun
+from typing import List
+from pydantic import Field
+
+class ClimateQARetriever(BaseRetriever):
+    vectorstore:VectorStore
+    sources:list = ["IPCC","IPBES"]
+    threshold:float = 22
+    k_summary:int = 3
+    k_total:int = 10
+    namespace:str = "vectors"
+
+    def get_relevant_documents(self, query: str) -> List[Document]:
+
+        # Check if all elements in the list are either IPCC or IPBES
+        assert isinstance(self.sources,list)
+        assert all([x in ["IPCC","IPBES"] for x in self.sources])
+        assert self.k_total > self.k_summary, "k_total should be greater than k_summary"
+
+        # Prepare base search kwargs
+        filters = {
+            "source": { "$in":self.sources},
+        }
+
+        # Search for k_summary documents in the summaries dataset
+        filters_summaries = {
+            **filters,
+            "report_type": { "$in":["SPM","TS"]},
+        }
+        docs_summaries = self.vectorstore.similarity_search_with_score(query=query,namespace = self.namespace,filter = filters_summaries,k = self.k_summary)
+        docs_summaries = [x for x in docs_summaries if x[1] > self.threshold]
+
+        # Search for k_total - k_summary documents in the full reports dataset
+        filters_full = {
+            **filters,
+            "report_type": { "$nin":["SPM","TS"]},
+        }
+        k_full = self.k_total - len(docs_summaries)
+        docs_full = self.vectorstore.similarity_search_with_score(query=query,namespace = self.namespace,filter = filters_full,k = k_full)
+
+        # Concatenate documents
+        docs = docs_summaries + docs_full
+
+        # Filter if scores are below threshold
+        docs = [x for x in docs if x[1] > self.threshold]
+
+        # Add score to metadata
+        results = []
+        for i,(doc,score) in enumerate(docs):
+            doc.metadata["similarity_score"] = score
+            doc.metadata["content"] = doc.page_content
+            doc.metadata["page_number"] = int(doc.metadata["page_number"])
+            doc.page_content = f"""Doc {i+1} - {doc.metadata['short_name']}: {doc.page_content}"""
+            results.append(doc)
+
+        return results
+
+
+
+
+
+# def filter_summaries(df,k_summary = 3,k_total = 10):
+#     # assert source in ["IPCC","IPBES","ALL"], "source arg should be in (IPCC,IPBES,ALL)"
+
+#     # # Filter by source
+#     # if source == "IPCC":
+#     #     df = df.loc[df["source"]=="IPCC"]
+#     # elif source == "IPBES":
+#     #     df = df.loc[df["source"]=="IPBES"]
+#     # else:
+#     #     pass
+
+#     # Separate summaries and full reports
+#     df_summaries = df.loc[df["report_type"].isin(["SPM","TS"])]
+#     df_full = df.loc[~df["report_type"].isin(["SPM","TS"])]
+
+#     # Find passages from summaries dataset
+#     passages_summaries = df_summaries.head(k_summary)
+
+#     # Find passages from full reports dataset
+#     passages_fullreports = df_full.head(k_total - len(passages_summaries))
+
+#     # Concatenate passages
+#     passages = pd.concat([passages_summaries,passages_fullreports],axis = 0,ignore_index = True)
+#     return passages
+
+
+
+
+# def retrieve_with_summaries(query,retriever,k_summary = 3,k_total = 10,sources = ["IPCC","IPBES"],max_k = 100,threshold = 0.555,as_dict = True,min_length = 300):
+#     assert max_k > k_total
+
+#     validated_sources = ["IPCC","IPBES"]
+#     sources = [x for x in sources if x in validated_sources]
+#     filters = {
+#         "source": { "$in": sources },
+#     }
+#     print(filters)
+
+#     # Retrieve documents
+#     docs = retriever.retrieve(query,top_k = max_k,filters = filters)
+
+#     # Filter by score
+#     docs = [{**x.meta,"score":x.score,"content":x.content} for x in docs if x.score > threshold]
+
+#     if len(docs) == 0:
+#         return []
+#     res = pd.DataFrame(docs)
+#     passages_df = filter_summaries(res,k_summary,k_total)
+#     if as_dict:
+#         contents = passages_df["content"].tolist()
+#         meta = passages_df.drop(columns = ["content"]).to_dict(orient = "records")
+#         passages = []
+#         for i in range(len(contents)):
+#             passages.append({"content":contents[i],"meta":meta[i]})
+#         return passages
+#     else:
+#         return passages_df
+
+
+
+# def retrieve(query,sources = ["IPCC"],threshold = 0.555,k = 10):
+
+
+#     print("hellooooo")
+
+#     # Reformulate queries
+#     reformulated_query,language = reformulate(query)
+
+#     print(reformulated_query)
+
+#     # Retrieve documents
+#     passages = retrieve_with_summaries(reformulated_query,retriever,k_total = k,k_summary = 3,as_dict = True,sources = sources,threshold = threshold)
+#     response = {
+#       "query":query,
+#       "reformulated_query":reformulated_query,
+#       "language":language,
+#       "sources":passages,
+#       "prompts":{"init_prompt":init_prompt,"sources_prompt":sources_prompt},
+#     }
+#     return response
+

climateqa/vectorstore.py

@@ -0,0 +1,42 @@
+# Pinecone
+# More info at https://docs.pinecone.io/docs/langchain
+# And https://python.langchain.com/docs/integrations/vectorstores/pinecone
+
+import pinecone
+from langchain.vectorstores import Pinecone
+
+# LOAD ENVIRONMENT VARIABLES
+from dotenv import load_dotenv
+import os
+load_dotenv()
+
+
+def get_pinecone_vectorstore(embeddings,text_key = "content"):
+
+    # initialize pinecone
+    pinecone.init(
+        api_key=os.getenv("PINECONE_API_KEY"),  # find at app.pinecone.io
+        environment=os.getenv("PINECONE_API_ENVIRONMENT"),  # next to api key in console
+    )
+
+    index_name = os.getenv("PINECONE_API_INDEX")
+    vectorstore = Pinecone.from_existing_index(index_name, embeddings,text_key = text_key)
+    return vectorstore
+
+
+# def get_pinecone_retriever(vectorstore,k = 10,namespace = "vectors",sources = ["IPBES","IPCC"]):
+
+#     assert isinstance(sources,list)
+
+#     # Check if all elements in the list are either IPCC or IPBES
+#     filter = {
+#         "source": { "$in":sources},
+#     }
+
+#     retriever = vectorstore.as_retriever(search_kwargs={
+#         "k": k,
+#         "namespace":"vectors",
+#         "filter":filter
+#     })
+
+#     return retriever

climateqa_v3.db

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:6c05065e5070dd68d05547fc4f330ced5bcc30762d4f566a3fbcfd70d950063f
-size 459763712

climateqa_v3.faiss

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2e3190d405db879ccfcd913d0f752fbbde35243ec439f684eed2654484550ab2
-size 227269677

climateqa_v3.json

@@ -1 +0,0 @@
-{"sql_url": "sqlite:///climateqa_v3.db", "faiss_index_factory_str": "Flat", "index": "climateqa_v3", "similarity": "dot_product", "embedding_dim": 768}

requirements.txt

@@ -1,6 +1,4 @@
-faiss-cpu==1.7.2
-farm-haystack==1.14.0
-gradio==3.22.1
+gradio==3.47.1
 openai==0.27.0
 azure-storage-file-share==12.11.1
 python-dotenv==1.0.0

style.css

@@ -1,7 +1,7 @@
 
-:root {
+/* :root {
     --user-image: url('https://ih1.redbubble.net/image.4776899543.6215/st,small,507x507-pad,600x600,f8f8f8.jpg');
-  }
+  } */
 
 .warning-box {
     background-color: #fff3cd;
@@ -147,7 +147,7 @@ label > span{
 }
 
 /* Pseudo-element for the circularly cropped picture */
-.message.bot::before {
+/* .message.bot::before {
     content: '';
     position: absolute;
     top: -10px;
@@ -160,5 +160,8 @@ label > span{
     border-radius: 50%;
     z-index: 10;
   }
-  
-  
+   */
+   
+label.selected{
+  background:none !important;
+}