Update main.py

main.py

@@ -1883,18 +1883,18 @@ def get_price_trends():
 def ai_chat():
     auth_header = request.headers.get("Authorization", "");
     uid = None
-    response_obj_gemini = None
-    intent = "unknown"
+    response_obj_gemini = None 
+    intent = "unknown"  
     try:
         if not FIREBASE_INITIALIZED or not gemini_client:
             return jsonify({'error': 'Server or AI service not ready.'}), 503
-
+        
         uid = verify_token(auth_header)
-
+        
         data = request.get_json()
         if not data:
             return jsonify({"error": "Invalid request data."}), 400
-
+            
         user_message = data.get("message", "").strip()
         if not user_message:
             return jsonify({"error": "Message cannot be empty."}), 400
@@ -1905,24 +1905,22 @@ def ai_chat():
         - "price_trend_query": User is asking about historical price trends, market analysis, or future price predictions. (e.g., "how are maize prices trending?", "what was the price of sorghum last month?")
         - "general_agri_info": User is asking for general agricultural advice, farming techniques, business tips, or information not tied to current platform data or specific historical trends. (e.g., "how to grow tomatoes?", "best fertilizer for sandy soils", "exporting crops in SADC")
         - "other": If the query doesn't fit well or is a simple greeting/chit-chat.
-
         User Query: "{user_message}"
-
         Return ONLY the category string (e.g., "platform_data_query").
         """
-
+        
         response_obj_gemini = gemini_client.generate_content( # Corrected: use generate_content directly
             model='gemini-2.0-flash',
             contents=[{'parts': [{'text': classify_prompt}]}] # Pass contents as a list
         )
-
+        
         intent = response_obj_gemini.text.strip().replace('"', '') if response_obj_gemini.text else "general_agri_info"
-
+        
         valid_intents = ["platform_data_query", "price_trend_query", "general_agri_info", "other"]
         if intent not in valid_intents:
             logger.warning(f"AI Chat: Unexpected intent: '{intent}' for query '{user_message}'. Defaulting to general_agri_info.")
             intent = "general_agri_info"
-
+        
         context_for_gemini = ""
         platform_data_summary = ""
         trend_analysis_summary = ""
@@ -1931,7 +1929,7 @@ def ai_chat():
             platform_data_summary, _ = _fetch_platform_data_for_chat(user_message)
             if platform_data_summary:
                 context_for_gemini += f"Current Platform Data Context (summarized for you):\n{platform_data_summary}\n\n"
-
+        
         elif intent == "price_trend_query":
             crop_match = re.search(r"(?:trend for|prices of|price of|trend of)\s+([\w\s]+?)(?:\s+in\s+([\w\s]+?))?(?:\s|$)", user_message.lower())
             trend_crop, trend_location = (None, None)
@@ -1939,37 +1937,34 @@ def ai_chat():
                 trend_crop = crop_match.group(1).strip()
                 if crop_match.group(2):
                     trend_location = crop_match.group(2).strip()
-
+            
             trend_analysis_summary, _ = _get_price_trend_analysis_for_chat(crop_type=trend_crop, location=trend_location)
             if trend_analysis_summary:
                 context_for_gemini += f"Price Trend Analysis Context (generated for you):\n{trend_analysis_summary}\n\n"
 
         main_prompt = f"""
-        You are Tunasonga Agri Assistant, an AI for an agricultural marketplace in Zimbabwe and SADC.
-        Your goal is to provide helpful, concise, and accurate information. Your persona is professional, friendly, and supportive of farmers and agri-businesses.
-
-        The user's original query intent was classified as: {intent}
-        {context_for_gemini}
-        Based on the user's query and any provided context above, please formulate your answer to the user.
-        User Query: "{user_message}"
-
-        Specific Instructions based on intent:
-
-        If the intent was 'platform_data_query': Use the "Current Platform Data Context" to answer. If the context says no items were found, relay that and suggest they browse the marketplace or refine their search. Do not invent listings.
-        If the intent was 'price_trend_query': Use the "Price Trend Analysis Context". If the context says trends couldn't be generated, relay that. Do not invent trends.
-        For 'general_agri_info': Use your broad agricultural knowledge. Focus on practices relevant to the SADC region, smallholder farmers, climate-smart agriculture, market access, and agri-business development. Provide actionable advice if possible.
-        If the query is unclear, classified as "other", or if the context is insufficient for a specific query: Provide a polite general response, ask for clarification, or gently guide the user on how you can help (e.g., "I can help with finding produce, getting price trends, or general farming advice. What would you like to know?").
-        Keep your answers clear and easy to understand. Avoid overly technical jargon unless necessary and explain it.
-        Answer:
-        """
-
+You are Tunasonga Agri Assistant, an AI for an agricultural marketplace in Zimbabwe and SADC.
+Your goal is to provide helpful, concise, and accurate information. Your persona is professional, friendly, and supportive of farmers and agri-businesses.
+The user's original query intent was classified as: {intent}
+{context_for_gemini}
+Based on the user's query and any provided context above, please formulate your answer to the user.
+User Query: "{user_message}"
+Specific Instructions based on intent:
+- If the intent was 'platform_data_query': Use the "Current Platform Data Context" to answer. If the context says no items were found, relay that and suggest they browse the marketplace or refine their search. Do not invent listings.
+- If the intent was 'price_trend_query': Use the "Price Trend Analysis Context". If the context says trends couldn't be generated, relay that. Do not invent trends.
+- For 'general_agri_info': Use your broad agricultural knowledge. Focus on practices relevant to the SADC region, smallholder farmers, climate-smart agriculture, market access, and agri-business development. Provide actionable advice if possible.
+- If the query is unclear, classified as "other", or if the context is insufficient for a specific query: Provide a polite general response, ask for clarification, or gently guide the user on how you can help (e.g., "I can help with finding produce, getting price trends, or general farming advice. What would you like to know?").
+Keep your answers clear and easy to understand. Avoid overly technical jargon unless necessary and explain it.
+Answer:
+"""
+        
         response_obj_gemini = gemini_client.generate_content( # Corrected
             model='gemini-2.0-flash',
             contents=[{'parts': [{'text': main_prompt}]}]
         )
-
+        
         ai_response_text = response_obj_gemini.text.strip() if response_obj_gemini.text else "I'm having trouble generating a response right now. Please try again."
-
+        
         try:
             db.reference(f'ai_chat_history/{uid}/{str(uuid.uuid4())}', app=db_app).set({
                 'user_message': user_message,
@@ -1979,14 +1974,14 @@ def ai_chat():
             })
         except Exception as chat_history_error:
             logger.error(f"Failed to store chat history for UID {uid}: {chat_history_error}")
-
+            
         return jsonify({"response": ai_response_text, "intent": intent})
 
-    except AttributeError as ae:
+    except AttributeError as ae:  
         logger.error(f"Gemini Response Attribute Error in ai_chat (UID: {uid}): {ae}. Response object: {response_obj_gemini}")
         ai_response_text = "I'm having a little trouble understanding that. Could you try rephrasing?"
         return jsonify({"response": ai_response_text, "intent": intent, "error_detail": "AI_RESPONSE_FORMAT_ISSUE"}), 200
-    except Exception as e:
+    except Exception as e:  
         return handle_route_errors(e, uid_context=uid)
 
 @app.route('/api/user/ai-chat-history', methods=['GET'])