Update app.py

app.py

@@ -325,34 +325,16 @@ def whatsapp_webhook():
 
     if num_media > 0:
         media_url = request.values.get('MediaUrl0')
-        response_text = media_url
         content_type = request.values.get('MediaContentType0')
+
         if content_type.startswith('image/'):
+            # Handle image processing (disease/pest detection)
             filepath = convert_img(media_url, account_sid, auth_token)
-            try:
-                disease = predict_disease(filepath)
-            except:
-                disease = None
-            try:
-                pest = predict_pest(filepath)
-            except:
-                pest = None
-
-            if disease:
-                response_text = f"Detected disease: {disease}"
-                disease_info = generate_response(f"Provide brief information about {disease} in plants", chat_history)
-                response_text += f"\n\nAdditional information: {disease_info}"
-            elif pest:
-                response_text = f"Detected pest: {pest}"
-                pest_info = generate_response(f"Provide brief information about {pest} in agriculture", chat_history)
-                response_text += f"\n\nAdditional information: {pest_info}"
-            else:
-                response_text = "Please upload another image with good quality."
-
+            response_text = handle_image(filepath)
         else:
+            # Handle PDF processing
             filepath = download_and_save_as_txt(media_url, account_sid, auth_token)
-            response_text = query_rag(filepath)
-    
+            response_text = process_and_query_pdf(filepath)
     elif ('weather' in incoming_msg.lower()) or ('climate' in incoming_msg.lower()) or (
             'temperature' in incoming_msg.lower()):
         response_text = get_weather(incoming_msg.lower())
@@ -366,16 +348,40 @@ def whatsapp_webhook():
         news = get_news()
         response_text = generate_response(incoming_msg + ' data is ' + str(news), chat_history)
     else:
-        # Check if the query is related to a PDF document
-        if 'from pdf' in incoming_msg or 'in pdf' in incoming_msg:
-            response_text = query_rag(incoming_msg)
-        else:
-            response_text = generate_response(incoming_msg, chat_history)
+        response_text = query_rag(incoming_msg)
 
     conversation_memory.add_to_memory({"user": incoming_msg, "assistant": response_text})
     send_message(sender, response_text)
     return '', 204
 
+def handle_image(filepath):
+    try:
+        disease = predict_disease(filepath)
+    except:
+        disease = None
+    try:
+        pest = predict_pest(filepath)
+    except:
+        pest = None
+
+    if disease:
+        response_text = f"Detected disease: {disease}"
+        disease_info = generate_response(f"Provide brief information about {disease} in plants", chat_history)
+        response_text += f"\n\nAdditional information: {disease_info}"
+    elif pest:
+        response_text = f"Detected pest: {pest}"
+        pest_info = generate_response(f"Provide brief information about {pest} in agriculture", chat_history)
+        response_text += f"\n\nAdditional information: {pest_info}"
+    else:
+        response_text = "Please upload another image with good quality."
+    
+    return response_text
+
+def process_and_query_pdf(filepath):
+    # Assuming the PDF processing and embedding are handled here.
+    add_to_chroma(load_documents())
+    return query_rag("from pdf")  # Replace with a more specific query if needed
+
 
 def send_message(to, body):
     try: