Update other_function.py

other_function.py

@@ -212,11 +212,42 @@ def respond_pdf(query):
     for chunk in AI71(AI71_API_KEY).chat.completions.create(
         model="tiiuae/falcon-11b",
         messages=[
-            {"role": "system", "content": "You are a pdf answering assistant."},
+            {"role": "system", "content": "You are a agriculture pdf answering assistant."},
             {"role": "user", "content": f"Content:{extracted_text},Query:{query}"},
         ],
         stream=True,
     ):
         if chunk.choices[0].delta.content:
             res += chunk.choices[0].delta.content
+    return ( res.replace("User:",'').strip())
+
+
+def booktask(data):
+    res = ''
+    for chunk in AI71(AI71_API_KEY).chat.completions.create(
+        model="tiiuae/falcon-11b",
+        messages=[
+            {"role": "system", "content": "You are a agriculture bookkeeping assistant."},
+            {"role": "user", "content": f"My bookkeeping data is {data}.Provide the data in points."},
+        ],
+        stream=True,
+    ):
+        if chunk.choices[0].delta.content:
+            res += chunk.choices[0].delta.content
+    return ( res.replace("User:",'').strip())
+
+
+
+def return_bookdata(querry,data):
+    res = ''
+    for chunk in AI71(AI71_API_KEY).chat.completions.create(
+        model="tiiuae/falcon-11b",
+        messages=[
+            {"role": "system", "content": "You are a agriculture bookkeeping responding assistant."},
+            {"role": "user", "content": f"My bookkeeping data is {data}.do what user is telling. user:{querry"},
+        ],
+        stream=True,
+    ):
+        if chunk.choices[0].delta.content:
+            res += chunk.choices[0].delta.content
     return ( res.replace("User:",'').strip())