rag edit

appStore/rag.py

@@ -10,13 +10,26 @@ from tenacity import retry, stop_after_attempt, wait_random_exponential
 
 # Get openai API key
 openai.api_key = os.environ["OPENAI_API_KEY"]
-model_select = "gpt-3.5-turbo-1106"
+model_select = "gpt-3.5-turbo-0125"
 
 
 # define a special function for putting the prompt together (as we can't use haystack)
+# def get_prompt(context, label):
+#   base_prompt="Summarize the following context efficiently in bullet points, the less the better - but keep concrete goals. \
+#   Summarize only activities that address the vulnerability of "+label+" to climate change. \
+#   Formatting example: \
+#     - Bullet point 1 \
+#     - Bullet point 2 \
+# "
+
+#   # Add the meta data for references
+#   # context = ' - '.join([d.content for d in docs])
+#   prompt = base_prompt+"; Context: "+context+"; Answer:"
+  
+#   return prompt
+
 def get_prompt(context, label):
-  base_prompt="Summarize the following context efficiently in bullet points, the less the better- but keep concrete goals. \
-  Summarize only activities that address the vulnerability of "+label+" to climate change. \
+  base_prompt="Summarize the following context efficiently in bullet points, the less the better. \
   Formatting example: \
     - Bullet point 1 \
     - Bullet point 2 \

appStore/target.py

@@ -92,10 +92,15 @@ def target_display():
     ### RAG Output by group ##
 
     # Expand the DataFrame
-    df_expand = df.explode('Vulnerability Label')
+    df_expand = (
+        df.query("`Specific action/target/measure mentioned` == 'YES'")
+        .explode('Vulnerability Label')
+        )
     # Group by 'Vulnerability Label' and concatenate 'text'
     df_agg = df_expand.groupby('Vulnerability Label')['text'].agg('; '.join).reset_index()
 
+    # st.write(df_agg)
+
     st.markdown("----")
     st.markdown('**DOCUMENT FINDINGS SUMMARY BY VULNERABILITY LABEL:**')