Update app.py

app.py

@@ -5,7 +5,7 @@ import json
 import os
 import requests
 
-
+# Securely load the API key from Hugging Face Secrets
 API_KEY = os.environ.get("GOOGLE_API_KEY")
 LLM_LOADED = bool(API_KEY)
 
@@ -74,7 +74,7 @@ def generate_case_reply(case_data_json):
     if not LLM_LOADED:
         return "Google API Key not loaded. Please configure it in the Space secrets."
 
-    # --- FIX: Corrected model name in URL ---
+    # --- THIS IS THE CORRECTED URL ---
     url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.0-pro:generateContent?key={API_KEY}"
     
     internal_data = json.loads(case_data_json)
@@ -82,25 +82,11 @@ def generate_case_reply(case_data_json):
     internal_data.pop('case_id')
 
     system_prompt = """
-        You are an expert AI assistant for "RecoEngine", specializing in Amazon customer support.
-        Your task is to analyze a customer's query and associated internal data to generate a structured JSON object.
-        This JSON object will be used to populate a standard reply template.
-        Your FINAL and ONLY output must be a single, well-formed JSON object. Do not add any text before or after the JSON.
-        Here is the required JSON schema:
-        {
-          "customer_name": "string",
-          "paraphrased_issue": "string (A concise summary of the customer's problem)",
-          "empathy_adjective": "string (e.g., 'frustrating', 'disappointing', 'concerning')",
-          "facts_statement": "string (A statement of fact based on internal data, e.g., 'Your order is delayed in transit.')",
-          "action_taken": "string (The resolution step that was performed, e.g., 'I have issued a full refund of ₹XXX.')",
-          "next_steps": ["an array of strings detailing what the customer should do or expect next"]
-        }
+        You are an expert AI assistant for "RecoEngine", specializing in Amazon customer support... (Your prompt remains the same)
         """
     user_prompt = f"### Customer Query ###\n{customer_query}\n### Internal Data ###\n{json.dumps(internal_data)}"
 
-    payload = {
-        "contents": [{"parts": [{"text": f"{system_prompt}\n{user_prompt}"}]}]
-    }
+    payload = { "contents": [{"parts": [{"text": f"{system_prompt}\n{user_prompt}"}]}] }
     headers = {'Content-Type': 'application/json'}
 
     try:
@@ -120,32 +106,8 @@ def generate_case_reply(case_data_json):
         parsed_json = json.loads(json_output_str)
 
         template = f"""
-          Subject: Regarding your recent inquiry about your order
-
-          Dear {parsed_json.get('customer_name', 'Valued Customer')},
-          
-          Thank you for contacting Amazon Support.
-          
-          I understand that you are concerned about {parsed_json.get('paraphrased_issue', 'your recent issue')}. I know how {parsed_json.get('empathy_adjective', 'concerning')} this can be, and I'm here to help you with this.
-          
-          After reviewing the details, I can see that {parsed_json.get('facts_statement', 'we have reviewed your case.')}.
-          
-          To resolve this for you, {parsed_json.get('action_taken', 'we have taken the necessary action.')}.
-          
-          Here is a summary of the next steps:
+          Subject: Regarding your recent inquiry about your order... (Your email template remains the same)
           """
-        for i, step in enumerate(parsed_json.get('next_steps', []), 1):
-            template += f"{i}. {step}\n"
-        template += """
-            If there is anything else you need assistance with, please don't hesitate to reply to this email. We're always here to help.
-            
-            Thank you for your patience and understanding.
-            
-            Best regards,
-            
-            Anjali
-            Amazon.in Customer Service
-            """
         return template
     except Exception as e:
         return f"Error parsing API output: {e}\nRaw Output:\n{generated_text}"
@@ -162,19 +124,13 @@ def generate_prep_center_email(client_name, shipment_id, status, arrival_date, n
     if not LLM_LOADED:
         return "Google API Key not loaded. Please configure it in the Space secrets."
 
-    
+    # --- THIS IS THE CORRECTED URL ---
     url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.0-pro:generateContent?key={API_KEY}"
     
     shipment_data = {"client_name": client_name, "shipment_id": shipment_id, "status": status, "expected_arrival_date": arrival_date, "num_cartons": num_cartons, "labels_provided": labels_provided}
-    
-    system_prompt = """
-      You are an expert AI operations assistant... (Your prompt remains the same)
-      """
+    system_prompt = "You are an expert AI operations assistant..."
     user_prompt = f"Analyze this data and generate the JSON: {json.dumps(shipment_data)}"
-
-    payload = {
-        "contents": [{"parts": [{"text": f"{system_prompt}\n{user_prompt}"}]}]
-    }
+    payload = { "contents": [{"parts": [{"text": f"{system_prompt}\n{user_prompt}"}]}] }
     headers = {'Content-Type': 'application/json'}
 
     try:
@@ -193,34 +149,7 @@ def generate_prep_center_email(client_name, shipment_id, status, arrival_date, n
         json_output_str = generated_text[start_index:end_index]
         parsed_json = json.loads(json_output_str)
         template = f"""
-            Subject: {parsed_json.get('urgency_tag')} - Action Required for Your Shipment {shipment_id} - {client_name}
-            
-            Dear {client_name},
-            
-            This is an important reminder from PrepCenter India regarding your upcoming shipment.
-            
-            {parsed_json.get('main_call_to_action')}.
-            
-            **Shipment Details:**
-            * **Shipment ID:** {shipment_id}
-            * **Status:** {status}
-            * **Expected Arrival Date:** {arrival_date}
-            * **Number of Cartons:** {num_cartons}
-            
-            **Action Required:**
-            {parsed_json.get('detailed_action')}
-            
-            **Why is this important?**
-            {parsed_json.get('reason_for_action')}
-            
-            **How to complete this action:**
-            {parsed_json.get('instructions')}
-            
-            If you have already completed this action, please disregard this email. If you have any questions, please reply to this message.
-            
-            Best regards,
-            
-            The PrepCenter India Team
+            Subject: {parsed_json.get('urgency_tag')} - Action Required for Your Shipment... (Your email template remains the same)
             """
         return template
     except Exception as e: