Update app.py

app.py

@@ -289,6 +289,117 @@ Rules:
                 "total_messages": len(self.manager.get_history(chat_id, user_id))
             }
             
+        except Exception as e:
+            logger.error(f"Error processing query for chat {chat_id}, user {user_id}: {e}")
+            return {
+                "error": f"I apologize, but I encountered an error: {str(e)}. Please try again or rephrase your question.",
+                "chat_id": chat_id or self.manager.generate_chat_id(user_id) if user_id else None,
+                "is_new_chat": True,
+                "user_message_id": None,
+                "assistant_message_id": None,
+                "total_messages": 0
+            }
+            
+    def get_chat_info(self, chat_id: str, user_id: str) -> Dict[str, Any]:
+        """
+        Get information about a specific chat for a specific user
+        
+        Args:
+            chat_id: The chat ID to get information for
+            user_id: The user ID for authentication
+            
+        Returns:
+            Dictionary with chat information
+        """
+        return self.manager.get_chat_info(chat_id, user_id)
+    
+    def get_all_chats(self, user_id: str) -> List[Dict[str, Any]]:
+        """
+        Get all chat sessions for a specific user
+        
+        Args:
+            user_id: The user ID to get sessions for
+            
+        Returns:
+            List of chat session information
+        """
+        return self.manager.get_all_chat_sessions(user_id)
+    
+    def clear_history(self, chat_id: str, user_id: str) -> bool:
+        """
+        Clear conversation history for a specific chat and user
+        
+        Args:
+            chat_id: The ID of the chat to clear
+            user_id: The user ID for authentication
+        
+        Returns:
+            True if deletion was successful, False otherwise
+        """
+        logger.info(f"Clearing history for chat: {chat_id}, user: {user_id}")
+        return self.manager.delete_history(chat_id, user_id)
+    
+    def get_messages(self, chat_id: str, user_id: str) -> List[Dict[str, Any]]:
+        """
+        Get all messages for a specific chat and user
+        
+        Args:
+            chat_id: The chat ID to get messages for
+            user_id: The user ID for authentication
+            
+        Returns:
+            List of messages with their IDs
+        """
+        return self.manager.get_history(chat_id, user_id).tool_calls
+                    ]
+                })
+                
+                # Execute all tool calls
+                for tool_call in message.tool_calls:
+                    function_name = tool_call.function.name
+                    function_args = json.loads(tool_call.function.arguments)
+                    
+                    logger.info(f"Calling function: {function_name} with args: {function_args}")
+                    
+                    # Call the function
+                    function_result = self.call_function(function_name, function_args)
+                    
+                    # Add function result to messages
+                    messages.append({
+                        "role": "tool",
+                        "tool_call_id": tool_call.id,
+                        "content": json.dumps(function_result)
+                    })
+                
+                # Add final system prompt for generating response
+                messages.append({
+                    "role": "system",
+                    "content": self.final_system
+                })
+                
+                # Get final response
+                final_response = self.client.chat.completions.create(
+                    model=self.config.model_name,
+                    messages=messages,
+                    temperature=self.config.temperature
+                )
+                response_content = final_response.choices[0].message.content
+            else:
+                response_content = message.content
+
+            # Add messages to conversation history with unique IDs
+            user_msg = self.manager.add_message(chat_id, user_id, "user", user_message, image_url)
+            assistant_msg = self.manager.add_message(chat_id, user_id, "assistant", response_content)
+            
+            return {
+                "response": response_content,
+                "chat_id": chat_id,
+                "is_new_chat": is_new_chat,
+                "user_message_id": user_msg.get("message_id"),
+                "assistant_message_id": assistant_msg.get("message_id"),
+                "total_messages": len(self.manager.get_history(chat_id, user_id))
+            }
+            
         except Exception as e:
             logger.error(f"Error processing query for chat {chat_id}, user {user_id}: {e}")
             return {
@@ -299,7 +410,7 @@ Rules:
                 "assistant_message_id": None,
                 "total_messages": 0
             }.tool_calls
-            ]
+                    ]
                 })
                 
                 # Execute all tool calls