Update app.py

app.py

@@ -54,6 +54,7 @@ class UserProfile(Base):
     __tablename__ = "user_profiles"
     id = Column(Integer, primary_key=True, index=True)
     user_id = Column(String, unique=True, index=True)
+    phone_number = Column(String, unique=True, index=True, nullable=True)  # New field for phone numbers
     name = Column(String, default="Valued Customer")
     email = Column(String, default="unknown@example.com")
     preferences = Column(Text, default="")  # e.g., favorite dishes, dietary restrictions
@@ -241,8 +242,8 @@ def process_order_flow(user_id: str, message: str) -> str:
                 asyncio.create_task(save_order())
                 # Clear conversation state for order flow.
                 del user_state[user_id]
-                # For personalization, try retrieving the user's email.
-                email = "customer@example.com"  # In a real system, use UserProfile info.
+                # Retrieve email from user profile if available (here using a placeholder)
+                email = "customer@example.com"
                 payment_data = create_paystack_payment_link(email, total_price * 100, order_id)
                 if payment_data.get("status"):
                     payment_link = payment_data["data"]["authorization_url"]
@@ -257,15 +258,19 @@ def process_order_flow(user_id: str, message: str) -> str:
     return ""
 
 # --- User Profile Functions ---
-async def get_or_create_user_profile(user_id: str) -> UserProfile:
-    """Retrieve an existing profile or create a new one."""
+async def get_or_create_user_profile(user_id: str, phone_number: str = None) -> UserProfile:
+    """Retrieve an existing profile or create a new one with user_id and phone_number."""
     async with async_session() as session:
         result = await session.execute(
             UserProfile.__table__.select().where(UserProfile.user_id == user_id)
         )
         profile = result.scalar_one_or_none()
         if profile is None:
-            profile = UserProfile(user_id=user_id, last_interaction=datetime.utcnow())
+            profile = UserProfile(
+                user_id=user_id,
+                phone_number=phone_number,
+                last_interaction=datetime.utcnow()
+            )
             session.add(profile)
             await session.commit()
         return profile
@@ -284,8 +289,6 @@ async def update_user_last_interaction(user_id: str):
 # --- Proactive Engagement: Warm Greetings ---
 async def send_proactive_greeting(user_id: str):
     """Simulate sending a proactive greeting if the user has been inactive."""
-    # In a real system, you might schedule this using a job scheduler.
-    # Here we just simulate a warm greeting message.
     greeting = "Hi again! We miss you. Would you like to see our new menu items or get personalized recommendations?"
     await log_chat_to_db(user_id, "outbound", greeting)
     return greeting
@@ -305,11 +308,13 @@ async def chatbot_response(request: Request, background_tasks: BackgroundTasks):
     Supports text queries, image queries, and advanced logic.
     Expects JSON payload with:
       - 'user_id'
+      - 'phone_number'
       - 'message'
       - Optionally, 'is_image': true and 'image_base64'
     """
     data = await request.json()
     user_id = data.get("user_id")
+    phone_number = data.get("phone_number")  # New field for phone number
     user_message = data.get("message", "").strip()
     is_image = data.get("is_image", False)
     image_b64 = data.get("image_base64", None)
@@ -319,11 +324,11 @@ async def chatbot_response(request: Request, background_tasks: BackgroundTasks):
 
     # Log inbound message
     background_tasks.add_task(log_chat_to_db, user_id, "inbound", user_message)
-    # Update user last interaction and profile
+    # Update user last interaction and create profile with phone number
     await update_user_last_interaction(user_id)
-    await get_or_create_user_profile(user_id)
+    await get_or_create_user_profile(user_id, phone_number)
 
-    # Handle voice/image queries if applicable
+    # Handle image queries
     if is_image and image_b64:
         if len(image_b64) >= 180_000:
             raise HTTPException(status_code=400, detail="Image too large.")
@@ -334,7 +339,7 @@ async def chatbot_response(request: Request, background_tasks: BackgroundTasks):
     sentiment_score = analyze_sentiment(user_message)
     background_tasks.add_task(log_sentiment, user_id, user_message, sentiment_score)
 
-    # Adjust response tone based on sentiment (for demonstration)
+    # Adjust response tone based on sentiment
     sentiment_modifier = ""
     if sentiment_score < -0.3:
         sentiment_modifier = "I'm sorry if you're having a tough time. "
@@ -426,6 +431,7 @@ async def get_user_profile(user_id: str):
     profile = await get_or_create_user_profile(user_id)
     return {
         "user_id": profile.user_id,
+        "phone_number": profile.phone_number,
         "name": profile.name,
         "email": profile.email,
         "preferences": profile.preferences,
@@ -465,9 +471,8 @@ async def process_voice(file: UploadFile = File(...)):
     Accept a voice file upload, perform speech-to-text (simulated), and process the resulting text.
     In production, integrate with a real STT service.
     """
-    # Simulated Speech-to-Text: read file bytes and decode (for demo, just return a fixed string)
     contents = await file.read()
-    # In real implementation, send `contents` to an STT service.
+    # Simulated Speech-to-Text: In real implementation, send `contents` to an STT service.
     simulated_text = "Simulated speech-to-text conversion result."
     return {"transcription": simulated_text}
 
@@ -480,7 +485,7 @@ async def payment_callback(request: Request):
     """
     data = await request.json()
     # Extract order reference and update order status accordingly.
-    # This is a stub – in production, verify callback signature, extract data, and update DB.
+    # In production, verify callback signature and extract data.
     order_id = data.get("reference")
     new_status = data.get("status", "Paid")
     async with async_session() as session: