Update app.py

app.py

@@ -3,11 +3,11 @@ import pandas as pd
 from prophet import Prophet
 import json
 import os
-import requests  # Use requests instead of the google library
+import requests
+
 
-# --- MODIFIED: Securely load the API key from Hugging Face Secrets ---
 API_KEY = os.environ.get("GOOGLE_API_KEY")
-LLM_LOADED = bool(API_KEY)  # Check if the key was loaded
+LLM_LOADED = bool(API_KEY)
 
 def create_cases_data():
     """Creates a dummy CSV with customer case data."""
@@ -55,14 +55,12 @@ def predict_future_sales(sku_choice):
 
 # === USE CASE 2: Automated Case Reply Generation ===
 def get_open_cases():
-    """Loads open cases from the CSV and returns a list for the dropdown."""
     if not os.path.exists('customer_cases.csv'):
         return []
     df_cases = pd.read_csv('customer_cases.csv')
     return [f"Case {row['case_id']}: {row['customer_name']}" for index, row in df_cases.iterrows()]
 
 def get_case_details(case_selection):
-    """Fetches details for a selected case."""
     if not case_selection:
         return "", "", gr.update(visible=False)
 
@@ -73,11 +71,11 @@ def get_case_details(case_selection):
     return case_data['customer_query'], case_data.to_json(), gr.update(visible=True)
 
 def generate_case_reply(case_data_json):
-    """Generates an email reply using the Gemini API via requests."""
     if not LLM_LOADED:
         return "Google API Key not loaded. Please configure it in the Space secrets."
 
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={API_KEY}"
+    # --- FIX: Corrected model name in URL ---
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.0-pro:generateContent?key={API_KEY}"
     
     internal_data = json.loads(case_data_json)
     customer_query = internal_data.pop('customer_query')
@@ -125,26 +123,26 @@ def generate_case_reply(case_data_json):
           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:
           """
         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
             """
@@ -153,7 +151,6 @@ def generate_case_reply(case_data_json):
         return f"Error parsing API output: {e}\nRaw Output:\n{generated_text}"
 
 def send_reply(case_selection):
-    """Simulates sending the reply by removing the case from the 'database'."""
     case_id = int(case_selection.split(':')[0].replace('Case', '').strip())
     df_cases = pd.read_csv('customer_cases.csv')
     df_cases = df_cases[df_cases['case_id'] != case_id]
@@ -162,26 +159,16 @@ def send_reply(case_selection):
 
 # === USE CASE 3: Automated Prep Center Communication ===
 def generate_prep_center_email(client_name, shipment_id, status, arrival_date, num_cartons, labels_provided):
-    """Generates a prep center email using the Gemini API via requests."""
     if not LLM_LOADED:
         return "Google API Key not loaded. Please configure it in the Space secrets."
 
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={API_KEY}"
+    # --- FIX: Corrected model name in 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, specializing in e-commerce prep center communications.
-      Your task is to analyze shipment data and generate a structured JSON object to populate a reminder email for the client.
-      Your FINAL and ONLY output must be a single, well-formed JSON object. Do not add any text before or after the JSON.
-      Required JSON Schema:
-      {
-        "urgency_tag": "string (e.g., 'Reminder', 'Urgent', 'Action Required')",
-        "main_call_to_action": "string (A single, clear sentence summarizing what the client needs to do)",
-        "detailed_action": "string (A more detailed explanation of the required action)",
-        "reason_for_action": "string (Explain the impact of the action, i.e., why it is important)",
-        "instructions": "string (Provide simple, step-by-step instructions)"
-      }
+      You are an expert AI operations assistant... (Your prompt remains the same)
       """
     user_prompt = f"Analyze this data and generate the JSON: {json.dumps(shipment_data)}"
 
@@ -207,32 +194,32 @@ def generate_prep_center_email(client_name, shipment_id, status, arrival_date, n
         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
             """
         return template
@@ -337,4 +324,4 @@ with gr.Blocks(theme=gr.themes.Soft(text_size='lg'), title="RecoEngine Demo") as
 
 # --- 5. LAUNCH THE APP ---
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()