Merge pull request #158 from eye-on-surveillance/AI/request-flow-fix

AI/request-flow-fix
eye-on-surveillance · Nov 15, 2023 · 37c1034 · 37c1034
2 parents 01d7932 + bd963df
commit 37c1034
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/packages/googlecloud/functions/getanswer/main.py b/packages/googlecloud/functions/getanswer/main.py
@@ -25,9 +25,18 @@
 
 
 def update_supabase(responses, citations, card_id):
+    transformed_citations = []
+    for citation in citations:
+        transformed_citations.append({
+            "source_title": citation.get("Title"),
+            "source_name": citation.get("Name"),
+            "source_publish_date": citation.get("Published"),
+            "source_url": citation.get("URL")
+        })
+
     try:
         supabase.table("cards").update(
-            {"responses": responses, "citations": citations}
+            {"responses": responses, "citations": transformed_citations}
         ).eq("id", card_id).execute()
         logging.info("Data successfully updated in Supabase")
     except Exception as e:

diff --git a/packages/web/components/NewQuery.tsx b/packages/web/components/NewQuery.tsx
@@ -61,6 +61,21 @@ export default function NewQuery() {
     }
   };
 
+  const fetchUpdatedCard = async (cardId: string) => {
+    const { data, error } = await supabase
+        .from('cards')
+        .select('*')
+        .eq('id', cardId)
+        .single();
+
+    if (error) {
+        console.error('Error fetching updated card:', error);
+    } else if (data) {
+        setCard(data);
+    }
+};
+
+
   const sendQueryToFunction = async (newCard: ICard) => {
     // Start processing question
     const answerResp = await fetch(apiEndpoint, {
@@ -91,6 +106,11 @@ export default function NewQuery() {
     card = card as ICard;
     setQuery("");
     setIsProcessing(false);
+    if (newCard.id) {
+      await fetchUpdatedCard(newCard.id);
+  } else {
+      console.error('Card ID is undefined');
+  }
   };
 
   useEffect(() => {