Update app.py

app.py

@@ -457,7 +457,8 @@ async def chatbot_response(request: Request, background_tasks: BackgroundTasks):
         if selected_dish:
             state = ConversationState()
             state.flow = "order"
-            state.step = 1
+            # **** FIX: Set step to 2 since the dish is already selected ****
+            state.step = 2  
             state.data["dish"] = selected_dish
             state.update_last_active()
             user_state[user_id] = state
@@ -496,7 +497,6 @@ async def chatbot_response(request: Request, background_tasks: BackgroundTasks):
         return JSONResponse(content={"response": sentiment_modifier + response_text})
 
     # --- Fallback: LLM Response Streaming with Conversation Context ---
-    # Build recent conversation history (e.g., last 5 messages)
     recent_context = conversation_context.get(user_id, [])[-5:]
     context_str = "\n".join([f"{entry['role'].capitalize()}: {entry['message']}" for entry in recent_context])
     prompt = f"Conversation context:\n{context_str}\nUser query: {user_message}\nGenerate a helpful, personalized response for a restaurant chatbot."
@@ -505,15 +505,13 @@ async def chatbot_response(request: Request, background_tasks: BackgroundTasks):
             yield chunk
     fallback_log = f"LLM fallback response for prompt: {prompt}"
     background_tasks.add_task(log_chat_to_db, user_id, "outbound", fallback_log)
-    # (Optionally, you might want to capture and add the streamed response to conversation_context)
     return StreamingResponse(stream_response(), media_type="text/plain")
 
-    # --- Chat History Endpoint ---
+# --- Other Endpoints (Chat History, Order Details, User Profile, Analytics, Voice, Payment Callback) ---
+# ... (Implement other endpoints as needed) ...
+
 @app.get("/chat_history/{user_id}")
 async def get_chat_history(user_id: str):
-    """
-    Retrieve chat history for a user.
-    """
     async with async_session() as session:
         result = await session.execute(
             ChatHistory.__table__.select().where(ChatHistory.user_id == user_id)
@@ -521,12 +519,8 @@ async def get_chat_history(user_id: str):
         history = result.fetchall()
         return [dict(row) for row in history]
 
-# --- Order Details Endpoint ---
 @app.get("/order/{order_id}")
 async def get_order(order_id: str):
-    """
-    Retrieve details for a specific order.
-    """
     async with async_session() as session:
         result = await session.execute(
             Order.__table__.select().where(Order.order_id == order_id)
@@ -537,12 +531,8 @@ async def get_order(order_id: str):
         else:
             raise HTTPException(status_code=404, detail="Order not found.")
 
-# --- User Profile Endpoint ---
 @app.get("/user_profile/{user_id}")
 async def get_user_profile(user_id: str):
-    """
-    Retrieve the user profile.
-    """
     profile = await get_or_create_user_profile(user_id)
     return {
         "user_id": profile.user_id,
@@ -553,54 +543,30 @@ async def get_user_profile(user_id: str):
         "last_interaction": profile.last_interaction.isoformat()
     }
 
-# --- Analytics Endpoint ---
 @app.get("/analytics")
 async def get_analytics():
-    """
-    Simple analytics dashboard endpoint.
-    Returns counts of messages, orders, and average sentiment.
-    """
     async with async_session() as session:
-        # Total messages count
         msg_result = await session.execute(ChatHistory.__table__.count())
         total_messages = msg_result.scalar() or 0
-
-        # Total orders count
         order_result = await session.execute(Order.__table__.count())
         total_orders = order_result.scalar() or 0
-
-        # Average sentiment score
         sentiment_result = await session.execute("SELECT AVG(sentiment_score) FROM sentiment_logs")
         avg_sentiment = sentiment_result.scalar() or 0
-
     return {
         "total_messages": total_messages,
         "total_orders": total_orders,
         "average_sentiment": avg_sentiment
     }
 
-# --- Voice Integration Endpoint ---
 @app.post("/voice")
 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.
-    """
     contents = await file.read()
-    # 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}
 
-# --- Payment Callback Endpoint (Stub) ---
 @app.post("/payment_callback")
 async def payment_callback(request: Request):
-    """
-    Endpoint to handle payment callbacks from Paystack.
-    Update order status based on callback data.
-    """
     data = await request.json()
-    # Extract order reference and update order status accordingly.
-    # 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:
@@ -617,4 +583,4 @@ async def payment_callback(request: Request):
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)